internal/cvss: add many tests

1 files changed, 41 insertions, 7 deletions

diff --git a/internal/cvss/v2vector_test.go b/internal/cvss/v2vector_test.go
index 3ef4194..a943957 100644
--- a/internal/cvss/v2vector_test.go
+++ b/internal/cvss/v2vector_test.go
@@ -1,17 +1,51 @@
 package cvss
 
-import "testing"
+import (
+  "testing"
+)
 
-// test v2 vector
-var testCvssV2 = "AV:L/AC:L/Au:N/C:N/I:N/A:P"
-
-// Test cvss v2 parser
+// Test cvss v2
 func TestCvssV2Parser(t *testing.T) {
+  // test v2 vector
+  test := "AV:L/AC:L/Au:N/C:N/I:N/A:P"
+  expMetrics := []string { "AV:L", "AC:L", "Au:N", "C:N", "I:N", "A:P" }
+
   t.Run("TestV2Parser", func(t *testing.T) {
-    // parse vector, check for error
-    _, err := NewVector(testCvssV2)
+    // parse vector
+    vec, err := NewVector(test)
     if err != nil {
       t.Error(err)
+      return
+    }
+
+    // check version
+    if vec.Version() != V20 {
+      t.Errorf("got %s, exp %s", vec.Version(), V20)
+      return
+    }
+
+    for i, m := range(vec.Metrics()) {
+      got := m.String()
+      if got != expMetrics[i] {
+        t.Errorf("got %s, exp %s", got, expMetrics[i])
+      }
     }
   })
 }
+
+// Test cvss v2
+func TestInvalidV2Vector(t *testing.T) {
+  // test invalid vector
+  test := "AV:L/junk/Au:N/C:N/I:N/A:P"
+  exp := "invalid CVSS 2.0 metric: junk"
+
+  // parse vector
+  got, err := NewVector(test)
+  if err != nil && err.Error() != exp {
+    t.Errorf("got \"%s\", exp \"%s\"", err.Error(), err)
+    return
+  } else if err == nil {
+    t.Errorf("got \"%s\", exp badMetric", got)
+    return
+  }
+}