forked from owasp-amass/amass
-
Notifications
You must be signed in to change notification settings - Fork 0
/
set_test.go
91 lines (81 loc) · 2.07 KB
/
set_test.go
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
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
package stringset
import (
"testing"
)
func TestSetHas(t *testing.T) {
set := New("test1", "test2", "test3", "test1", "test2")
if !set.Has("test1") {
t.Errorf("Set missing expected value")
}
}
func TestSetInsert(t *testing.T) {
expected := 3
set := New()
set.Insert("test1")
set.Insert("test2")
set.Insert("test3")
set.Insert("test3")
set.Insert("test2")
set.Insert("test1")
if len(set) != expected {
t.Errorf("Got %d, expected %d", len(set), expected)
}
}
func TestSetInsertMany(t *testing.T) {
expected := 3
set := New()
set.InsertMany("test1", "test2", "test3", "test1", "test2")
if len(set) != expected {
t.Errorf("Got %d, expected %d", len(set), expected)
}
}
func TestSetRemove(t *testing.T) {
expected := 2
set := New("test1", "test2", "test3", "test1", "test2")
set.Remove("test1")
if len(set) != expected {
t.Errorf("Got %d, expected %d", len(set), expected)
}
}
func TestSetSlice(t *testing.T) {
expected := 3
set := New("test1", "test2", "test3", "test1", "test2")
slice := set.Slice()
if len(slice) != expected {
t.Errorf("Got %d, expected %d", len(set), expected)
}
}
func TestSetLen(t *testing.T) {
expected := 3
set := New("test1", "test2", "test3", "test1", "test2")
if len(set) != expected {
t.Errorf("Got %d, expected %d", len(set), expected)
}
}
func TestSetUnion(t *testing.T) {
expected := 6
set1 := New("test1", "test2", "test3", "test6")
set2 := New("test1", "test2", "test3", "test4", "test5")
set1.Union(set2)
if len(set1) != expected {
t.Errorf("Got %d, expected %d", len(set1), expected)
}
}
func TestSetIntersect(t *testing.T) {
expected := 3
set1 := New("test1", "test2", "test3", "test6")
set2 := New("test1", "test2", "test3", "test4", "test5")
set1.Intersect(set2)
if len(set1) != expected {
t.Errorf("Got %d, expected %d", len(set1), expected)
}
}
func TestSetSubtract(t *testing.T) {
expected := 1
set1 := New("test1", "test2", "test3", "test6")
set2 := New("test1", "test2", "test3", "test4", "test5")
set1.Subtract(set2)
if len(set1) != expected {
t.Errorf("Got %d, expected %d", len(set1), expected)
}
}