895 lines
27 KiB
Go
895 lines
27 KiB
Go
// 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 |
|
|
// *============*===========*=========*
|
|
}
|
|
|
|
// TestNumLines to test the numbers of lines
|
|
func TestNumLines(t *testing.T) {
|
|
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"},
|
|
}
|
|
|
|
buf := &bytes.Buffer{}
|
|
table := NewWriter(buf)
|
|
table.SetHeader([]string{"Name", "Sign", "Rating"})
|
|
|
|
for i, v := range data {
|
|
table.Append(v)
|
|
if i+1 != table.NumLines() {
|
|
t.Errorf("Number of lines failed\ngot:\n[%d]\nwant:\n[%d]\n", table.NumLines(), i+1)
|
|
}
|
|
}
|
|
|
|
if len(data) != table.NumLines() {
|
|
t.Errorf("Number of lines failed\ngot:\n[%d]\nwant:\n[%d]\n", table.NumLines(), len(data))
|
|
}
|
|
}
|
|
|
|
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)
|
|
}
|
|
}
|