File: writer_test.go

package info (click to toggle)
golang-github-muesli-ansi 0.0~git20211031.c9f0611-2
  • links: PTS, VCS
  • area: main
  • in suites: bookworm, forky, sid, trixie
  • size: 120 kB
  • sloc: makefile: 2
file content (81 lines) | stat: -rw-r--r-- 1,983 bytes parent folder | download
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
package compressor

import (
	"bytes"
	"errors"
	"io/ioutil"
	"testing"
)

func TestWriter_Write(t *testing.T) {
	tt := []struct {
		Input    []byte
		Expected []byte
	}{
		// nothing to compress
		{
			[]byte("\x1B[38;2;249;38;114mre\x1B[0m\x1B[38;2;249;38;115mflow\x1B[0m"),
			[]byte("\x1B[38;2;249;38;114mre\x1B[0m\x1B[38;2;249;38;115mflow\x1B[0m"),
		},
		// reset sequence before setting the same styles again
		{
			[]byte("\x1B[38;2;249;38;114mre\x1B[0m\x1B[38;2;249;38;114mfl\x1B[0m\x1B[38;2;249;38;114mow\x1B[0m"),
			[]byte("\x1B[38;2;249;38;114mreflow\x1B[0m"),
		},
	}

	for i, tc := range tt {
		forward := &bytes.Buffer{}
		w := &Writer{Forward: forward}

		n, err := w.Write(tc.Input)
		if err != nil {
			t.Fatalf("Test %d, err should be nil, but got %v", i, err)
		}
		if err := w.Close(); err != nil {
			t.Fatalf("Test %d, err should be nil, but got %v", i, err)
		}

		if l := len(tc.Input); n != l {
			t.Fatalf("Test %d, n should be %d, got %d", i, l, n)
		}

		if b := forward.Bytes(); !bytes.Equal(b, tc.Expected) {
			// fmt.Println(strings.ReplaceAll(string(tc.Expected), string(ansi.Marker), ""))
			// fmt.Println(strings.ReplaceAll(string(b), string(ansi.Marker), ""))

			t.Fatalf("Test %d, forward should be wrote by %v, but got %v", i, tc.Expected, b)
		}
	}
}

var fakeErr = errors.New("fake error")

type fakeWriter struct{}

func (fakeWriter) Write(_ []byte) (int, error) {
	return 0, fakeErr
}

func TestWriter_Write_Error(t *testing.T) {
	t.Parallel()

	w := &Writer{Forward: fakeWriter{}}

	_, err := w.Write([]byte("foo"))
	if err != fakeErr {
		t.Fatalf("err should be fakeErr, but got %v", err)
	}
}

// go test -bench=BenchmarkWriter_Write -benchmem -count=4
func BenchmarkWriter_Write(b *testing.B) {
	buf := []byte("\x1B[38;2;249;38;114mre\x1B[0m\x1B[38;2;249;38;114mfl\x1B[0m\x1B[38;2;249;38;114mow\x1B[0m")
	w := &Writer{Forward: ioutil.Discard}

	b.ReportAllocs()
	b.ResetTimer()
	for i := 0; i < b.N; i++ {
		_, _ = w.Write(buf)
	}
}