Add middleware to strip ansi escape sequences

1 files changed, 50 insertions, 0 deletions

diff --git a/api/docker_test.go b/api/docker_test.go
new file mode 100644
index 0000000..acd92bc
--- /dev/null
+++ b/api/docker_test.go
@@ -0,0 +1,50 @@
+package api_test
+
+import (
+	"io"
+	"nonsense-time/api"
+	"strings"
+	"testing"
+)
+
+type stringReadCloser struct {
+	*strings.Reader
+}
+
+func newStringReadCloser(s string) stringReadCloser {
+	rc := stringReadCloser{}
+	rc.Reader = strings.NewReader(s)
+
+	return rc
+}
+
+func (stringReadCloser) Close() error {
+	return nil
+}
+
+func testAnsiStripper(t *testing.T, input string, expected string) {
+	reader := newStringReadCloser(input)
+	cleanReader := api.StripAnsi(reader)
+	defer cleanReader.Close()
+
+	buf := new(strings.Builder)
+	n, err := io.Copy(buf, cleanReader)
+	if err != nil {
+		t.Fatal("Error while copying cleaned text to output", err)
+	}
+
+	result := buf.String()
+
+	if n != int64(len(expected)) {
+		t.Errorf("Expected to write %d characters but wrote %d\n", 3, n)
+	}
+
+	if result != expected {
+		t.Errorf("Expected string `%s` but wrote `%s`", "abc", result)
+	}
+}
+
+func TestStripAnsiColors(t *testing.T) {
+	testAnsiStripper(t, "a\x1b[31mbc", "abc")
+    testAnsiStripper(t, "[\x1b[32minfo\x1b[39m]", "[info]")
+}