internal/cvss: more tests

1 files changed, 18 insertions, 0 deletions

diff --git a/internal/cvss/v3key_test.go b/internal/cvss/v3key_test.go
index ccf2b8f..517eebd 100644
--- a/internal/cvss/v3key_test.go
+++ b/internal/cvss/v3key_test.go
@@ -122,3 +122,21 @@ func TestV3KeyCategory(t *testing.T) {
     })
   }
 }
+
+func TestInvalidV3KeyName(t *testing.T) {
+  exp := "invalid"
+  got := v3Key(255).Name()
+
+  if got != exp {
+    t.Errorf("got: \"%s\", exp: \"%s\"", got, exp)
+  }
+}
+
+func TestInvalidV3KeyCategory(t *testing.T) {
+  exp := InvalidCategory
+  got := v3Key(255).Category()
+
+  if got != exp {
+    t.Errorf("got: \"%s\", exp: \"%s\"", got, exp)
+  }
+}