dualfroz
d75166c01a
the width cache key does not include the tab width, so after SetTabWidth, Width returned stale results for strings containing tabs. purge the cache like SetForceNarrow does, and add a test comparing Width with WidthNoCache after changing the tab width.
54 lines
1.3 KiB
Go
54 lines
1.3 KiB
Go
package twwidth_test
|
|
|
|
import (
|
|
"fmt"
|
|
"testing"
|
|
|
|
"github.com/olekukonko/tablewriter/pkg/twwidth"
|
|
)
|
|
|
|
func TestTabinalSetWidth(t *testing.T) {
|
|
t.Setenv("TABWIDTH", "3")
|
|
|
|
for _, initialized := range []bool{false, true} {
|
|
for _, width := range []int{1, 2, 32, 0, -1, 33} {
|
|
t.Run(fmt.Sprintf("initialized=%v/width=%d", initialized, width), func(t *testing.T) {
|
|
var tab twwidth.Tabinal
|
|
if initialized {
|
|
if got := tab.Size(); got != 3 {
|
|
t.Fatalf("initial Size() = %d, want 3", got)
|
|
}
|
|
}
|
|
|
|
tab.SetWidth(width)
|
|
want := width
|
|
if width <= 0 || width > 32 {
|
|
want = 3
|
|
}
|
|
if got := tab.Size(); got != want {
|
|
t.Fatalf("Size() after SetWidth(%d) = %d, want %d", width, got, want)
|
|
}
|
|
|
|
tab.SetWidth(5)
|
|
tab.SetWidth(0)
|
|
if got := tab.Size(); got != 5 {
|
|
t.Fatalf("Size() after changing width = %d, want 5", got)
|
|
}
|
|
})
|
|
}
|
|
}
|
|
}
|
|
|
|
func TestSetTabWidthPurgesWidthCache(t *testing.T) {
|
|
orig := twwidth.TabWidth()
|
|
t.Cleanup(func() { twwidth.SetTabWidth(orig) })
|
|
|
|
const s = "a\tb"
|
|
for _, width := range []int{2, 4, 6} {
|
|
twwidth.SetTabWidth(width)
|
|
if got, want := twwidth.Width(s), twwidth.WidthNoCache(s); got != want {
|
|
t.Errorf("Width(%q) after SetTabWidth(%d) = %d, want %d", s, width, got, want)
|
|
}
|
|
}
|
|
}
|