initial commit

pkg/op/mock/storage.mock.impl.go

@@ -0,0 +1,142 @@
+package mock
+
+import (
+	"context"
+	"errors"
+	"testing"
+	"time"
+
+	"gopkg.in/square/go-jose.v2"
+
+	"github.com/golang/mock/gomock"
+
+	"github.com/caos/oidc/pkg/op"
+)
+
+func NewStorage(t *testing.T) op.Storage {
+	return NewMockStorage(gomock.NewController(t))
+}
+
+func NewMockStorageExpectValidClientID(t *testing.T) op.Storage {
+	m := NewStorage(t)
+	ExpectValidClientID(m)
+	return m
+}
+
+func NewMockStorageExpectInvalidClientID(t *testing.T) op.Storage {
+	m := NewStorage(t)
+	ExpectInvalidClientID(m)
+	return m
+}
+
+func NewMockStorageAny(t *testing.T) op.Storage {
+	m := NewStorage(t)
+	mockS := m.(*MockStorage)
+	mockS.EXPECT().GetClientByClientID(gomock.Any(), gomock.Any()).AnyTimes().Return(&ConfClient{}, nil)
+	mockS.EXPECT().AuthorizeClientIDSecret(gomock.Any(), gomock.Any(), gomock.Any()).AnyTimes().Return(nil)
+	return m
+}
+
+func NewMockStorageSigningKeyError(t *testing.T) op.Storage {
+	m := NewStorage(t)
+	ExpectSigningKeyError(m)
+	return m
+}
+
+func NewMockStorageSigningKeyInvalid(t *testing.T) op.Storage {
+	m := NewStorage(t)
+	ExpectSigningKeyInvalid(m)
+	return m
+}
+func NewMockStorageSigningKey(t *testing.T) op.Storage {
+	m := NewStorage(t)
+	ExpectSigningKey(m)
+	return m
+}
+
+func ExpectInvalidClientID(s op.Storage) {
+	mockS := s.(*MockStorage)
+	mockS.EXPECT().GetClientByClientID(gomock.Any(), gomock.Any()).Return(nil, errors.New("client not found"))
+}
+
+func ExpectValidClientID(s op.Storage) {
+	mockS := s.(*MockStorage)
+	mockS.EXPECT().GetClientByClientID(gomock.Any(), gomock.Any()).DoAndReturn(
+		func(_ context.Context, id string) (op.Client, error) {
+			var appType op.ApplicationType
+			var authMethod op.AuthMethod
+			var accessTokenType op.AccessTokenType
+			switch id {
+			case "web_client":
+				appType = op.ApplicationTypeWeb
+				authMethod = op.AuthMethodBasic
+				accessTokenType = op.AccessTokenTypeBearer
+			case "native_client":
+				appType = op.ApplicationTypeNative
+				authMethod = op.AuthMethodNone
+				accessTokenType = op.AccessTokenTypeBearer
+			case "useragent_client":
+				appType = op.ApplicationTypeUserAgent
+				authMethod = op.AuthMethodBasic
+				accessTokenType = op.AccessTokenTypeJWT
+			}
+			return &ConfClient{id: id, appType: appType, authMethod: authMethod, accessTokenType: accessTokenType}, nil
+		})
+}
+
+func ExpectSigningKeyError(s op.Storage) {
+	mockS := s.(*MockStorage)
+	mockS.EXPECT().GetSigningKey(gomock.Any()).Return(nil, errors.New("error"))
+}
+
+func ExpectSigningKeyInvalid(s op.Storage) {
+	mockS := s.(*MockStorage)
+	mockS.EXPECT().GetSigningKey(gomock.Any()).Return(&jose.SigningKey{}, nil)
+}
+
+func ExpectSigningKey(s op.Storage) {
+	mockS := s.(*MockStorage)
+	mockS.EXPECT().GetSigningKey(gomock.Any()).Return(&jose.SigningKey{Algorithm: jose.HS256, Key: []byte("key")}, nil)
+}
+
+type ConfClient struct {
+	id              string
+	appType         op.ApplicationType
+	authMethod      op.AuthMethod
+	accessTokenType op.AccessTokenType
+}
+
+func (c *ConfClient) RedirectURIs() []string {
+	return []string{
+		"https://registered.com/callback",
+		"http://registered.com/callback",
+		"http://localhost:9999/callback",
+		"custom://callback",
+	}
+}
+
+func (c *ConfClient) LoginURL(id string) string {
+	return "login?id=" + id
+}
+
+func (c *ConfClient) ApplicationType() op.ApplicationType {
+	return c.appType
+}
+
+func (c *ConfClient) GetAuthMethod() op.AuthMethod {
+	return c.authMethod
+}
+
+func (c *ConfClient) GetID() string {
+	return c.id
+}
+
+func (c *ConfClient) AccessTokenLifetime() time.Duration {
+	return time.Duration(5 * time.Minute)
+}
+func (c *ConfClient) IDTokenLifetime() time.Duration {
+	return time.Duration(5 * time.Minute)
+}
+func (c *ConfClient) AccessTokenType() op.AccessTokenType {
+	return c.accessTokenType
+}