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
92
93
94
95
96
97
98
99
100
|
package mockoidc
import "sync"
// UserQueue manages the queue of Users returned for each
// call to the authorize endpoint
type UserQueue struct {
sync.Mutex
Queue []User
}
// CodeQueue manages the queue of codes returned for each
// call to the authorize endpoint
type CodeQueue struct {
sync.Mutex
Queue []string
}
// ErrorQueue manages the queue of errors for handlers to return
type ErrorQueue struct {
sync.Mutex
Queue []*ServerError
}
// ServerError is a tester-defined error for a handler to return
type ServerError struct {
Code int
Error string
Description string
}
// Push adds a User to the Queue to be set in subsequent calls to the
// `authorization_endpoint`
func (q *UserQueue) Push(user User) {
q.Lock()
defer q.Unlock()
q.Queue = append(q.Queue, user)
}
// Pop a User from the Queue. If empty, return `DefaultUser()`
func (q *UserQueue) Pop() User {
q.Lock()
defer q.Unlock()
if len(q.Queue) == 0 {
return DefaultUser()
}
var user User
user, q.Queue = q.Queue[0], q.Queue[1:]
return user
}
// Push adds a code to the Queue to be returned by subsequent
// `authorization_endpoint` calls as the code
func (q *CodeQueue) Push(code string) {
q.Lock()
defer q.Unlock()
q.Queue = append(q.Queue, code)
}
// Pop a `code` from the Queue. If empty, return a random code
func (q *CodeQueue) Pop() (string, error) {
q.Lock()
defer q.Unlock()
if len(q.Queue) == 0 {
code, err := randomNonce(24)
if err != nil {
return "", err
}
return code, nil
}
var code string
code, q.Queue = q.Queue[0], q.Queue[1:]
return code, nil
}
// Push adds a ServerError to the Queue to be returned in subsequent
// handler calls
func (q *ErrorQueue) Push(se *ServerError) {
q.Lock()
defer q.Unlock()
q.Queue = append(q.Queue, se)
}
// Pop a ServerError from the Queue. If empty, return nil
func (q *ErrorQueue) Pop() *ServerError {
q.Lock()
defer q.Unlock()
if len(q.Queue) == 0 {
return nil
}
var se *ServerError
se, q.Queue = q.Queue[0], q.Queue[1:]
return se
}
|
