125 lines
2.9 KiB
Go
125 lines
2.9 KiB
Go
package mock
|
|
|
|
import (
|
|
"errors"
|
|
"testing"
|
|
|
|
"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()).AnyTimes().Return(&ConfClient{}, nil)
|
|
mockS.EXPECT().AuthorizeClientIDSecret(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()).Return(nil, errors.New("client not found"))
|
|
}
|
|
|
|
func ExpectValidClientID(s op.Storage) {
|
|
mockS := s.(*MockStorage)
|
|
mockS.EXPECT().GetClientByClientID(gomock.Any()).DoAndReturn(
|
|
func(id string) (op.Client, error) {
|
|
var appType op.ApplicationType
|
|
var authMethod op.AuthMethod
|
|
switch id {
|
|
case "web_client":
|
|
appType = op.ApplicationTypeWeb
|
|
authMethod = op.AuthMethodBasic
|
|
case "native_client":
|
|
appType = op.ApplicationTypeNative
|
|
authMethod = op.AuthMethodNone
|
|
case "useragent_client":
|
|
appType = op.ApplicationTypeUserAgent
|
|
authMethod = op.AuthMethodBasic
|
|
}
|
|
return &ConfClient{id: id, appType: appType, authMethod: authMethod}, nil
|
|
})
|
|
}
|
|
|
|
func ExpectSigningKeyError(s op.Storage) {
|
|
mockS := s.(*MockStorage)
|
|
mockS.EXPECT().GetSigningKey().Return(nil, errors.New("error"))
|
|
}
|
|
|
|
func ExpectSigningKeyInvalid(s op.Storage) {
|
|
mockS := s.(*MockStorage)
|
|
mockS.EXPECT().GetSigningKey().Return(&jose.SigningKey{}, nil)
|
|
}
|
|
|
|
func ExpectSigningKey(s op.Storage) {
|
|
mockS := s.(*MockStorage)
|
|
mockS.EXPECT().GetSigningKey().Return(&jose.SigningKey{Algorithm: jose.HS256, Key: []byte("key")}, nil)
|
|
}
|
|
|
|
type ConfClient struct {
|
|
id string
|
|
appType op.ApplicationType
|
|
authMethod op.AuthMethod
|
|
}
|
|
|
|
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
|
|
}
|