Tests && gomod

termext/termcolor_test.go

@@ -0,0 +1,40 @@
+package termext
+
+import (
+	"math/rand"
+	"testing"
+)
+
+func init() {
+	rand.Seed(0)
+}
+
+func TestSupportsColors(t *testing.T) {
+	SupportsColors() // should not error
+}
+
+func TestColor(t *testing.T) {
+	assertEqual(t, Red("test"), "\033[31mtest\u001B[0m")
+	assertEqual(t, Green("test"), "\033[32mtest\u001B[0m")
+	assertEqual(t, Yellow("test"), "\033[33mtest\u001B[0m")
+	assertEqual(t, Blue("test"), "\033[34mtest\u001B[0m")
+	assertEqual(t, Purple("test"), "\033[35mtest\u001B[0m")
+	assertEqual(t, Cyan("test"), "\033[36mtest\u001B[0m")
+	assertEqual(t, Gray("test"), "\033[37mtest\u001B[0m")
+	assertEqual(t, White("test"), "\033[97mtest\u001B[0m")
+
+	assertEqual(t, CleanString(Red("test")), "test")
+	assertEqual(t, CleanString(Green("test")), "test")
+	assertEqual(t, CleanString(Yellow("test")), "test")
+	assertEqual(t, CleanString(Blue("test")), "test")
+	assertEqual(t, CleanString(Purple("test")), "test")
+	assertEqual(t, CleanString(Cyan("test")), "test")
+	assertEqual(t, CleanString(Gray("test")), "test")
+	assertEqual(t, CleanString(White("test")), "test")
+}
+
+func assertEqual(t *testing.T, actual string, expected string) {
+	if actual != expected {
+		t.Errorf("values differ: Actual: '%v', Expected: '%v'", actual, expected)
+	}
+}