mirror of
https://github.com/taigrr/wtf
synced 2025-01-18 04:03:14 -08:00
45 lines
900 B
Go
45 lines
900 B
Go
package view
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
)
|
|
|
|
func makeMap() map[string]string {
|
|
m := make(map[string]string)
|
|
m["foo"] = "val1"
|
|
m["bar"] = "val2"
|
|
m["baz"] = "val3"
|
|
|
|
return m
|
|
}
|
|
|
|
func newTestTable(height int) *InfoTable {
|
|
var headers [2]string
|
|
|
|
headers[0] = "hdr0"
|
|
headers[1] = "hdr1"
|
|
|
|
table := NewInfoTable(
|
|
headers[:],
|
|
makeMap(),
|
|
1,
|
|
1,
|
|
height,
|
|
)
|
|
return table
|
|
}
|
|
|
|
func Test_RenderSimpleInfoTable(t *testing.T) {
|
|
table := newTestTable(4).Render()
|
|
|
|
assert.Equal(t, " ----- ------ \n HDR0 HDR1 \n ----- ------ \n bar val2 \n baz val3 \n foo val1 \n ----- ------ \n", table)
|
|
}
|
|
|
|
func Test_RenderPaddedInfoTable(t *testing.T) {
|
|
table := newTestTable(6).Render()
|
|
|
|
assert.Equal(t, " ----- ------ \n HDR0 HDR1 \n ----- ------ \n bar val2 \n baz val3 \n foo val1 \n \n \n ----- ------ \n", table)
|
|
}
|