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
|
package cors
import (
"strings"
"testing"
)
func TestWildcard(t *testing.T) {
w := wildcard{"foo", "bar"}
if !w.match("foobar") {
t.Error("foo*bar should match foobar")
}
if !w.match("foobazbar") {
t.Error("foo*bar should match foobazbar")
}
if w.match("foobaz") {
t.Error("foo*bar should not match foobaz")
}
w = wildcard{"foo", "oof"}
if w.match("foof") {
t.Error("foo*oof should not match foof")
}
}
func TestConvert(t *testing.T) {
s := convert([]string{"A", "b", "C"}, strings.ToLower)
e := []string{"a", "b", "c"}
if s[0] != e[0] || s[1] != e[1] || s[2] != e[2] {
t.Errorf("%v != %v", s, e)
}
}
func TestParseHeaderList(t *testing.T) {
h := parseHeaderList("header, second-header, THIRD-HEADER, Numb3r3d-H34d3r")
e := []string{"Header", "Second-Header", "Third-Header", "Numb3r3d-H34d3r"}
if h[0] != e[0] || h[1] != e[1] || h[2] != e[2] {
t.Errorf("%v != %v", h, e)
}
}
func TestParseHeaderListEmpty(t *testing.T) {
if len(parseHeaderList("")) != 0 {
t.Error("should be empty sclice")
}
if len(parseHeaderList(" , ")) != 0 {
t.Error("should be empty sclice")
}
}
func BenchmarkParseHeaderList(b *testing.B) {
b.ReportAllocs()
for i := 0; i < b.N; i++ {
parseHeaderList("header, second-header, THIRD-HEADER")
}
}
func BenchmarkParseHeaderListSingle(b *testing.B) {
b.ReportAllocs()
for i := 0; i < b.N; i++ {
parseHeaderList("header")
}
}
func BenchmarkParseHeaderListNormalized(b *testing.B) {
b.ReportAllocs()
for i := 0; i < b.N; i++ {
parseHeaderList("Header1, Header2, Third-Header")
}
}
func BenchmarkWildcard(b *testing.B) {
w := wildcard{"foo", "bar"}
b.Run("match", func(b *testing.B) {
b.ReportAllocs()
for i := 0; i < b.N; i++ {
w.match("foobazbar")
}
})
b.Run("too short", func(b *testing.B) {
b.ReportAllocs()
for i := 0; i < b.N; i++ {
w.match("fobar")
}
})
}
|
