adds test cast for olekukonko#214

eryx · Apr 15, 2023 · c41d317 · c41d317
1 parent 8532778
commit c41d317
Showing 1 changed file with 26 additions and 0 deletions.
diff --git a/table_test.go b/table_test.go
@@ -1016,6 +1016,32 @@ func TestMoreFooterColumnsThanHeaders(t *testing.T) {
 	checkEqual(t, buf.String(), want)
 }
 
+func TestLessFooterColumnsThanHeaders(t *testing.T) {
+	var (
+		buf    = &bytes.Buffer{}
+		table  = NewWriter(buf)
+		header = []string{"A", "B", "C"}
+		data   = [][]string{
+			{"1", "2", "3"},
+		}
+		footer = []string{"a", "b"}
+		want   = `+---+---+---+
+| A | B | C |
++---+---+---+
+| 1 | 2 | 3 |
++---+---+---+
+| A | B |   |
++---+---+---+
+`
+	)
+	table.SetHeader(header)
+	table.SetFooter(footer)
+	table.AppendBulk(data)
+	table.Render()
+
+	checkEqual(t, buf.String(), want)
+}
+
 func TestSetColMinWidth(t *testing.T) {
 	var (
 		buf    = &bytes.Buffer{}