From 358ea1fc254c33e4bebc8e2919c095288770a4ab Mon Sep 17 00:00:00 2001
From: bashbunni <bunni@bashbunni.dev>
Date: Thu, 15 Aug 2024 14:25:32 -0700
Subject: [PATCH] test(table): add TestStyleFunc

---
 go.mod                              |  2 ++
 go.sum                              |  4 ++++
 table/table_test.go                 | 36 +++++++++++++++++++++++------
 table/testdata/TestStyleFunc.golden | 21 +++++++++++++++++
 4 files changed, 56 insertions(+), 7 deletions(-)
 create mode 100644 table/testdata/TestStyleFunc.golden

diff --git a/go.mod b/go.mod
index 03a0f7b..1b14080 100644
--- a/go.mod
+++ b/go.mod
@@ -9,6 +9,7 @@ require (
 	github.com/charmbracelet/harmonica v0.2.0
 	github.com/charmbracelet/lipgloss v0.12.1
 	github.com/charmbracelet/x/ansi v0.1.4
+	github.com/charmbracelet/x/exp/golden v0.0.0-20240815200342-61de596daa2b
 	github.com/dustin/go-humanize v1.0.1
 	github.com/lucasb-eyer/go-colorful v1.2.0
 	github.com/mattn/go-runewidth v0.0.16
@@ -19,6 +20,7 @@ require (
 
 require (
 	github.com/aymanbagabas/go-osc52/v2 v2.0.1 // indirect
+	github.com/aymanbagabas/go-udiff v0.2.0 // indirect
 	github.com/charmbracelet/x/input v0.1.0 // indirect
 	github.com/charmbracelet/x/term v0.1.1 // indirect
 	github.com/charmbracelet/x/windows v0.1.0 // indirect
diff --git a/go.sum b/go.sum
index 3118009..f555560 100644
--- a/go.sum
+++ b/go.sum
@@ -4,6 +4,8 @@ github.com/atotto/clipboard v0.1.4 h1:EH0zSVneZPSuFR11BlR9YppQTVDbh5+16AmcJi4g1z
 github.com/atotto/clipboard v0.1.4/go.mod h1:ZY9tmq7sm5xIbd9bOK4onWV4S6X0u6GY7Vn0Yu86PYI=
 github.com/aymanbagabas/go-osc52/v2 v2.0.1 h1:HwpRHbFMcZLEVr42D4p7XBqjyuxQH5SMiErDT4WkJ2k=
 github.com/aymanbagabas/go-osc52/v2 v2.0.1/go.mod h1:uYgXzlJ7ZpABp8OJ+exZzJJhRNQ2ASbcXHWsFqH8hp8=
+github.com/aymanbagabas/go-udiff v0.2.0 h1:TK0fH4MteXUDspT88n8CKzvK0X9O2xu9yQjWpi6yML8=
+github.com/aymanbagabas/go-udiff v0.2.0/go.mod h1:RE4Ex0qsGkTAJoQdQQCA0uG+nAzJO/pI/QwceO5fgrA=
 github.com/charmbracelet/bubbletea v0.26.6 h1:zTCWSuST+3yZYZnVSvbXwKOPRSNZceVeqpzOLN2zq1s=
 github.com/charmbracelet/bubbletea v0.26.6/go.mod h1:dz8CWPlfCCGLFbBlTY4N7bjLiyOGDJEnd2Muu7pOWhk=
 github.com/charmbracelet/harmonica v0.2.0 h1:8NxJWRWg/bzKqqEaaeFNipOu77YR5t8aSwG4pgaUBiQ=
@@ -12,6 +14,8 @@ github.com/charmbracelet/lipgloss v0.12.1 h1:/gmzszl+pedQpjCOH+wFkZr/N90Snz40J/N
 github.com/charmbracelet/lipgloss v0.12.1/go.mod h1:V2CiwIuhx9S1S1ZlADfOj9HmxeMAORuz5izHb0zGbB8=
 github.com/charmbracelet/x/ansi v0.1.4 h1:IEU3D6+dWwPSgZ6HBH+v6oUuZ/nVawMiWj5831KfiLM=
 github.com/charmbracelet/x/ansi v0.1.4/go.mod h1:dk73KoMTT5AX5BsX0KrqhsTqAnhZZoCBjs7dGWp4Ktw=
+github.com/charmbracelet/x/exp/golden v0.0.0-20240815200342-61de596daa2b h1:MnAMdlwSltxJyULnrYbkZpp4k58Co7Tah3ciKhSNo0Q=
+github.com/charmbracelet/x/exp/golden v0.0.0-20240815200342-61de596daa2b/go.mod h1:wDlXFlCrmJ8J+swcL/MnGUuYnqgQdW9rhSD61oNMb6U=
 github.com/charmbracelet/x/input v0.1.0 h1:TEsGSfZYQyOtp+STIjyBq6tpRaorH0qpwZUj8DavAhQ=
 github.com/charmbracelet/x/input v0.1.0/go.mod h1:ZZwaBxPF7IG8gWWzPUVqHEtWhc1+HXJPNuerJGRGZ28=
 github.com/charmbracelet/x/term v0.1.1 h1:3cosVAiPOig+EV4X9U+3LDgtwwAoEzJjNdwbXDjF6yI=
diff --git a/table/table_test.go b/table/table_test.go
index 24f075f..21ecb13 100644
--- a/table/table_test.go
+++ b/table/table_test.go
@@ -5,6 +5,8 @@ import (
 	"testing"
 
 	"github.com/charmbracelet/lipgloss"
+	"github.com/charmbracelet/x/ansi"
+	"github.com/charmbracelet/x/exp/golden"
 )
 
 func TestFromValues(t *testing.T) {
@@ -58,13 +60,11 @@ func deepEqual(a, b []Row) bool {
 	return true
 }
 
-var (
-	cols = []Column{
-		{Title: "col1", Width: 10},
-		{Title: "col2", Width: 10},
-		{Title: "col3", Width: 10},
-	}
-)
+var cols = []Column{
+	{Title: "col1", Width: 10},
+	{Title: "col2", Width: 10},
+	{Title: "col3", Width: 10},
+}
 
 func TestRenderRow(t *testing.T) {
 	tests := []struct {
@@ -109,6 +109,7 @@ func TestRenderRow(t *testing.T) {
 		})
 	}
 }
+
 func TestRenderRowStyleFunc(t *testing.T) {
 	tests := []struct {
 		name     string
@@ -139,3 +140,24 @@ func TestRenderRowStyleFunc(t *testing.T) {
 		})
 	}
 }
+
+func TestStyleFunc(t *testing.T) {
+	biscuits := New(
+		WithColumns([]Column{
+			{Title: "Name", Width: 25},
+			{Title: "Country of Origin", Width: 16},
+			{Title: "Dunk-able", Width: 12},
+		}),
+		WithRows([]Row{
+			{"Chocolate Digestives", "UK", "Yes"},
+			{"Tim Tams", "Australia", "No"},
+			{"Hobnobs", "UK", "Yes"},
+		}),
+		WithStyleFunc(func(row, col int, value string) lipgloss.Style {
+			return lipgloss.NewStyle()
+		}),
+	)
+
+	got := ansi.Strip(biscuits.View())
+	golden.RequireEqual(t, []byte(got))
+}
diff --git a/table/testdata/TestStyleFunc.golden b/table/testdata/TestStyleFunc.golden
new file mode 100644
index 0000000..6ba436c
--- /dev/null
+++ b/table/testdata/TestStyleFunc.golden
@@ -0,0 +1,21 @@
+ Name                       Country of Orig…  Dunk-able    
+ Chocolate Digestives       UK                Yes          
+ Tim Tams                   Australia         No           
+ Hobnobs                    UK                Yes          
+                                                           
+                                                           
+                                                           
+                                                           
+                                                           
+                                                           
+                                                           
+                                                           
+                                                           
+                                                           
+                                                           
+                                                           
+                                                           
+                                                           
+                                                           
+                                                           
+                                                           
\ No newline at end of file