1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
|
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 testAnsiFilterReadCloser(t *testing.T, input string, expected string) {
reader := newStringReadCloser(input)
cleanReader := util.NewAnsiFilterReadCloser(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 read %d bytes but read %d\n", len(expected), n)
}
if result != expected {
t.Errorf("Expected string `%s` but read `%s`\n", "abc", result)
}
}
func testAnsiFilterWriter(t *testing.T, input string, expected string) {
writer := new(strings.Builder)
w := util.NewAnsiFilterWriter(writer)
n, err := w.Write([]byte(input))
if err != nil {
t.Fatal("Error while writting input", err)
}
// NOTE: we expect the size of input to be written to indicate those bytes have been consumed by the writer
if n != len(input) {
t.Errorf("Expected to write %d bytes but wrote %d\n", len(input), n)
}
result := writer.String()
if result != expected {
t.Errorf("Expected to write `%s` but wrote `%s`\n", expected, result)
}
}
func TestAnsiFilterColorsReadCloser(t *testing.T) {
testAnsiFilterReadCloser(t, "a\x1b[31mbc", "abc")
testAnsiFilterReadCloser(t, "[\x1b[32minfo\x1b[39m]", "[info]")
testAnsiFilterReadCloser(t,
"FoundryVTT | 2024-10-06 20:11:14 | [\x1b[32minfo\x1b[39m] Running on Node.js - Version 20.17.0\n"+
"FoundryVTT | 2024-10-06 20:11:14 | [\x1b[32minfo\x1b[39m] Foundry Virtual Tabletop - Version 12 Build 331\n",
"FoundryVTT | 2024-10-06 20:11:14 | [info] Running on Node.js - Version 20.17.0\n"+
"FoundryVTT | 2024-10-06 20:11:14 | [info] Foundry Virtual Tabletop - Version 12 Build 331\n")
}
func TestAnsiFilterColorsWriter(t *testing.T) {
testAnsiFilterWriter(t, "a\x1b[31mbc", "abc")
testAnsiFilterWriter(t, "[\x1b[32minfo\x1b[39m]", "[info]")
testAnsiFilterWriter(t,
"FoundryVTT | 2024-10-06 20:11:14 | [\x1b[32minfo\x1b[39m] Running on Node.js - Version 20.17.0\n"+
"FoundryVTT | 2024-10-06 20:11:14 | [\x1b[32minfo\x1b[39m] Foundry Virtual Tabletop - Version 12 Build 331\n",
"FoundryVTT | 2024-10-06 20:11:14 | [info] Running on Node.js - Version 20.17.0\n"+
"FoundryVTT | 2024-10-06 20:11:14 | [info] Foundry Virtual Tabletop - Version 12 Build 331\n")
}
|
