From a40213480929a93a78239053a40a6ee93547c063 Mon Sep 17 00:00:00 2001
From: Paul Duncan <pabs@pablotron.org>
Date: Tue, 1 Feb 2022 13:23:44 -0500
Subject: internal/feed: add score, score tests

---
 internal/feed/feed.go       | 21 -----------
 internal/feed/score.go      | 34 +++++++++++++++++
 internal/feed/score_test.go | 92 +++++++++++++++++++++++++++++++++++++++++++++
 3 files changed, 126 insertions(+), 21 deletions(-)
 create mode 100644 internal/feed/score.go
 create mode 100644 internal/feed/score_test.go

(limited to 'internal/feed')

diff --git a/internal/feed/feed.go b/internal/feed/feed.go
index 9c27fb0..b8a3849 100644
--- a/internal/feed/feed.go
+++ b/internal/feed/feed.go
@@ -21,27 +21,6 @@ const (
 
 // TODO: parse cpe
 
-// CVSS score
-type Score float32
-
-// Unmarshal CVSS score from JSON.
-func (me *Score) UnmarshalJSON(b []byte) error {
-  // decode float, check for error
-  var v float32
-  if err := json.Unmarshal(b, &v); err != nil {
-    return err
-  }
-
-  // check score
-  if v < 0.0 || v > 10.0 {
-    return fmt.Errorf("score out of bounds: %f", v)
-  }
-
-  // save result, return success
-  *me = Score(v)
-  return nil
-}
-
 // CVSS severity
 type Severity int
 
diff --git a/internal/feed/score.go b/internal/feed/score.go
new file mode 100644
index 0000000..4e7da57
--- /dev/null
+++ b/internal/feed/score.go
@@ -0,0 +1,34 @@
+package feed
+
+import (
+  "encoding/json"
+  "fmt"
+  "math"
+  "strconv"
+)
+
+// CVSS score
+type Score float32
+
+// Unmarshal CVSS score from JSON.
+func (me *Score) UnmarshalJSON(b []byte) error {
+  // decode float, check for error
+  var v float32
+  if err := json.Unmarshal(b, &v); err != nil {
+    return err
+  }
+
+  // check score
+  if v < 0.0 || v > 10.0 {
+    return fmt.Errorf("CVSS score out of bounds: %2.1f", v)
+  }
+
+  // save result, return success
+  *me = Score(v)
+  return nil
+}
+
+func (me Score) String() string {
+  val := math.Trunc(10.0 * float64(me)) / 10.0
+  return strconv.FormatFloat(val, 'f', 1, 64)
+}
diff --git a/internal/feed/score_test.go b/internal/feed/score_test.go
new file mode 100644
index 0000000..0bda1b2
--- /dev/null
+++ b/internal/feed/score_test.go
@@ -0,0 +1,92 @@
+package feed
+
+import (
+  "encoding/json"
+  "testing"
+)
+
+func TestScoreUnmarshalInvalidData(t *testing.T) {
+  test := []byte(`{}`)
+  var val Score
+
+  if err := json.Unmarshal(test, &val); err == nil {
+    t.Errorf("got \"%s\", exp error", val)
+  }
+}
+
+func TestScoreUnmarshalInvalidValues(t *testing.T) {
+  tests := []struct {
+    val string
+    exp string
+  } {
+    { `-100.0`, "CVSS score out of bounds: -100.0" },
+    { `-90.0`, "CVSS score out of bounds: -90.0" },
+    { `-9.3`, "CVSS score out of bounds: -9.3" },
+    { `-1`, "CVSS score out of bounds: -1.0" },
+    { `10.1`, "CVSS score out of bounds: 10.1" },
+    { `100.0`, "CVSS score out of bounds: 100.0" },
+  }
+
+  for _, test := range(tests) {
+    t.Run(test.val, func(t *testing.T) {
+      var got Score
+
+      if err := json.Unmarshal([]byte(test.val), &got); err == nil {
+        t.Errorf("got \"%s\", exp error", got)
+      } else if err.Error() != test.exp {
+        t.Errorf("got \"%s\", exp \"%s\"", err.Error(), test.exp)
+      }
+    })
+  }
+}
+
+func TestScoreUnmarshalValidValues(t *testing.T) {
+  tests := []struct {
+    val string
+    exp float32
+  } {
+    { `0.0`, 0.0 },
+    { `0.1`, 0.1 },
+    { `1.2`, 1.2 },
+    { `5.9`, 5.9 },
+    { `9.9`, 9.9 },
+    { `10.0`, 10.0 },
+  }
+
+  for _, test := range(tests) {
+    t.Run(test.val, func(t *testing.T) {
+      var got Score
+
+      if err := json.Unmarshal([]byte(test.val), &got); err != nil {
+        t.Error(err)
+        return
+      } else if float32(got) != test.exp {
+        t.Errorf("got \"%f\", exp \"%f\"", float32(got), test.exp)
+      }
+    })
+  }
+}
+
+func TestScoreString(t *testing.T) {
+  tests := []struct {
+    val float32
+    exp string
+  } {
+    { 0.0, "0.0" },
+    { 0.01, "0.0" },
+    { 0.09, "0.0" },
+    { 1.2222, "1.2" },
+    { 5.9, "5.9" },
+    { 9.99, "9.9" },
+    { 10.0, "10.0" },
+  }
+
+  for _, test := range(tests) {
+    t.Run(test.exp, func(t *testing.T) {
+      got := Score(test.val).String()
+      if got != test.exp {
+        t.Errorf("got \"%s\", exp \"%s\"", got, test.exp)
+      }
+    })
+  }
+}
-- 
cgit v1.2.3