add more tests

gookit · Apr 1, 2021 · f7f679a · f7f679a
1 parent 9e0126d
commit f7f679a
Show file tree

Hide file tree

Showing 3 changed files with 34 additions and 1 deletion.
diff --git a/color_test.go b/color_test.go
@@ -121,6 +121,26 @@ func TestSet(t *testing.T) {
 	is.Equal("\x1b[0m", str)
 }
 
+func TestSupportColor(t *testing.T) {
+	is := assert.New(t)
+
+	if is.True(SupportTrueColor()) {
+		is.True(SupportColor())
+		is.True(Support256Color())
+	}
+
+	if is.True(Support256Color()) {
+		is.True(SupportColor())
+	} else {
+		is.False(SupportTrueColor())
+	}
+
+	if is.False(SupportColor()) {
+		is.False(Support256Color())
+		is.False(SupportTrueColor())
+	}
+}
+
 func TestRenderCode(t *testing.T) {
 	// force open color render for testing
 	oldVal = ForceColor()

diff --git a/detect_env.go b/detect_env.go
@@ -45,7 +45,7 @@ func detectTermColorLevel() (level terminfo.ColorLevel, needVTP bool) {
 	// - support true-color
 	// 	TERMINAL_EMULATOR=JetBrains-JediTerm
 	if val := os.Getenv("TERMINAL_EMULATOR"); val == "JetBrains-JediTerm" {
-		debugf("on JetBrains-JediTerm, is win: %v", isWin)
+		debugf("True Color support on JetBrains-JediTerm, is win: %v", isWin)
 		return terminfo.ColorLevelMillions, isWin
 	}
 

diff --git a/utils_test.go b/utils_test.go
@@ -52,6 +52,13 @@ func TestDetectColorLevel(t *testing.T) {
 		is.True(IsSupportColor())
 	})
 
+	// TERMINAL_EMULATOR=JetBrains-JediTerm
+	mockEnvValue("TERMINAL_EMULATOR", "JetBrains-JediTerm", func(val string) {
+		is.Equal(LevelRgb, DetectColorLevel())
+		is.True(IsSupport16Color())
+		is.True(IsSupportColor())
+	})
+
 	// "COLORTERM=truecolor"
 	mockEnvValue("COLORTERM", "truecolor", func(val string) {
 		is.Equal("truecolor", val)
@@ -80,9 +87,15 @@ func TestIsDetectColorLevel_win(t *testing.T) {
 	mockEnvValue("ConEmuANSI", "ON", func(_ string) {
 		is.Equal(LevelRgb, DetectColorLevel())
 		is.True(IsSupportColor())
+		is.True(IsSupport256Color())
 		is.True(IsSupportTrueColor())
 	})
 
+	// WSL_DISTRO_NAME=Debian
+	mockEnvValue("WSL_DISTRO_NAME", "Debian", func(_ string) {
+		is.True(IsSupportColor())
+	})
+
 	// ANSICON
 	mockEnvValue("ANSICON", "189x2000 (189x43)", func(_ string) {
 		is.True(IsSupportColor())