aboutsummaryrefslogtreecommitdiff
path: root/internal/cvss/v30vector_test.go
blob: fae3436ee794943652dc7168318a16448d0d6072 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
package cvss

import (
  "testing"
)

func TestV30Vector(t *testing.T) {
  // test v3 vector
  test := "CVSS:3.0/AV:L/AC:H/PR:L/UI:N/S:U/C:H/I:N/A:N"
  expMetrics := []string { "AV:L", "AC:H", "PR:L", "UI:N", "S:U", "C:H", "I:N", "A:N" }

  t.Run("TestV30Vector", func(t *testing.T) {
    // parse vector
    vec, err := NewVector(test)
    if err != nil {
      t.Error(err)
      return
    }

    // check version
    if vec.Version() != V30 {
      t.Errorf("got %s, exp %s", vec.Version(), V30)
      return
    }

    for i, m := range(vec.Metrics()) {
      got := m.String()
      if got != expMetrics[i] {
        t.Errorf("got %s, exp %s", got, expMetrics[i])
      }
    }
  })
}

func TestInvalidV30Vector(t *testing.T) {
  // test invalid vector
  test := "CVSS:3.0/AV:L/junk/PR:L/UI:N/S:U/C:H/I:N/A:N"
  exp := "invalid CVSS 3.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
  }
}