add tests
diff --git a/colorable_test.go b/colorable_test.go
index 3069869..7244639 100644
--- a/colorable_test.go
+++ b/colorable_test.go
@@ -64,6 +64,24 @@
}
}
+func TestNonColorableESC(t *testing.T) {
+ var b bytes.Buffer
+ c := NewNonColorable(&b)
+ c.Write([]byte{0x1b})
+ if b.Len() > 0 {
+ t.Fatalf("0 bytes expected, got %d", b.Len())
+ }
+}
+
+func TestNonColorableBadESC(t *testing.T) {
+ var b bytes.Buffer
+ c := NewNonColorable(&b)
+ c.Write([]byte{0x1b, 0x1b})
+ if b.Len() > 0 {
+ t.Fatalf("0 bytes expected, got %d", b.Len())
+ }
+}
+
func TestColorable(t *testing.T) {
if runtime.GOOS == "windows" {
t.Skipf("skip this test on windows")
