add Makefile, internal/cvss/v{2,31}vector_test.go

2 files changed, 36 insertions, 0 deletions

diff --git a/internal/cvss/v2vector_test.go b/internal/cvss/v2vector_test.go
new file mode 100644
index 0000000..3ef4194
--- /dev/null
+++ b/internal/cvss/v2vector_test.go
@@ -0,0 +1,17 @@
+package cvss
+
+import "testing"
+
+// test v2 vector
+var testCvssV2 = "AV:L/AC:L/Au:N/C:N/I:N/A:P"
+
+// Test cvss v2 parser
+func TestCvssV2Parser(t *testing.T) {
+  t.Run("TestV2Parser", func(t *testing.T) {
+    // parse vector, check for error
+    _, err := NewVector(testCvssV2)
+    if err != nil {
+      t.Error(err)
+    }
+  })
+}
diff --git a/internal/cvss/v31vector_test.go b/internal/cvss/v31vector_test.go
new file mode 100644
index 0000000..8340123
--- /dev/null
+++ b/internal/cvss/v31vector_test.go
@@ -0,0 +1,19 @@
+package cvss
+
+import "testing"
+
+// test cvss v3.1 vector
+var testCvssV3 = "CVSS:3.1/AV:L/AC:L/PR:H/UI:N/S:U/C:N/I:N/A:H"
+
+// Test CVSS v3 parser.
+func TestCvssV3Parser(t *testing.T) {
+  t.Run("TestV3", func(t *testing.T) {
+    // parse vector, check for error
+    if _, err := NewVector(testCvssV3); err != nil {
+      t.Fatal(err)
+    }
+  })
+
+  // fmt.Println(v.Metrics())
+  // fmt.Println(v)
+}