git.iko.soy/crush
on 21ca2ae help sign in
crush/internal/exp/diffview/diffview_test.go
212 lines 5.6 kB history raw
1package diffview_test
3import (
4 _ "embed"
5 "fmt"
6 "strings"
7 "testing"
9 "github.com/charmbracelet/x/ansi"
10 "github.com/charmbracelet/x/exp/golden"
11 "github.com/opencode-ai/opencode/internal/exp/diffview"
14//go:embed testdata/TestDefault.before
15var TestDefaultBefore string
17//go:embed testdata/TestDefault.after
18var TestDefaultAfter string
20//go:embed testdata/TestMultipleHunks.before
21var TestMultipleHunksBefore string
23//go:embed testdata/TestMultipleHunks.after
24var TestMultipleHunksAfter string
26//go:embed testdata/TestNarrow.before
27var TestNarrowBefore string
29//go:embed testdata/TestNarrow.after
30var TestNarrowAfter string
32type (
33 TestFunc func(dv *diffview.DiffView) *diffview.DiffView
34 TestFuncs map[string]TestFunc
37var (
38 UnifiedFunc = func(dv *diffview.DiffView) *diffview.DiffView {
39 return dv.Unified()
40 }
41 SplitFunc = func(dv *diffview.DiffView) *diffview.DiffView {
42 return dv.Split()
43 }
45 DefaultFunc = func(dv *diffview.DiffView) *diffview.DiffView {
46 return dv.
47 Before("main.go", TestDefaultBefore).
48 After("main.go", TestDefaultAfter)
49 }
50 NoLineNumbersFunc = func(dv *diffview.DiffView) *diffview.DiffView {
51 return dv.
52 Before("main.go", TestDefaultBefore).
53 After("main.go", TestDefaultAfter).
54 LineNumbers(false)
55 }
56 MultipleHunksFunc = func(dv *diffview.DiffView) *diffview.DiffView {
57 return dv.
58 Before("main.go", TestMultipleHunksBefore).
59 After("main.go", TestMultipleHunksAfter)
60 }
61 CustomContextLinesFunc = func(dv *diffview.DiffView) *diffview.DiffView {
62 return dv.
63 Before("main.go", TestMultipleHunksBefore).
64 After("main.go", TestMultipleHunksAfter).
65 ContextLines(4)
66 }
67 NarrowFunc = func(dv *diffview.DiffView) *diffview.DiffView {
68 return dv.
69 Before("text.txt", TestNarrowBefore).
70 After("text.txt", TestNarrowAfter)
71 }
72 SmallWidthFunc = func(dv *diffview.DiffView) *diffview.DiffView {
73 return dv.
74 Before("main.go", TestMultipleHunksBefore).
75 After("main.go", TestMultipleHunksAfter).
76 Width(40)
77 }
78 LargeWidthFunc = func(dv *diffview.DiffView) *diffview.DiffView {
79 return dv.
80 Before("main.go", TestMultipleHunksBefore).
81 After("main.go", TestMultipleHunksAfter).
82 Width(120)
83 }
85 LightModeFunc = func(dv *diffview.DiffView) *diffview.DiffView {
86 return dv.Style(diffview.DefaultLightStyle)
87 }
88 DarkModeFunc = func(dv *diffview.DiffView) *diffview.DiffView {
89 return dv.Style(diffview.DefaultDarkStyle)
90 }
92 LayoutFuncs = TestFuncs{
93 "Unified": UnifiedFunc,
94 "Split": SplitFunc,
95 }
96 BehaviorFuncs = TestFuncs{
97 "Default": DefaultFunc,
98 "NoLineNumbers": NoLineNumbersFunc,
99 "MultipleHunks": MultipleHunksFunc,
100 "CustomContextLines": CustomContextLinesFunc,
101 "Narrow": NarrowFunc,
102 "SmallWidth": SmallWidthFunc,
103 "LargeWidth": LargeWidthFunc,
104 }
105 ThemeFuncs = TestFuncs{
106 "LightMode": LightModeFunc,
107 "DarkMode": DarkModeFunc,
108 }
111func TestDiffView(t *testing.T) {
112 for layoutName, layoutFunc := range LayoutFuncs {
113 t.Run(layoutName, func(t *testing.T) {
114 for behaviorName, behaviorFunc := range BehaviorFuncs {
115 t.Run(behaviorName, func(t *testing.T) {
116 for themeName, themeFunc := range ThemeFuncs {
117 t.Run(themeName, func(t *testing.T) {
118 dv := diffview.New()
119 dv = layoutFunc(dv)
120 dv = behaviorFunc(dv)
121 dv = themeFunc(dv)
123 output := dv.String()
124 golden.RequireEqual(t, []byte(output))
126 switch behaviorName {
127 case "SmallWidth":
128 assertLineWidth(t, 40, output)
129 case "LargeWidth":
130 assertLineWidth(t, 120, output)
131 }
132 })
133 }
134 })
135 }
136 })
137 }
140func TestDiffViewWidth(t *testing.T) {
141 for layoutName, layoutFunc := range LayoutFuncs {
142 t.Run(layoutName, func(t *testing.T) {
143 for themeName, themeFunc := range ThemeFuncs {
144 t.Run(themeName, func(t *testing.T) {
145 for width := 1; width <= 110; width++ {
146 if layoutName == "Unified" && width > 60 {
147 continue
148 }
150 t.Run(fmt.Sprintf("WidthOf%03d", width), func(t *testing.T) {
151 dv := diffview.New().
152 Before("main.go", TestMultipleHunksBefore).
153 After("main.go", TestMultipleHunksAfter).
154 Width(width)
155 dv = layoutFunc(dv)
156 dv = themeFunc(dv)
158 output := dv.String()
159 golden.RequireEqual(t, []byte(output))
161 assertLineWidth(t, width, output)
162 })
163 }
164 })
165 }
166 })
167 }
170func TestDiffViewHeight(t *testing.T) {
171 for layoutName, layoutFunc := range LayoutFuncs {
172 t.Run(layoutName, func(t *testing.T) {
173 for themeName, themeFunc := range ThemeFuncs {
174 t.Run(themeName, func(t *testing.T) {
175 for height := 1; height <= 20; height++ {
176 t.Run(fmt.Sprintf("HeightOf%03d", height), func(t *testing.T) {
177 dv := diffview.New().
178 Before("main.go", TestMultipleHunksBefore).
179 After("main.go", TestMultipleHunksAfter).
180 Height(height)
181 dv = layoutFunc(dv)
182 dv = themeFunc(dv)
184 output := dv.String()
185 golden.RequireEqual(t, []byte(output))
187 assertHeight(t, height, output)
188 })
189 }
190 })
191 }
192 })
193 }
196func assertLineWidth(t *testing.T, expected int, output string) {
197 var lineWidth int
198 for line := range strings.SplitSeq(output, "\n") {
199 lineWidth = max(lineWidth, ansi.StringWidth(line))
200 }
201 if lineWidth != expected {
202 t.Errorf("expected output width to be == %d, got %d", expected, lineWidth)
203 }
206func assertHeight(t *testing.T, expected int, output string) {
207 output = strings.TrimSuffix(output, "\n")
208 lines := strings.Count(output, "\n") + 1
209 if lines != expected {
210 t.Errorf("expected output height to be == %d, got %d", expected, lines)
211 }