-
Notifications
You must be signed in to change notification settings - Fork 0
/
options_test.go
62 lines (50 loc) · 1.54 KB
/
options_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
package csrf
// Options realization from
// https://github.com/gorilla/csrf/blob/master/options.go
// Copyright (c) 2015, Matt Silverlock (matt@eatsleeprepeat.net) All rights
// reserved.
import (
"net/http"
"reflect"
"testing"
)
// Tests that options functions are applied to the middleware.
func TestOptions(t *testing.T) {
var h http.Handler
age := 30
ageSec := age * 60
domain := "example.io"
header := "X-CSRF-Token"
errorHandler := unauthorizedHandler
cookie := "X-CSRF-Token"
testOpts := []Option{
MaxAge(age),
Domain(domain),
Secure(false),
RequestHeader(header),
ErrHandler(errorHandler),
CookieName(cookie),
}
// Parse our test options and check that they set the related struct fields.
cs := parseOptions(h, testOpts...)
if cs.opts.MaxAge != ageSec {
t.Errorf("MaxAge not set correctly: got %v want %v", cs.opts.MaxAge, ageSec)
}
if cs.opts.Domain != domain {
t.Errorf("Domain not set correctly: got %v want %v", cs.opts.Domain, domain)
}
if cs.opts.Secure != false {
t.Errorf("Secure not set correctly: got %v want %v", cs.opts.Secure, false)
}
if cs.opts.RequestHeader != header {
t.Errorf("RequestHeader not set correctly: got %v want %v", cs.opts.RequestHeader, header)
}
if !reflect.ValueOf(cs.opts.ErrHandler).IsValid() {
t.Errorf("ErrHandler not set correctly: got %v want %v",
reflect.ValueOf(cs.opts.ErrHandler).IsValid(), reflect.ValueOf(errorHandler).IsValid())
}
if cs.opts.CookieName != cookie {
t.Errorf("CookieName not set correctly: got %v want %v",
cs.opts.CookieName, cookie)
}
}