-
Notifications
You must be signed in to change notification settings - Fork 0
/
interceptor_test.go
57 lines (52 loc) · 1.36 KB
/
interceptor_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
// Copyright (c) 2019,CAO HONGJU. All rights reserved.
// Use of this source code is governed by a MIT-style
// license that can be found in the LICENSE file.
package apirouter_test
import (
"net/http"
"testing"
"github.com/cnotch/apirouter"
"github.com/stretchr/testify/assert"
)
func TestWrap(t *testing.T) {
signature := ""
unWrap := func(w http.ResponseWriter, r *http.Request, ps apirouter.Params) {
signature += "D"
}
h := apirouter.Wrap(unWrap,
apirouter.NewInterceptor(
func(w http.ResponseWriter, r *http.Request) bool {
signature += "A"
return true
}, func(r *http.Request) {
signature += "B"
}),
apirouter.PreInterceptor(func(w http.ResponseWriter, r *http.Request) bool {
signature += "C"
return true
}),
)
h(nil, nil, apirouter.Params{})
assert.Equal(t, "ACDB", signature)
}
func TestWrapAbort(t *testing.T) {
signature := ""
unWrap := func(w http.ResponseWriter, r *http.Request, ps apirouter.Params) {
signature += "D"
}
h := apirouter.Wrap(unWrap,
apirouter.PreInterceptor(func(w http.ResponseWriter, r *http.Request) bool {
signature += "A"
return true
}),
apirouter.NewInterceptor(
func(w http.ResponseWriter, r *http.Request) bool {
signature += "B"
return false
}, func(r *http.Request) {
signature += "C"
}),
)
h(nil, nil, apirouter.Params{})
assert.Equal(t, "AB", signature)
}