go/texttable/texttable_test.go at main · jim-barber-he/go · GitHub

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
86
87
88
89
90
91
package texttable_test

import (
	"bytes"
	"testing"

	"github.com/jim-barber-he/go/texttable"
)

type Row struct {
	Name  string `title:"NAME"`
	Value string `title:"VALUE"`
}

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

	t.Run("Append", func(t *testing.T) {
		t.Parallel()

		tbl := texttable.Table[*Row]{
			Rows: []*Row{
				{Name: "a", Value: "a"},
				{Name: "b", Value: "b"},
			},
		}
		tbl.Append(&Row{Name: "c", Value: "c"})

		expectedLen := 3
		if len(tbl.Rows) != expectedLen {
			t.Errorf("Append() failed, expected %d, got %d", expectedLen, len(tbl.Rows))
		}

		var buf bytes.Buffer
		tbl.Write(&buf)
		table := buf.String()

		expected := "NAME  VALUE\na     a\nb     b\nc     c\n"
		if table != expected {
			t.Errorf("Append() failed, expected %s, got %s", expected, table)
		}
	})
}

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

	t.Run("Write", func(t *testing.T) {
		t.Parallel()

		tbl := texttable.Table[*Row]{
			Rows: []*Row{
				{Name: "a", Value: "a"},
				{Name: "b", Value: "b"},
			},
		}

		var buf bytes.Buffer
		tbl.Write(&buf)
		table := buf.String()

		expected := "NAME  VALUE\na     a\nb     b\n"
		if table != expected {
			t.Errorf("Append() failed, expected %s, got %s", expected, table)
		}
	})
}

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

	t.Run("WriteNoHeader", func(t *testing.T) {
		t.Parallel()

		tbl := texttable.Table[*Row]{
			Rows: []*Row{
				{Name: "a", Value: "a"},
				{Name: "b", Value: "b"},
			},
		}

		var buf bytes.Buffer
		tbl.WriteNoHeaders(&buf)
		table := buf.String()

		expected := "a  a\nb  b\n"
		if table != expected {
			t.Errorf("Append() failed, expected %s, got %s", expected, table)
		}
	})
}