route/vendor/github.com/olekukonko/tablewriter/table_test.go

870 lines
26 KiB
Go
Raw Normal View History

2017-10-06 15:29:20 +00:00
// Copyright 2014 Oleku Konko All rights reserved.
// Use of this source code is governed by a MIT
// license that can be found in the LICENSE file.
// This module is a Table Writer API for the Go Programming Language.
// The protocols were written in pure Go and works on windows and unix systems
package tablewriter
import (
"bytes"
"fmt"
"io"
"os"
"reflect"
"strings"
"testing"
)
func ExampleShort() {
data := [][]string{
[]string{"A", "The Good", "500"},
[]string{"B", "The Very very Bad Man", "288"},
[]string{"C", "The Ugly", "120"},
[]string{"D", "The Gopher", "800"},
}
table := NewWriter(os.Stdout)
table.SetHeader([]string{"Name", "Sign", "Rating"})
for _, v := range data {
table.Append(v)
}
table.Render()
// Output: +------+-----------------------+--------+
// | NAME | SIGN | RATING |
// +------+-----------------------+--------+
// | A | The Good | 500 |
// | B | The Very very Bad Man | 288 |
// | C | The Ugly | 120 |
// | D | The Gopher | 800 |
// +------+-----------------------+--------+
}
func ExampleLong() {
data := [][]string{
[]string{"Learn East has computers with adapted keyboards with enlarged print etc", " Some Data ", " Another Data"},
[]string{"Instead of lining up the letters all ", "the way across, he splits the keyboard in two", "Like most ergonomic keyboards", "See Data"},
}
table := NewWriter(os.Stdout)
table.SetHeader([]string{"Name", "Sign", "Rating"})
table.SetCenterSeparator("*")
table.SetRowSeparator("=")
for _, v := range data {
table.Append(v)
}
table.Render()
}
func ExampleCSV() {
table, _ := NewCSV(os.Stdout, "test.csv", true)
table.SetCenterSeparator("*")
table.SetRowSeparator("=")
table.Render()
// Output: *============*===========*=========*
// | FIRST NAME | LAST NAME | SSN |
// *============*===========*=========*
// | John | Barry | 123456 |
// | Kathy | Smith | 687987 |
// | Bob | McCornick | 3979870 |
// *============*===========*=========*
}
func TestCSVInfo(t *testing.T) {
buf := &bytes.Buffer{}
table, err := NewCSV(buf, "test_info.csv", true)
if err != nil {
t.Error(err)
return
}
table.SetAlignment(ALIGN_LEFT)
table.SetBorder(false)
table.Render()
got := buf.String()
want := ` FIELD | TYPE | NULL | KEY | DEFAULT | EXTRA
+----------+--------------+------+-----+---------+----------------+
user_id | smallint(5) | NO | PRI | NULL | auto_increment
username | varchar(10) | NO | | NULL |
password | varchar(100) | NO | | NULL |
`
if got != want {
t.Errorf("CSV info failed\ngot:\n[%s]\nwant:\n[%s]\n", got, want)
}
}
func TestCSVSeparator(t *testing.T) {
buf := &bytes.Buffer{}
table, err := NewCSV(buf, "test.csv", true)
if err != nil {
t.Error(err)
return
}
table.SetRowLine(true)
table.SetCenterSeparator("+")
table.SetColumnSeparator("|")
table.SetRowSeparator("-")
table.SetAlignment(ALIGN_LEFT)
table.Render()
want := `+------------+-----------+---------+
| FIRST NAME | LAST NAME | SSN |
+------------+-----------+---------+
| John | Barry | 123456 |
+------------+-----------+---------+
| Kathy | Smith | 687987 |
+------------+-----------+---------+
| Bob | McCornick | 3979870 |
+------------+-----------+---------+
`
got := buf.String()
if got != want {
t.Errorf("CSV info failed\ngot:\n[%s]\nwant:\n[%s]\n", got, want)
}
}
func TestNoBorder(t *testing.T) {
data := [][]string{
[]string{"1/1/2014", "Domain name", "2233", "$10.98"},
[]string{"1/1/2014", "January Hosting", "2233", "$54.95"},
[]string{"", " (empty)\n (empty)", "", ""},
[]string{"1/4/2014", "February Hosting", "2233", "$51.00"},
[]string{"1/4/2014", "February Extra Bandwidth", "2233", "$30.00"},
[]string{"1/4/2014", " (Discount)", "2233", "-$1.00"},
}
var buf bytes.Buffer
table := NewWriter(&buf)
table.SetAutoWrapText(false)
table.SetHeader([]string{"Date", "Description", "CV2", "Amount"})
table.SetFooter([]string{"", "", "Total", "$145.93"}) // Add Footer
table.SetBorder(false) // Set Border to false
table.AppendBulk(data) // Add Bulk Data
table.Render()
want := ` DATE | DESCRIPTION | CV2 | AMOUNT
+----------+--------------------------+-------+---------+
1/1/2014 | Domain name | 2233 | $10.98
1/1/2014 | January Hosting | 2233 | $54.95
| (empty) | |
| (empty) | |
1/4/2014 | February Hosting | 2233 | $51.00
1/4/2014 | February Extra Bandwidth | 2233 | $30.00
1/4/2014 | (Discount) | 2233 | -$1.00
+----------+--------------------------+-------+---------+
TOTAL | $145 93
+-------+---------+
`
got := buf.String()
if got != want {
t.Errorf("border table rendering failed\ngot:\n%s\nwant:\n%s\n", got, want)
}
}
func TestWithBorder(t *testing.T) {
data := [][]string{
[]string{"1/1/2014", "Domain name", "2233", "$10.98"},
[]string{"1/1/2014", "January Hosting", "2233", "$54.95"},
[]string{"", " (empty)\n (empty)", "", ""},
[]string{"1/4/2014", "February Hosting", "2233", "$51.00"},
[]string{"1/4/2014", "February Extra Bandwidth", "2233", "$30.00"},
[]string{"1/4/2014", " (Discount)", "2233", "-$1.00"},
}
var buf bytes.Buffer
table := NewWriter(&buf)
table.SetAutoWrapText(false)
table.SetHeader([]string{"Date", "Description", "CV2", "Amount"})
table.SetFooter([]string{"", "", "Total", "$145.93"}) // Add Footer
table.AppendBulk(data) // Add Bulk Data
table.Render()
want := `+----------+--------------------------+-------+---------+
| DATE | DESCRIPTION | CV2 | AMOUNT |
+----------+--------------------------+-------+---------+
| 1/1/2014 | Domain name | 2233 | $10.98 |
| 1/1/2014 | January Hosting | 2233 | $54.95 |
| | (empty) | | |
| | (empty) | | |
| 1/4/2014 | February Hosting | 2233 | $51.00 |
| 1/4/2014 | February Extra Bandwidth | 2233 | $30.00 |
| 1/4/2014 | (Discount) | 2233 | -$1.00 |
+----------+--------------------------+-------+---------+
| TOTAL | $145 93 |
+----------+--------------------------+-------+---------+
`
got := buf.String()
if got != want {
t.Errorf("border table rendering failed\ngot:\n%s\nwant:\n%s\n", got, want)
}
}
func TestPrintingInMarkdown(t *testing.T) {
data := [][]string{
[]string{"1/1/2014", "Domain name", "2233", "$10.98"},
[]string{"1/1/2014", "January Hosting", "2233", "$54.95"},
[]string{"1/4/2014", "February Hosting", "2233", "$51.00"},
[]string{"1/4/2014", "February Extra Bandwidth", "2233", "$30.00"},
}
var buf bytes.Buffer
table := NewWriter(&buf)
table.SetHeader([]string{"Date", "Description", "CV2", "Amount"})
table.AppendBulk(data) // Add Bulk Data
table.SetBorders(Border{Left: true, Top: false, Right: true, Bottom: false})
table.SetCenterSeparator("|")
table.Render()
want := `| DATE | DESCRIPTION | CV2 | AMOUNT |
|----------|--------------------------|------|--------|
| 1/1/2014 | Domain name | 2233 | $10.98 |
| 1/1/2014 | January Hosting | 2233 | $54.95 |
| 1/4/2014 | February Hosting | 2233 | $51.00 |
| 1/4/2014 | February Extra Bandwidth | 2233 | $30.00 |
`
got := buf.String()
if got != want {
t.Errorf("border table rendering failed\ngot:\n%s\nwant:\n%s\n", got, want)
}
}
func TestPrintHeading(t *testing.T) {
var buf bytes.Buffer
table := NewWriter(&buf)
table.SetHeader([]string{"1", "2", "3", "4", "5", "6", "7", "8", "9", "a", "b", "c"})
table.printHeading()
want := `| 1 | 2 | 3 | 4 | 5 | 6 | 7 | 8 | 9 | A | B | C |
+---+---+---+---+---+---+---+---+---+---+---+---+
`
got := buf.String()
if got != want {
t.Errorf("header rendering failed\ngot:\n%s\nwant:\n%s\n", got, want)
}
}
func TestPrintHeadingWithoutAutoFormat(t *testing.T) {
var buf bytes.Buffer
table := NewWriter(&buf)
table.SetHeader([]string{"1", "2", "3", "4", "5", "6", "7", "8", "9", "a", "b", "c"})
table.SetAutoFormatHeaders(false)
table.printHeading()
want := `| 1 | 2 | 3 | 4 | 5 | 6 | 7 | 8 | 9 | a | b | c |
+---+---+---+---+---+---+---+---+---+---+---+---+
`
got := buf.String()
if got != want {
t.Errorf("header rendering failed\ngot:\n%s\nwant:\n%s\n", got, want)
}
}
func TestPrintFooter(t *testing.T) {
var buf bytes.Buffer
table := NewWriter(&buf)
table.SetHeader([]string{"1", "2", "3", "4", "5", "6", "7", "8", "9", "a", "b", "c"})
table.SetFooter([]string{"1", "2", "3", "4", "5", "6", "7", "8", "9", "a", "b", "c"})
table.printFooter()
want := `| 1 | 2 | 3 | 4 | 5 | 6 | 7 | 8 | 9 | A | B | C |
+---+---+---+---+---+---+---+---+---+---+---+---+
`
got := buf.String()
if got != want {
t.Errorf("footer rendering failed\ngot:\n%s\nwant:\n%s\n", got, want)
}
}
func TestPrintFooterWithoutAutoFormat(t *testing.T) {
var buf bytes.Buffer
table := NewWriter(&buf)
table.SetAutoFormatHeaders(false)
table.SetHeader([]string{"1", "2", "3", "4", "5", "6", "7", "8", "9", "a", "b", "c"})
table.SetFooter([]string{"1", "2", "3", "4", "5", "6", "7", "8", "9", "a", "b", "c"})
table.printFooter()
want := `| 1 | 2 | 3 | 4 | 5 | 6 | 7 | 8 | 9 | a | b | c |
+---+---+---+---+---+---+---+---+---+---+---+---+
`
got := buf.String()
if got != want {
t.Errorf("footer rendering failed\ngot:\n%s\nwant:\n%s\n", got, want)
}
}
func TestPrintShortCaption(t *testing.T) {
var buf bytes.Buffer
data := [][]string{
[]string{"A", "The Good", "500"},
[]string{"B", "The Very very Bad Man", "288"},
[]string{"C", "The Ugly", "120"},
[]string{"D", "The Gopher", "800"},
}
table := NewWriter(&buf)
table.SetHeader([]string{"Name", "Sign", "Rating"})
table.SetCaption(true, "Short caption.")
for _, v := range data {
table.Append(v)
}
table.Render()
want := `+------+-----------------------+--------+
| NAME | SIGN | RATING |
+------+-----------------------+--------+
| A | The Good | 500 |
| B | The Very very Bad Man | 288 |
| C | The Ugly | 120 |
| D | The Gopher | 800 |
+------+-----------------------+--------+
Short caption.
`
got := buf.String()
if got != want {
t.Errorf("long caption for short example rendering failed\ngot:\n%s\nwant:\n%s\n", got, want)
}
}
func TestPrintLongCaptionWithShortExample(t *testing.T) {
var buf bytes.Buffer
data := [][]string{
[]string{"A", "The Good", "500"},
[]string{"B", "The Very very Bad Man", "288"},
[]string{"C", "The Ugly", "120"},
[]string{"D", "The Gopher", "800"},
}
table := NewWriter(&buf)
table.SetHeader([]string{"Name", "Sign", "Rating"})
table.SetCaption(true, "This is a very long caption. The text should wrap. If not, we have a problem that needs to be solved.")
for _, v := range data {
table.Append(v)
}
table.Render()
want := `+------+-----------------------+--------+
| NAME | SIGN | RATING |
+------+-----------------------+--------+
| A | The Good | 500 |
| B | The Very very Bad Man | 288 |
| C | The Ugly | 120 |
| D | The Gopher | 800 |
+------+-----------------------+--------+
This is a very long caption. The text
should wrap. If not, we have a problem
that needs to be solved.
`
got := buf.String()
if got != want {
t.Errorf("long caption for short example rendering failed\ngot:\n%s\nwant:\n%s\n", got, want)
}
}
func TestPrintCaptionWithFooter(t *testing.T) {
data := [][]string{
[]string{"1/1/2014", "Domain name", "2233", "$10.98"},
[]string{"1/1/2014", "January Hosting", "2233", "$54.95"},
[]string{"1/4/2014", "February Hosting", "2233", "$51.00"},
[]string{"1/4/2014", "February Extra Bandwidth", "2233", "$30.00"},
}
var buf bytes.Buffer
table := NewWriter(&buf)
table.SetHeader([]string{"Date", "Description", "CV2", "Amount"})
table.SetFooter([]string{"", "", "Total", "$146.93"}) // Add Footer
table.SetCaption(true, "This is a very long caption. The text should wrap to the width of the table.") // Add caption
table.SetBorder(false) // Set Border to false
table.AppendBulk(data) // Add Bulk Data
table.Render()
want := ` DATE | DESCRIPTION | CV2 | AMOUNT
+----------+--------------------------+-------+---------+
1/1/2014 | Domain name | 2233 | $10.98
1/1/2014 | January Hosting | 2233 | $54.95
1/4/2014 | February Hosting | 2233 | $51.00
1/4/2014 | February Extra Bandwidth | 2233 | $30.00
+----------+--------------------------+-------+---------+
TOTAL | $146 93
+-------+---------+
This is a very long caption. The text should wrap to the
width of the table.
`
got := buf.String()
if got != want {
t.Errorf("border table rendering failed\ngot:\n%s\nwant:\n%s\n", got, want)
}
}
func TestPrintLongCaptionWithLongExample(t *testing.T) {
var buf bytes.Buffer
data := [][]string{
[]string{"Learn East has computers with adapted keyboards with enlarged print etc", "Some Data", "Another Data"},
[]string{"Instead of lining up the letters all", "the way across, he splits the keyboard in two", "Like most ergonomic keyboards"},
}
table := NewWriter(&buf)
table.SetCaption(true, "This is a very long caption. The text should wrap. If not, we have a problem that needs to be solved.")
table.SetHeader([]string{"Name", "Sign", "Rating"})
for _, v := range data {
table.Append(v)
}
table.Render()
want := `+--------------------------------+--------------------------------+-------------------------------+
| NAME | SIGN | RATING |
+--------------------------------+--------------------------------+-------------------------------+
| Learn East has computers | Some Data | Another Data |
| with adapted keyboards with | | |
| enlarged print etc | | |
| Instead of lining up the | the way across, he splits the | Like most ergonomic keyboards |
| letters all | keyboard in two | |
+--------------------------------+--------------------------------+-------------------------------+
This is a very long caption. The text should wrap. If not, we have a problem that needs to be
solved.
`
got := buf.String()
if got != want {
t.Errorf("long caption for long example rendering failed\ngot:\n%s\nwant:\n%s\n", got, want)
}
}
func TestPrintTableWithAndWithoutAutoWrap(t *testing.T) {
var buf bytes.Buffer
var multiline = `A multiline
string with some lines being really long.`
with := NewWriter(&buf)
with.Append([]string{multiline})
with.Render()
want := `+--------------------------------+
| A multiline string with some |
| lines being really long. |
+--------------------------------+
`
got := buf.String()
if got != want {
t.Errorf("multiline text rendering with wrapping failed\ngot:\n%s\nwant:\n%s\n", got, want)
}
buf.Truncate(0)
without := NewWriter(&buf)
without.SetAutoWrapText(false)
without.Append([]string{multiline})
without.Render()
want = `+-------------------------------------------+
| A multiline |
| string with some lines being really long. |
+-------------------------------------------+
`
got = buf.String()
if got != want {
t.Errorf("multiline text rendering without wrapping rendering failed\ngot:\n%s\nwant:\n%s\n", got, want)
}
}
func TestPrintLine(t *testing.T) {
header := make([]string, 12)
val := " "
want := ""
for i := range header {
header[i] = val
want = fmt.Sprintf("%s+-%s-", want, strings.Replace(val, " ", "-", -1))
val = val + " "
}
want = want + "+"
var buf bytes.Buffer
table := NewWriter(&buf)
table.SetHeader(header)
table.printLine(false)
got := buf.String()
if got != want {
t.Errorf("line rendering failed\ngot:\n%s\nwant:\n%s\n", got, want)
}
}
func TestAnsiStrip(t *testing.T) {
header := make([]string, 12)
val := " "
want := ""
for i := range header {
header[i] = "\033[43;30m" + val + "\033[00m"
want = fmt.Sprintf("%s+-%s-", want, strings.Replace(val, " ", "-", -1))
val = val + " "
}
want = want + "+"
var buf bytes.Buffer
table := NewWriter(&buf)
table.SetHeader(header)
table.printLine(false)
got := buf.String()
if got != want {
t.Errorf("line rendering failed\ngot:\n%s\nwant:\n%s\n", got, want)
}
}
func NewCustomizedTable(out io.Writer) *Table {
table := NewWriter(out)
table.SetCenterSeparator("")
table.SetColumnSeparator("")
table.SetRowSeparator("")
table.SetBorder(false)
table.SetAlignment(ALIGN_LEFT)
table.SetHeader([]string{})
return table
}
func TestSubclass(t *testing.T) {
buf := new(bytes.Buffer)
table := NewCustomizedTable(buf)
data := [][]string{
[]string{"A", "The Good", "500"},
[]string{"B", "The Very very Bad Man", "288"},
[]string{"C", "The Ugly", "120"},
[]string{"D", "The Gopher", "800"},
}
for _, v := range data {
table.Append(v)
}
table.Render()
output := string(buf.Bytes())
want := ` A The Good 500
B The Very very Bad Man 288
C The Ugly 120
D The Gopher 800
`
if output != want {
t.Error(fmt.Sprintf("Unexpected output '%v' != '%v'", output, want))
}
}
func TestAutoMergeRows(t *testing.T) {
data := [][]string{
[]string{"A", "The Good", "500"},
[]string{"A", "The Very very Bad Man", "288"},
[]string{"B", "The Very very Bad Man", "120"},
[]string{"B", "The Very very Bad Man", "200"},
}
var buf bytes.Buffer
table := NewWriter(&buf)
table.SetHeader([]string{"Name", "Sign", "Rating"})
for _, v := range data {
table.Append(v)
}
table.SetAutoMergeCells(true)
table.Render()
want := `+------+-----------------------+--------+
| NAME | SIGN | RATING |
+------+-----------------------+--------+
| A | The Good | 500 |
| | The Very very Bad Man | 288 |
| B | | 120 |
| | | 200 |
+------+-----------------------+--------+
`
got := buf.String()
if got != want {
t.Errorf("\ngot:\n%s\nwant:\n%s\n", got, want)
}
buf.Reset()
table = NewWriter(&buf)
table.SetHeader([]string{"Name", "Sign", "Rating"})
for _, v := range data {
table.Append(v)
}
table.SetAutoMergeCells(true)
table.SetRowLine(true)
table.Render()
want = `+------+-----------------------+--------+
| NAME | SIGN | RATING |
+------+-----------------------+--------+
| A | The Good | 500 |
+ +-----------------------+--------+
| | The Very very Bad Man | 288 |
+------+ +--------+
| B | | 120 |
+ + +--------+
| | | 200 |
+------+-----------------------+--------+
`
got = buf.String()
if got != want {
t.Errorf("\ngot:\n%s\nwant:\n%s\n", got, want)
}
buf.Reset()
table = NewWriter(&buf)
table.SetHeader([]string{"Name", "Sign", "Rating"})
dataWithlongText := [][]string{
[]string{"A", "The Good", "500"},
[]string{"A", "The Very very very very very Bad Man", "288"},
[]string{"B", "The Very very very very very Bad Man", "120"},
[]string{"C", "The Very very Bad Man", "200"},
}
table.AppendBulk(dataWithlongText)
table.SetAutoMergeCells(true)
table.SetRowLine(true)
table.Render()
want = `+------+--------------------------------+--------+
| NAME | SIGN | RATING |
+------+--------------------------------+--------+
| A | The Good | 500 |
+------+--------------------------------+--------+
| A | The Very very very very very | 288 |
| | Bad Man | |
+------+ +--------+
| B | | 120 |
| | | |
+------+--------------------------------+--------+
| C | The Very very Bad Man | 200 |
+------+--------------------------------+--------+
`
got = buf.String()
if got != want {
t.Errorf("\ngot:\n%s\nwant:\n%s\n", got, want)
}
}
func TestClearRows(t *testing.T) {
data := [][]string{
[]string{"1/1/2014", "Domain name", "2233", "$10.98"},
}
var buf bytes.Buffer
table := NewWriter(&buf)
table.SetAutoWrapText(false)
table.SetHeader([]string{"Date", "Description", "CV2", "Amount"})
table.SetFooter([]string{"", "", "Total", "$145.93"}) // Add Footer
table.AppendBulk(data) // Add Bulk Data
table.Render()
originalWant := `+----------+-------------+-------+---------+
| DATE | DESCRIPTION | CV2 | AMOUNT |
+----------+-------------+-------+---------+
| 1/1/2014 | Domain name | 2233 | $10.98 |
+----------+-------------+-------+---------+
| TOTAL | $145 93 |
+----------+-------------+-------+---------+
`
want := originalWant
got := buf.String()
if got != want {
t.Errorf("table clear rows failed\ngot:\n%s\nwant:\n%s\n", got, want)
}
buf.Reset()
table.ClearRows()
table.Render()
want = `+----------+-------------+-------+---------+
| DATE | DESCRIPTION | CV2 | AMOUNT |
+----------+-------------+-------+---------+
+----------+-------------+-------+---------+
| TOTAL | $145 93 |
+----------+-------------+-------+---------+
`
got = buf.String()
if got != want {
t.Errorf("table clear failed\ngot:\n%s\nwant:\n%s\n", got, want)
}
buf.Reset()
table.AppendBulk(data) // Add Bulk Data
table.Render()
want = `+----------+-------------+-------+---------+
| DATE | DESCRIPTION | CV2 | AMOUNT |
+----------+-------------+-------+---------+
| 1/1/2014 | Domain name | 2233 | $10.98 |
+----------+-------------+-------+---------+
| TOTAL | $145 93 |
+----------+-------------+-------+---------+
`
got = buf.String()
if got != want {
t.Errorf("table clear rows failed\ngot:\n%s\nwant:\n%s\n", got, want)
}
}
func TestClearFooters(t *testing.T) {
data := [][]string{
[]string{"1/1/2014", "Domain name", "2233", "$10.98"},
}
var buf bytes.Buffer
table := NewWriter(&buf)
table.SetAutoWrapText(false)
table.SetHeader([]string{"Date", "Description", "CV2", "Amount"})
table.SetFooter([]string{"", "", "Total", "$145.93"}) // Add Footer
table.AppendBulk(data) // Add Bulk Data
table.Render()
buf.Reset()
table.ClearFooter()
table.Render()
want := `+----------+-------------+-------+---------+
| DATE | DESCRIPTION | CV2 | AMOUNT |
+----------+-------------+-------+---------+
| 1/1/2014 | Domain name | 2233 | $10.98 |
+----------+-------------+-------+---------+
`
got := buf.String()
if got != want {
t.Errorf("table clear rows failed\ngot:\n%s\nwant:\n%s\n", got, want)
}
}
func TestMoreDataColumnsThanHeaders(t *testing.T) {
var (
buf = &bytes.Buffer{}
table = NewWriter(buf)
header = []string{"A", "B", "C"}
data = [][]string{
[]string{"a", "b", "c", "d"},
[]string{"1", "2", "3", "4"},
}
want = `+---+---+---+---+
| A | B | C | |
+---+---+---+---+
| a | b | c | d |
| 1 | 2 | 3 | 4 |
+---+---+---+---+
`
)
table.SetHeader(header)
// table.SetFooter(ctx.tableCtx.footer)
table.AppendBulk(data)
table.Render()
got := buf.String()
if got != want {
t.Errorf("\ngot:\n%s\nwant:\n%s\n", got, want)
}
}
func TestMoreFooterColumnsThanHeaders(t *testing.T) {
var (
buf = &bytes.Buffer{}
table = NewWriter(buf)
header = []string{"A", "B", "C"}
data = [][]string{
[]string{"a", "b", "c", "d"},
[]string{"1", "2", "3", "4"},
}
footer = []string{"a", "b", "c", "d", "e"}
want = `+---+---+---+---+---+
| A | B | C | | |
+---+---+---+---+---+
| a | b | c | d |
| 1 | 2 | 3 | 4 |
+---+---+---+---+---+
| A | B | C | D | E |
+---+---+---+---+---+
`
)
table.SetHeader(header)
table.SetFooter(footer)
table.AppendBulk(data)
table.Render()
got := buf.String()
if got != want {
t.Errorf("\ngot:\n%s\nwant:\n%s\n", got, want)
}
}
func TestSetColMinWidth(t *testing.T) {
var (
buf = &bytes.Buffer{}
table = NewWriter(buf)
header = []string{"AAA", "BBB", "CCC"}
data = [][]string{
[]string{"a", "b", "c"},
[]string{"1", "2", "3"},
}
footer = []string{"a", "b", "cccc"}
want = `+-----+-----+-------+
| AAA | BBB | CCC |
+-----+-----+-------+
| a | b | c |
| 1 | 2 | 3 |
+-----+-----+-------+
| A | B | CCCC |
+-----+-----+-------+
`
)
table.SetHeader(header)
table.SetFooter(footer)
table.AppendBulk(data)
table.SetColMinWidth(2, 5)
table.Render()
got := buf.String()
if got != want {
t.Errorf("\ngot:\n%s\nwant:\n%s\n", got, want)
}
}
func TestWrapString(t *testing.T) {
want := []string{"ああああああああああああああああああああああああ", "あああああああ"}
got, _ := WrapString("ああああああああああああああああああああああああ あああああああ", 55)
if !reflect.DeepEqual(got, want) {
t.Errorf("\ngot:\n%v\nwant:\n%v\n", got, want)
}
}
func TestCustomAlign(t *testing.T) {
var (
buf = &bytes.Buffer{}
table = NewWriter(buf)
header = []string{"AAA", "BBB", "CCC"}
data = [][]string{
[]string{"a", "b", "c"},
[]string{"1", "2", "3"},
}
footer = []string{"a", "b", "cccc"}
want = `+-----+-----+-------+
| AAA | BBB | CCC |
+-----+-----+-------+
| a | b | c |
| 1 | 2 | 3 |
+-----+-----+-------+
| A | B | CCCC |
+-----+-----+-------+
`
)
table.SetHeader(header)
table.SetFooter(footer)
table.AppendBulk(data)
table.SetColMinWidth(2, 5)
table.SetColumnAlignment([]int{ALIGN_LEFT, ALIGN_CENTER, ALIGN_RIGHT})
table.Render()
got := buf.String()
if got != want {
t.Errorf("\ngot:\n%s\nwant:\n%s\n", got, want)
}
}