From e24e6af76a1dcd64889f47e3fae03d7c3fa6f7a3 Mon Sep 17 00:00:00 2001
From: Paul Duncan <pabs@pablotron.org>
Date: Mon, 7 Feb 2022 12:09:55 -0500
Subject: cvss: add score severity and tests

---
 cvss/score_test.go | 48 ++++++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 48 insertions(+)

(limited to 'cvss/score_test.go')

diff --git a/cvss/score_test.go b/cvss/score_test.go
index d35c4c8..5bff327 100644
--- a/cvss/score_test.go
+++ b/cvss/score_test.go
@@ -90,3 +90,51 @@ func TestScoreFloat(t *testing.T) {
     })
   }
 }
+
+func TestScoreSeverity(t *testing.T) {
+  passTests := []struct {
+    val float64
+    exp Severity
+  } {
+    { 0.0, None },
+    { 0.1, Low },
+    { 3.9, Low },
+    { 4.0, Medium },
+    { 6.9, Medium },
+    { 7.0, High },
+    { 8.9, High },
+    { 9.0, Critical },
+    { 10.0, Critical },
+  }
+
+  for _, test := range(passTests) {
+    t.Run(strconv.FormatFloat(test.val, 'f', 2, 64), func(t *testing.T) {
+      s, err := NewScore(test.val)
+      if err != nil {
+        t.Error(err)
+        return
+      }
+
+      got := s.Severity()
+      if got != test.exp {
+        t.Errorf("got \"%s\", exp \"%s\"", got, test.exp)
+      }
+    })
+  }
+
+  failTests := []struct {
+    val Score
+    exp Severity
+  } {
+    { Score(uint8(110)), Unknown },
+  }
+
+  for _, test := range(failTests) {
+    t.Run(test.val.String(), func(t *testing.T) {
+      got := test.val.Severity()
+      if got != test.exp {
+        t.Errorf("got \"%s\", exp \"%s\"", got, test.exp)
+      }
+    })
+  }
+}
-- 
cgit v1.2.3