From 13c771c84e6b86cebe6df14d0e92001f940581ba Mon Sep 17 00:00:00 2001
From: JP Appel <jeanpierre.appel01@gmail.com>
Date: Tue, 8 Oct 2024 13:17:03 -0400
Subject: Move ansi stripper to util package

---
 util/util_test.go | 50 ++++++++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 50 insertions(+)
 create mode 100644 util/util_test.go

(limited to 'util/util_test.go')

diff --git a/util/util_test.go b/util/util_test.go
new file mode 100644
index 0000000..1c4ae39
--- /dev/null
+++ b/util/util_test.go
@@ -0,0 +1,50 @@
+package util_test;
+
+import (
+	"io"
+	"nonsense-time/util"
+	"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 := util.NewAnsiStripper(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]")
+}
-- 
cgit v1.2.3