Fix panic with footer columns less than header columns

Adopted from olekukonko#228
alcionai · Oct 9, 2024 · f7b26e0 · f7b26e0
1 parent df64c4b
commit f7b26e0
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/table.go b/table.go
@@ -692,7 +692,7 @@ func (t *Table) printFooter() {
 	max := t.rs[footerRowIdx]
 
 	// Print Footer
-	for i := 0; i < (len(t.cs) - len(t.footers)); i++ {
+	for len(t.cs)-len(t.footers) > 0 {
 		lines := t.parseDimension(" ", len(t.footers), footerRowIdx)
 		t.footers = append(t.footers, lines)
 	}

diff --git a/table_test.go b/table_test.go
@@ -1309,13 +1309,13 @@ func TestLessFooterColumnsThanHeaders(t *testing.T) {
 		data   = [][]string{
 			{"1", "2", "3"},
 		}
-		footer = []string{"a", "b"}
+		footer = []string{"a"}
 		want   = `+---+---+---+
 | A | B | C |
 +---+---+---+
 | 1 | 2 | 3 |
 +---+---+---+
-| A | B |   |
+| A |   |   |
 +---+---+---+
 `
 	)