1 files changed, 100 insertions, 0 deletions
diff --git a/vendor/github.com/oauth2-proxy/mockoidc/queue.go b/vendor/github.com/oauth2-proxy/mockoidc/queue.go
new file mode 100644
index 0000000..bd706d3
--- /dev/null
+++ b/vendor/github.com/oauth2-proxy/mockoidc/queue.go
@@ -0,0 +1,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
+}