backup

pkg/op/mock/authorizer.mock.impl.go

@@ -0,0 +1,86 @@
+package mock
+
+import (
+	http "net/http"
+	"testing"
+
+	"github.com/stretchr/testify/require"
+
+	"github.com/golang/mock/gomock"
+	"github.com/gorilla/schema"
+
+	oidc "github.com/caos/oidc/pkg/oidc"
+	"github.com/caos/oidc/pkg/op"
+	u "github.com/caos/oidc/pkg/op/u"
+)
+
+func NewAuthorizer(t *testing.T) op.Authorizer {
+	return NewMockAuthorizer(gomock.NewController(t))
+}
+
+func NewAuthorizerExpectValid(t *testing.T, wantErr bool) op.Authorizer {
+	m := NewAuthorizer(t)
+	ExpectDecoder(m)
+	ExpectEncoder(m)
+	ExpectSigner(m, t)
+	ExpectStorage(m, t)
+	ExpectErrorHandler(m, t, wantErr)
+	return m
+}
+
+// func NewAuthorizerExpectDecoderFails(t *testing.T) op.Authorizer {
+// 	m := NewAuthorizer(t)
+// 	ExpectDecoderFails(m)
+// 	ExpectEncoder(m)
+// 	ExpectSigner(m, t)
+// 	ExpectStorage(m, t)
+// 	ExpectErrorHandler(m, t)
+// 	return m
+// }
+
+func ExpectDecoder(a op.Authorizer) {
+	mockA := a.(*MockAuthorizer)
+	mockA.EXPECT().Decoder().AnyTimes().Return(schema.NewDecoder())
+}
+
+func ExpectEncoder(a op.Authorizer) {
+	mockA := a.(*MockAuthorizer)
+	mockA.EXPECT().Encoder().AnyTimes().Return(schema.NewEncoder())
+}
+
+func ExpectSigner(a op.Authorizer, t *testing.T) {
+	mockA := a.(*MockAuthorizer)
+	mockA.EXPECT().Signe().DoAndReturn(
+		func() u.Signer {
+			return &Sig{}
+		})
+}
+
+func ExpectErrorHandler(a op.Authorizer, t *testing.T, wantErr bool) {
+	mockA := a.(*MockAuthorizer)
+	mockA.EXPECT().ErrorHandler().AnyTimes().
+		Return(func(w http.ResponseWriter, r *http.Request, authReq *oidc.AuthRequest, err error) {
+			if wantErr {
+				require.Error(t, err)
+				return
+			}
+			require.NoError(t, err)
+		})
+}
+
+type Sig struct{}
+
+func (s *Sig) Sign(*oidc.IDTokenClaims) (string, error) {
+	return "", nil
+}
+
+func ExpectStorage(a op.Authorizer, t *testing.T) {
+	mockA := a.(*MockAuthorizer)
+	mockA.EXPECT().Storage().AnyTimes().Return(NewMockStorageAny(t))
+}
+
+// func NewMockSignerAny(t *testing.T) op.Signer {
+// 	m := NewMockSigner(gomock.NewController(t))
+// 	m.EXPECT().Sign(gomock.Any()).AnyTimes().Return("", nil)
+// 	return m
+// }