test: add extra test for removing column separators

charmbracelet · Sep 19, 2023 · 830a3e1 · 830a3e1
1 parent 176e89e
commit 830a3e1
Showing 1 changed file with 33 additions and 0 deletions.
diff --git a/table/table_test.go b/table/table_test.go
@@ -250,6 +250,39 @@ func TestTableNoColumnSeparatorsWithHeaders(t *testing.T) {
 	}
 }
 
+func TestBorderColumnsWithExtraRows(t *testing.T) {
+	rows := [][]any{
+		{"Chinese", "Nǐn hǎo", "Nǐ hǎo"},
+		{"French", "Bonjour", "Salut", "Salut"},
+		{"Japanese", "こんにちは", "やあ"},
+		{"Russian", "Zdravstvuyte", "Privet", "Privet", "Privet"},
+		{"Spanish", "Hola", "¿Qué tal?"},
+	}
+
+	table := New().
+		Border(lipgloss.NormalBorder()).
+		BorderColumn(false).
+		StyleFunc(TableStyle).
+		Headers("LANGUAGE", "FORMAL").
+		Rows(rows...)
+
+	expected := strings.TrimSpace(`
+┌───────────────────────────────────────────────────┐
+│ LANGUAGE     FORMAL                               │
+├───────────────────────────────────────────────────┤
+│ Chinese   Nǐn hǎo       Nǐ hǎo                    │
+│ French    Bonjour       Salut      Salut          │
+│ Japanese  こんにちは    やあ                      │
+│ Russian   Zdravstvuyte  Privet     Privet  Privet │
+│ Spanish   Hola          ¿Qué tal?                 │
+└───────────────────────────────────────────────────┘
+`)
+
+	if table.String() != expected {
+		t.Fatalf("expected:\n\n%s\n\ngot:\n\n%s", expected, table.String())
+	}
+}
+
 func TestNew(t *testing.T) {
 	table := New()
 	expected := ""