test(tree): add TestInheritedStyles for items added to existing Tree

charmbracelet · Jan 10, 2025 · 54aafad · 54aafad
1 parent b0718a4
commit 54aafad
Show file tree

Hide file tree

Showing 2 changed files with 44 additions and 0 deletions.
diff --git a/tree/testdata/TestInheritedStyles.golden b/tree/testdata/TestInheritedStyles.golden
@@ -0,0 +1,12 @@
+[38;5;35m⁜ Makeup[0m
+[38;5;63m├──[0m [38;5;212mGlossier[0m
+[38;5;63m│  [0m [38;5;63m├──[0m [38;5;212mApparel[0m
+[38;5;63m│  [0m [38;5;63m│  [0m [38;5;63m├──[0m [38;5;212mPink Hoodie[0m
+[38;5;63m│  [0m [38;5;63m│  [0m [38;5;63m╰──[0m [38;5;212mBaseball Cap[0m
+[38;5;63m│  [0m [38;5;63m╰──[0m [38;5;212mMakeup[0m
+[38;5;63m├──[0m [38;5;212mFenty Beauty[0m
+[38;5;63m│  [0m [38;5;63m├──[0m [38;5;212mGloss Bomb Universal Lip Luminizer[0m
+[38;5;63m│  [0m [38;5;63m╰──[0m [38;5;212mHot Cheeks Velour Blushlighter[0m
+[38;5;63m├──[0m [38;5;212mNyx[0m
+[38;5;63m├──[0m [38;5;212mMac[0m
+[38;5;63m╰──[0m [38;5;212mMilk[0m
diff --git a/tree/tree_test.go b/tree/tree_test.go
@@ -426,6 +426,38 @@ func TestRootStyle(t *testing.T) {
 	golden.RequireEqual(t, []byte(tree.String()))
 }
 
+func TestInheritedStyles(t *testing.T) {
+	enumeratorStyle := lipgloss.NewStyle().Foreground(lipgloss.Color("63")).MarginRight(1)
+	rootStyle := lipgloss.NewStyle().Foreground(lipgloss.Color("35"))
+	itemStyle := lipgloss.NewStyle().Foreground(lipgloss.Color("212"))
+
+	tr := tree.
+		Root("⁜ Makeup").
+		Child(
+			"Glossier",
+			"Fenty Beauty",
+			tree.New().Child(
+				"Gloss Bomb Universal Lip Luminizer",
+				"Hot Cheeks Velour Blushlighter",
+			),
+			"Nyx",
+			"Mac",
+			"Milk",
+		).
+		Enumerator(tree.RoundedEnumerator).
+		EnumeratorStyle(enumeratorStyle).
+		RootStyle(rootStyle).
+		ItemStyle(itemStyle)
+	// Add a Tree as a Child of "Glossier"
+	tr.Replace(0, tr.Children().At(0).Child(
+		tree.Root("Apparel").Child("Pink Hoodie", "Baseball Cap"),
+	))
+
+	// Add a Leaf as a Child of "Glossier"
+	tr.Children().At(0).Child("Makeup")
+	golden.RequireEqual(t, []byte(tr.String()))
+}
+
 func TestAt(t *testing.T) {
 	data := tree.NewStringData("Foo", "Bar")