diff --git a/console/palette_test.go b/console/palette_test.go new file mode 100644 index 0000000..55a205e --- /dev/null +++ b/console/palette_test.go @@ -0,0 +1,32 @@ +package console + +import "testing" + +func TestGetPalette(t *testing.T) { + var ( + darkPalette = Palette{ + BaseColor: ColorWhite, + ReverseColor: ColorBlack, + } + lightPalette = Palette{ + BaseColor: ColorBlack, + ReverseColor: ColorWhite, + } + ) + + tests := []struct { + name string + input Theme + want Palette + }{ + {"should return dark theme with base color white", ThemeDark, darkPalette}, + {"should return light theme with base color black", ThemeLight, lightPalette}, + } + + for _, test := range tests { + palette := GetPalette(test.input) + if got := palette.BaseColor; got != test.want.BaseColor { + t.Errorf("GetPalette(%q) = %d, want %d", test.input, got, test.want.BaseColor) + } + } +}