test: rounded enumerator test

charmbracelet · Apr 18, 2024 · 2889c21 · 2889c21
1 parent 3a498da
commit 2889c21
Showing 1 changed file with 33 additions and 0 deletions.
diff --git a/tree/tree_test.go b/tree/tree_test.go
@@ -42,6 +42,39 @@ func TestTree(t *testing.T) {
 	require.Equal(t, expected, tree.String())
 }
 
+func TestTreeRounded(t *testing.T) {
+	tree := tree.New().
+		Items(
+			"Foo",
+			tree.New().
+				Root("Bar").
+				Items(
+					"Qux",
+					tree.New().
+						Root("Quux").
+						Items(
+							"Foo",
+							"Bar",
+						),
+					"Quuux",
+				),
+			"Baz",
+		).
+		Enumerator(tree.RoundedEnumerator)
+
+	expected := `
+├── Foo
+├── Bar
+│   ├── Qux
+│   ├── Quux
+│   │   ├── Foo
+│   │   ╰── Bar
+│   ╰── Quuux
+╰── Baz
+	`
+	require.Equal(t, expected, tree.String())
+}
+
 func TestTreeHidden(t *testing.T) {
 	tree := tree.New().
 		Items(