Improve test coverage for color pkg

chelnak · Jul 18, 2022 · 5f7a7a7 · 5f7a7a7
1 parent 5175349
commit 5f7a7a7
Show file tree

Hide file tree

Showing 2 changed files with 20 additions and 1 deletion.
diff --git a/pkg/colors/colors.go b/pkg/colors/colors.go
@@ -41,5 +41,9 @@ func GetColor(c Color) *color.Color {
 		return nil
 	}
 
-	return color.New(lookup[c])
+	if val, ok := lookup[c]; ok {
+		return color.New(val)
+	}
+
+	return nil
 }
diff --git a/pkg/colors/colors_test.go b/pkg/colors/colors_test.go
@@ -34,6 +34,21 @@ func TestGetColor(t *testing.T) {
 			c:    colors.FgHiRed,
 			want: color.New(color.FgHiRed),
 		},
+		{
+			name: "GetColor returns the correct mapping for FgHiWhite",
+			c:    colors.FgHiWhite,
+			want: color.New(color.FgHiWhite),
+		},
+		{
+			name: "GetColor returns the correct mapping for NoColor",
+			c:    colors.NoColor,
+			want: nil,
+		},
+		{
+			name: "GetColor returns nil for an unknown color",
+			c:    colors.Color(100),
+			want: nil,
+		},
 	}
 
 	for _, tt := range tests {