mirror of
https://github.com/zoriya/bubbles.git
synced 2026-08-08 22:26:09 +00:00
114 lines
2.5 KiB
Go
114 lines
2.5 KiB
Go
package table
|
|
|
|
import (
|
|
"reflect"
|
|
"testing"
|
|
|
|
"github.com/charmbracelet/lipgloss"
|
|
"github.com/charmbracelet/x/ansi"
|
|
"github.com/charmbracelet/x/exp/golden"
|
|
)
|
|
|
|
func TestFromValues(t *testing.T) {
|
|
input := "foo1,bar1\nfoo2,bar2\nfoo3,bar3"
|
|
table := New().
|
|
Headers("Foo", "Bar")
|
|
table.FromValues(input, ",")
|
|
|
|
if len(table.rows) != 3 {
|
|
t.Fatalf("expect table to have 3 rows but it has %d", len(table.rows))
|
|
}
|
|
|
|
expect := [][]string{
|
|
{"foo1", "bar1"},
|
|
{"foo2", "bar2"},
|
|
{"foo3", "bar3"},
|
|
}
|
|
if !reflect.DeepEqual(table.rows, expect) {
|
|
t.Fatal("table rows is not equals to the input")
|
|
}
|
|
}
|
|
|
|
func TestFromValuesWithTabSeparator(t *testing.T) {
|
|
input := "foo1.\tbar1\nfoo,bar,baz\tbar,2"
|
|
table := New(WithColumns([]Column{{Title: "Foo"}, {Title: "Bar"}}))
|
|
table.FromValues(input, "\t")
|
|
|
|
if len(table.rows) != 2 {
|
|
t.Fatalf("expect table to have 2 rows but it has %d", len(table.rows))
|
|
}
|
|
|
|
expect := []Row{
|
|
{"foo1.", "bar1"},
|
|
{"foo,bar,baz", "bar,2"},
|
|
}
|
|
if !reflect.DeepEqual(table.rows, expect) {
|
|
t.Fatal("table rows is not equals to the input")
|
|
}
|
|
}
|
|
|
|
func deepEqual(a, b []Row) bool {
|
|
if len(a) != len(b) {
|
|
return false
|
|
}
|
|
for i, r := range a {
|
|
for j, f := range r {
|
|
if f != b[i][j] {
|
|
return false
|
|
}
|
|
}
|
|
}
|
|
return true
|
|
}
|
|
|
|
// func TestWithColumns(t *testing.T) {
|
|
// t.Run("With columns")
|
|
//
|
|
// t.Run("set headers directly")
|
|
// }
|
|
|
|
func TestTableAlignment(t *testing.T) {
|
|
t.Run("No border", func(t *testing.T) {
|
|
s := DefaultStyles()
|
|
s.BorderHeader = false
|
|
biscuits := New(
|
|
WithHeight(5),
|
|
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"},
|
|
}),
|
|
WithStyles(s),
|
|
)
|
|
got := ansi.Strip(biscuits.View())
|
|
golden.RequireEqual(t, []byte(got))
|
|
})
|
|
t.Run("With border", func(t *testing.T) {
|
|
s := DefaultStyles()
|
|
s.Border = lipgloss.NormalBorder()
|
|
s.BorderStyle = lipgloss.NewStyle().BorderForeground(lipgloss.Color("240"))
|
|
|
|
biscuits := New(
|
|
WithHeight(5),
|
|
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"},
|
|
}),
|
|
WithStyles(s),
|
|
)
|
|
got := ansi.Strip(biscuits.View())
|
|
golden.RequireEqual(t, []byte(got))
|
|
})
|
|
}
|