diff options
Diffstat (limited to 'internal/cvss/v2vector_test.go')
-rw-r--r-- | internal/cvss/v2vector_test.go | 48 |
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 + } +} |