Rainbond/util/termtables/cell_test.go
2020-09-06 11:11:11 +08:00

114 lines
2.6 KiB
Go

// Copyright 2012-2015 Apcera Inc. All rights reserved.
package termtables
import (
"testing"
)
func TestCellRenderString(t *testing.T) {
style := &renderStyle{TableStyle: TableStyle{}, cellWidths: map[int]int{}}
cell := createCell(0, "foobar", nil)
output := cell.Render(style)
if output != "foobar" {
t.Fatal("Unexpected output:", output)
}
}
func TestCellRenderBool(t *testing.T) {
style := &renderStyle{TableStyle: TableStyle{}, cellWidths: map[int]int{}}
cell := createCell(0, true, nil)
output := cell.Render(style)
if output != "true" {
t.Fatal("Unexpected output:", output)
}
}
func TestCellRenderInteger(t *testing.T) {
style := &renderStyle{TableStyle: TableStyle{}, cellWidths: map[int]int{}}
cell := createCell(0, 12345, nil)
output := cell.Render(style)
if output != "12345" {
t.Fatal("Unexpected output:", output)
}
}
func TestCellRenderFloat(t *testing.T) {
style := &renderStyle{TableStyle: TableStyle{}, cellWidths: map[int]int{}}
cell := createCell(0, 12.345, nil)
output := cell.Render(style)
if output != "12.35" {
t.Fatal("Unexpected output:", output)
}
}
func TestCellRenderPadding(t *testing.T) {
style := &renderStyle{TableStyle: TableStyle{PaddingLeft: 3, PaddingRight: 4}, cellWidths: map[int]int{}}
cell := createCell(0, "foobar", nil)
output := cell.Render(style)
if output != " foobar " {
t.Fatal("Unexpected output:", output)
}
}
type foo struct {
v string
}
func (f *foo) String() string {
return f.v
}
func TestCellRenderStringerStruct(t *testing.T) {
style := &renderStyle{TableStyle: TableStyle{}, cellWidths: map[int]int{}}
cell := createCell(0, &foo{v: "bar"}, nil)
output := cell.Render(style)
if output != "bar" {
t.Fatal("Unexpected output:", output)
}
}
type fooString string
func TestCellRenderGeneric(t *testing.T) {
style := &renderStyle{TableStyle: TableStyle{}, cellWidths: map[int]int{}}
cell := createCell(0, fooString("baz"), nil)
output := cell.Render(style)
if output != "baz" {
t.Fatal("Unexpected output:", output)
}
}
func TestFilterColorCodes(t *testing.T) {
tests := []struct {
in string
out string
}{
{"abc", "abc"},
{"", ""},
{"\033[31m\033[0m", ""},
{"a\033[31mb\033[0mc", "abc"},
{"\033[31mabc\033[0m", "abc"},
{"\033[31mfoo\033[0mbar", "foobar"},
{"\033[31mfoo\033[mbar", "foobar"},
{"\033[31mfoo\033[0;0mbar", "foobar"},
{"\033[31;4mfoo\033[0mbar", "foobar"},
{"\033[31;4;43mfoo\033[0mbar", "foobar"},
}
for _, test := range tests {
got := filterColorCodes(test.in)
if got != test.out {
t.Errorf("Invalid color-code filter result; expected %q but got %q from input %q",
test.out, got, test.in)
}
}
}