From 2c8c1a47cdfb230c67d2cfc235fa10eec479c557 Mon Sep 17 00:00:00 2001
From: Daniel Milde <daniel@milde.cz>
Date: Sun, 29 Dec 2024 17:55:59 +0100
Subject: [PATCH] test: cover new style options

---
 cmd/gdu/app/app_test.go | 14 ++++++++++++++
 tui/tui_test.go         | 36 +++++++++++++++++++++++-------------
 2 files changed, 37 insertions(+), 13 deletions(-)

diff --git a/cmd/gdu/app/app_test.go b/cmd/gdu/app/app_test.go
index 55b428ac1..377a140f4 100644
--- a/cmd/gdu/app/app_test.go
+++ b/cmd/gdu/app/app_test.go
@@ -275,6 +275,20 @@ func TestAnalyzePathWithStyle(t *testing.T) {
 				ProgressModal: ProgressModalOpts{
 					CurrentItemNameMaxLen: 10,
 				},
+				Footer: FooterColorStyle{
+					TextColor:       "black",
+					BackgroundColor: "red",
+					NumberColor:     "white",
+				},
+				Header: HeaderColorStyle{
+					TextColor:       "black",
+					BackgroundColor: "red",
+					Hidden:          true,
+				},
+				ResultRow: ResultRowColorStyle{
+					NumberColor:    "orange",
+					DirectoryColor: "blue",
+				},
 				UseOldSizeBar: true,
 			},
 		},
diff --git a/tui/tui_test.go b/tui/tui_test.go
index a3a7503b0..2b4274a42 100644
--- a/tui/tui_test.go
+++ b/tui/tui_test.go
@@ -700,28 +700,38 @@ func TestMin(t *testing.T) {
 	assert.Equal(t, 3, min(4, 3))
 }
 
-func TestSetSelectedBackgroundColor(t *testing.T) {
+func TestSetStyles(t *testing.T) {
 	simScreen := testapp.CreateSimScreen()
 	defer simScreen.Fini()
 
-	app := testapp.CreateMockedApp(true)
-	ui := CreateUI(app, simScreen, &bytes.Buffer{}, false, true, false, false, false)
-
-	ui.SetSelectedBackgroundColor(tcell.ColorRed)
-
-	assert.Equal(t, ui.selectedBackgroundColor, tcell.ColorRed)
-}
-
-func TestSetSelectedTextColor(t *testing.T) {
-	simScreen := testapp.CreateSimScreen()
-	defer simScreen.Fini()
+	opts := []Option{}
+	opts = append(opts, func(ui *UI) {
+		ui.SetHeaderHidden()
+	})
 
 	app := testapp.CreateMockedApp(true)
-	ui := CreateUI(app, simScreen, &bytes.Buffer{}, false, true, false, false, false)
+	ui := CreateUI(app, simScreen, &bytes.Buffer{}, false, true, false, false, false, opts...)
 
+	ui.SetSelectedBackgroundColor(tcell.ColorRed)
 	ui.SetSelectedTextColor(tcell.ColorRed)
+	ui.SetFooterTextColor("red")
+	ui.SetFooterBackgroundColor("red")
+	ui.SetFooterNumberColor("red")
+	ui.SetHeaderTextColor("red")
+	ui.SetHeaderBackgroundColor("red")
+	ui.SetResultRowDirectoryColor("red")
+	ui.SetResultRowNumberColor("red")
 
+	assert.Equal(t, ui.selectedBackgroundColor, tcell.ColorRed)
 	assert.Equal(t, ui.selectedTextColor, tcell.ColorRed)
+	assert.Equal(t, ui.footerTextColor, "red")
+	assert.Equal(t, ui.footerBackgroundColor, "red")
+	assert.Equal(t, ui.footerNumberColor, "red")
+	assert.Equal(t, ui.headerTextColor, "red")
+	assert.Equal(t, ui.headerBackgroundColor, "red")
+	assert.Equal(t, ui.headerHidden, true)
+	assert.Equal(t, ui.resultRow.DirectoryColor, "red")
+	assert.Equal(t, ui.resultRow.NumberColor, "red")
 }
 
 func TestSetCurrentItemNameMaxLen(t *testing.T) {