From 13c771c84e6b86cebe6df14d0e92001f940581ba Mon Sep 17 00:00:00 2001 From: JP Appel Date: Tue, 8 Oct 2024 13:17:03 -0400 Subject: Move ansi stripper to util package --- api/docker_test.go | 50 -------------------------------------------------- 1 file changed, 50 deletions(-) delete mode 100644 api/docker_test.go (limited to 'api/docker_test.go') diff --git a/api/docker_test.go b/api/docker_test.go deleted file mode 100644 index 7d85803..0000000 --- a/api/docker_test.go +++ /dev/null @@ -1,50 +0,0 @@ -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.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