76 lines
2.5 KiB
Go
76 lines
2.5 KiB
Go
|
//
|
||
|
|
// config_test.go
|
||
|
|
// Copyright (C) 2023 tiglog <me@tiglog.com>
|
||
|
|
//
|
||
|
|
// Distributed under terms of the MIT license.
|
||
|
|
//
|
||
|
|
|
||
|
|
package table
|
||
|
|
|
||
|
|
import (
|
||
|
|
"testing"
|
||
|
|
|
||
|
|
"git.hexq.cn/tiglog/golib/helper/text"
|
||
|
|
"github.com/stretchr/testify/assert"
|
||
|
|
)
|
||
|
|
|
||
|
|
func TestColumnConfig_getWidthMaxEnforcer(t *testing.T) {
|
||
|
|
t.Run("no width enforcer", func(t *testing.T) {
|
||
|
|
cc := ColumnConfig{}
|
||
|
|
|
||
|
|
widthEnforcer := cc.getWidthMaxEnforcer()
|
||
|
|
assert.Equal(t, "1234567890", widthEnforcer("1234567890", 0))
|
||
|
|
assert.Equal(t, "1234567890", widthEnforcer("1234567890", 1))
|
||
|
|
assert.Equal(t, "1234567890", widthEnforcer("1234567890", 5))
|
||
|
|
assert.Equal(t, "1234567890", widthEnforcer("1234567890", 10))
|
||
|
|
assert.Equal(t, "1234567890", widthEnforcer("1234567890", 100))
|
||
|
|
assert.Equal(t, "1234567890", widthEnforcer("1234567890", 1000))
|
||
|
|
})
|
||
|
|
|
||
|
|
t.Run("default width enforcer", func(t *testing.T) {
|
||
|
|
cc := ColumnConfig{
|
||
|
|
WidthMax: 10,
|
||
|
|
}
|
||
|
|
|
||
|
|
widthEnforcer := cc.getWidthMaxEnforcer()
|
||
|
|
assert.Equal(t, "", widthEnforcer("1234567890", 0))
|
||
|
|
assert.Equal(t, "1\n2\n3\n4\n5\n6\n7\n8\n9\n0", widthEnforcer("1234567890", 1))
|
||
|
|
assert.Equal(t, "12345\n67890", widthEnforcer("1234567890", 5))
|
||
|
|
assert.Equal(t, "1234567890", widthEnforcer("1234567890", 10))
|
||
|
|
assert.Equal(t, "1234567890", widthEnforcer("1234567890", 100))
|
||
|
|
assert.Equal(t, "1234567890", widthEnforcer("1234567890", 1000))
|
||
|
|
})
|
||
|
|
|
||
|
|
t.Run("custom width enforcer (1)", func(t *testing.T) {
|
||
|
|
cc := ColumnConfig{
|
||
|
|
WidthMax: 10,
|
||
|
|
WidthMaxEnforcer: text.Trim,
|
||
|
|
}
|
||
|
|
|
||
|
|
widthEnforcer := cc.getWidthMaxEnforcer()
|
||
|
|
assert.Equal(t, text.Trim("1234567890", 0), widthEnforcer("1234567890", 0))
|
||
|
|
assert.Equal(t, text.Trim("1234567890", 1), widthEnforcer("1234567890", 1))
|
||
|
|
assert.Equal(t, text.Trim("1234567890", 5), widthEnforcer("1234567890", 5))
|
||
|
|
assert.Equal(t, text.Trim("1234567890", 10), widthEnforcer("1234567890", 10))
|
||
|
|
assert.Equal(t, text.Trim("1234567890", 100), widthEnforcer("1234567890", 100))
|
||
|
|
assert.Equal(t, text.Trim("1234567890", 1000), widthEnforcer("1234567890", 1000))
|
||
|
|
})
|
||
|
|
|
||
|
|
t.Run("custom width enforcer (2)", func(t *testing.T) {
|
||
|
|
cc := ColumnConfig{
|
||
|
|
WidthMax: 10,
|
||
|
|
WidthMaxEnforcer: func(col string, maxLen int) string {
|
||
|
|
return "foo"
|
||
|
|
},
|
||
|
|
}
|
||
|
|
|
||
|
|
widthEnforcer := cc.getWidthMaxEnforcer()
|
||
|
|
assert.Equal(t, "foo", widthEnforcer("1234567890", 0))
|
||
|
|
assert.Equal(t, "foo", widthEnforcer("1234567890", 1))
|
||
|
|
assert.Equal(t, "foo", widthEnforcer("1234567890", 5))
|
||
|
|
assert.Equal(t, "foo", widthEnforcer("1234567890", 10))
|
||
|
|
assert.Equal(t, "foo", widthEnforcer("1234567890", 100))
|
||
|
|
assert.Equal(t, "foo", widthEnforcer("1234567890", 1000))
|
||
|
|
})
|
||
|
|
}
|