forked from ory/fosite
/
authorize_response_handler_test.go
78 lines (75 loc) · 2.35 KB
/
authorize_response_handler_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
package fosite_test
import (
"github.com/go-errors/errors"
"github.com/golang/mock/gomock"
. "github.com/ory-am/fosite"
. "github.com/ory-am/fosite/internal"
"github.com/stretchr/testify/assert"
"golang.org/x/net/context"
"net/http"
"testing"
)
func TestNewAuthorizeResponse(t *testing.T) {
ctrl := gomock.NewController(t)
handlers := []*MockAuthorizeEndpointHandler{NewMockAuthorizeEndpointHandler(ctrl)}
ar := NewMockAuthorizeRequester(ctrl)
defer ctrl.Finish()
ctx := context.Background()
oauth2 := &Fosite{
AuthorizeEndpointHandlers: AuthorizeEndpointHandlers{"a": handlers[0]},
}
duo := &Fosite{
AuthorizeEndpointHandlers: AuthorizeEndpointHandlers{"a": handlers[0], "b": handlers[0]},
}
ar.EXPECT().SetSession(gomock.Eq(struct{}{})).AnyTimes()
fooErr := errors.New("foo")
for k, c := range []struct {
isErr bool
mock func()
expectErr error
}{
{
mock: func() {
handlers[0].EXPECT().HandleAuthorizeEndpointRequest(gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any()).Return(fooErr)
},
isErr: true,
expectErr: fooErr,
},
{
mock: func() {
handlers[0].EXPECT().HandleAuthorizeEndpointRequest(gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any()).Return(nil)
ar.EXPECT().DidHandleAllResponseTypes().Return(true)
},
isErr: false,
},
{
mock: func() {
oauth2 = duo
handlers[0].EXPECT().HandleAuthorizeEndpointRequest(gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any()).Return(nil)
handlers[0].EXPECT().HandleAuthorizeEndpointRequest(gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any()).Return(nil)
ar.EXPECT().DidHandleAllResponseTypes().Return(true)
},
isErr: false,
},
{
mock: func() {
oauth2 = duo
handlers[0].EXPECT().HandleAuthorizeEndpointRequest(gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any()).Return(nil)
handlers[0].EXPECT().HandleAuthorizeEndpointRequest(gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any()).Return(fooErr)
},
isErr: true,
expectErr: fooErr,
},
} {
c.mock()
responder, err := oauth2.NewAuthorizeResponse(ctx, &http.Request{}, ar, struct{}{})
assert.Equal(t, c.isErr, err != nil, "%d: %s", k, err)
if err != nil {
assert.Equal(t, c.expectErr, err, "%d: %s", k, err)
assert.Nil(t, responder, "%d", k)
} else {
assert.NotNil(t, responder, "%d", k)
}
t.Logf("Passed test case %d", k)
}
}