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
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
|
package cvss
import (
"testing"
)
func TestNewV30Vector(t *testing.T) {
passTests := []struct {
val string
exp []string
} {
{
val: "CVSS:3.0/AV:L/AC:H/PR:L/UI:N/S:U/C:H/I:N/A:N",
exp: []string { "AV:L", "AC:H", "PR:L", "UI:N", "S:U", "C:H", "I:N", "A:N" },
},
}
for _, test := range(passTests) {
t.Run(test.val, func(t *testing.T) {
// parse vector
vec, err := newV30Vector(test.val)
if err != nil {
t.Error(err)
return
}
// check version
if vec.Version() != V30 {
t.Errorf("got %s, exp %s", vec.Version(), V30)
return
}
// check metric length
if len(vec.Metrics()) != len(test.exp) {
t.Errorf("got %d, exp %d", len(vec.Metrics()), len(test.exp))
return
}
// check metrics
for i, m := range(vec.Metrics()) {
got := m.String()
if got != test.exp[i] {
t.Errorf("got %s, exp %s", got, test.exp[i])
}
}
})
}
failTests := []struct {
val string
exp string
} {
{
val: "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",
},
}
for _, test := range(failTests) {
t.Run(test.val, func(t *testing.T) {
got, err := newV30Vector(test.val)
if err != nil && err.Error() != test.exp {
t.Errorf("got \"%s\", exp \"%s\"", err.Error(), test.exp)
} else if err == nil {
t.Errorf("got \"%s\", exp badMetric", got)
}
})
}
}
|
