mirror of
https://github.com/zitadel/zitadel.git
synced 2025-04-25 13:40:51 +00:00
368 lines
12 KiB
Go
368 lines
12 KiB
Go
|
//go:build integration
|
||
|
|
|
||
|
|
package saml_test
|
||
|
|
|
||
|
|
import (
|
||
|
|
"context"
|
||
|
|
"net/url"
|
||
|
|
"os"
|
||
|
|
"regexp"
|
||
|
|
"testing"
|
||
|
|
"time"
|
||
|
|
|
||
|
|
"github.com/brianvoe/gofakeit/v6"
|
||
|
|
"github.com/crewjam/saml"
|
||
|
|
"github.com/muhlemmer/gu"
|
||
|
|
"github.com/stretchr/testify/assert"
|
||
|
|
"github.com/stretchr/testify/require"
|
||
|
|
"google.golang.org/protobuf/types/known/timestamppb"
|
||
|
|
|
||
|
|
"github.com/zitadel/zitadel/internal/integration"
|
||
|
|
"github.com/zitadel/zitadel/pkg/grpc/object/v2"
|
||
|
|
oidc_pb "github.com/zitadel/zitadel/pkg/grpc/oidc/v2"
|
||
|
|
saml_pb "github.com/zitadel/zitadel/pkg/grpc/saml/v2"
|
||
|
|
"github.com/zitadel/zitadel/pkg/grpc/session/v2"
|
||
|
|
)
|
||
|
|
|
||
|
|
var (
|
||
|
|
CTX context.Context
|
||
|
|
Instance *integration.Instance
|
||
|
|
Client saml_pb.SAMLServiceClient
|
||
|
|
)
|
||
|
|
|
||
|
|
func TestMain(m *testing.M) {
|
||
|
|
os.Exit(func() int {
|
||
|
|
ctx, cancel := context.WithTimeout(context.Background(), 15*time.Minute)
|
||
|
|
defer cancel()
|
||
|
|
|
||
|
|
Instance = integration.NewInstance(ctx)
|
||
|
|
Client = Instance.Client.SAMLv2
|
||
|
|
|
||
|
|
CTX = Instance.WithAuthorization(ctx, integration.UserTypeOrgOwner)
|
||
|
|
return m.Run()
|
||
|
|
}())
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestServer_GetAuthRequest(t *testing.T) {
|
||
|
|
rootURL := "https://sp.example.com"
|
||
|
|
idpMetadata, err := Instance.GetSAMLIDPMetadata()
|
||
|
|
require.NoError(t, err)
|
||
|
|
spMiddlewareRedirect, err := integration.CreateSAMLSP(rootURL, idpMetadata, saml.HTTPRedirectBinding)
|
||
|
|
require.NoError(t, err)
|
||
|
|
spMiddlewarePost, err := integration.CreateSAMLSP(rootURL, idpMetadata, saml.HTTPPostBinding)
|
||
|
|
require.NoError(t, err)
|
||
|
|
|
||
|
|
acsRedirect := idpMetadata.IDPSSODescriptors[0].SingleSignOnServices[0]
|
||
|
|
acsPost := idpMetadata.IDPSSODescriptors[0].SingleSignOnServices[1]
|
||
|
|
|
||
|
|
project, err := Instance.CreateProject(CTX)
|
||
|
|
require.NoError(t, err)
|
||
|
|
_, err = Instance.CreateSAMLClient(CTX, project.GetId(), spMiddlewareRedirect)
|
||
|
|
require.NoError(t, err)
|
||
|
|
_, err = Instance.CreateSAMLClient(CTX, project.GetId(), spMiddlewarePost)
|
||
|
|
require.NoError(t, err)
|
||
|
|
|
||
|
|
now := time.Now()
|
||
|
|
|
||
|
|
tests := []struct {
|
||
|
|
name string
|
||
|
|
dep func() (string, error)
|
||
|
|
want *oidc_pb.GetAuthRequestResponse
|
||
|
|
wantErr bool
|
||
|
|
}{
|
||
|
|
{
|
||
|
|
name: "Not found",
|
||
|
|
dep: func() (string, error) {
|
||
|
|
return "123", nil
|
||
|
|
},
|
||
|
|
wantErr: true,
|
||
|
|
},
|
||
|
|
{
|
||
|
|
name: "success, redirect binding",
|
||
|
|
dep: func() (string, error) {
|
||
|
|
return Instance.CreateSAMLAuthRequest(spMiddlewareRedirect, Instance.Users[integration.UserTypeOrgOwner].ID, acsRedirect, gofakeit.BitcoinAddress(), saml.HTTPRedirectBinding)
|
||
|
|
},
|
||
|
|
},
|
||
|
|
{
|
||
|
|
name: "success, post binding",
|
||
|
|
dep: func() (string, error) {
|
||
|
|
return Instance.CreateSAMLAuthRequest(spMiddlewarePost, Instance.Users[integration.UserTypeOrgOwner].ID, acsPost, gofakeit.BitcoinAddress(), saml.HTTPPostBinding)
|
||
|
|
},
|
||
|
|
},
|
||
|
|
}
|
||
|
|
for _, tt := range tests {
|
||
|
|
t.Run(tt.name, func(t *testing.T) {
|
||
|
|
authRequestID, err := tt.dep()
|
||
|
|
require.NoError(t, err)
|
||
|
|
|
||
|
|
retryDuration, tick := integration.WaitForAndTickWithMaxDuration(CTX, time.Minute)
|
||
|
|
require.EventuallyWithT(t, func(ttt *assert.CollectT) {
|
||
|
|
got, err := Client.GetSAMLRequest(CTX, &saml_pb.GetSAMLRequestRequest{
|
||
|
|
SamlRequestId: authRequestID,
|
||
|
|
})
|
||
|
|
if tt.wantErr {
|
||
|
|
assert.Error(ttt, err)
|
||
|
|
return
|
||
|
|
}
|
||
|
|
assert.NoError(ttt, err)
|
||
|
|
authRequest := got.GetSamlRequest()
|
||
|
|
assert.NotNil(ttt, authRequest)
|
||
|
|
assert.Equal(ttt, authRequestID, authRequest.GetId())
|
||
|
|
assert.WithinRange(ttt, authRequest.GetCreationDate().AsTime(), now.Add(-time.Second), now.Add(time.Second))
|
||
|
|
}, retryDuration, tick, "timeout waiting for expected saml request result")
|
||
|
|
})
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestServer_CreateResponse(t *testing.T) {
|
||
|
|
idpMetadata, err := Instance.GetSAMLIDPMetadata()
|
||
|
|
require.NoError(t, err)
|
||
|
|
rootURLRedirect := "spredirect.example.com"
|
||
|
|
spMiddlewareRedirect, err := integration.CreateSAMLSP("https://"+rootURLRedirect, idpMetadata, saml.HTTPRedirectBinding)
|
||
|
|
require.NoError(t, err)
|
||
|
|
rootURLPost := "sppost.example.com"
|
||
|
|
spMiddlewarePost, err := integration.CreateSAMLSP("https://"+rootURLPost, idpMetadata, saml.HTTPPostBinding)
|
||
|
|
require.NoError(t, err)
|
||
|
|
|
||
|
|
acsRedirect := idpMetadata.IDPSSODescriptors[0].SingleSignOnServices[0]
|
||
|
|
acsPost := idpMetadata.IDPSSODescriptors[0].SingleSignOnServices[1]
|
||
|
|
|
||
|
|
project, err := Instance.CreateProject(CTX)
|
||
|
|
require.NoError(t, err)
|
||
|
|
_, err = Instance.CreateSAMLClient(CTX, project.GetId(), spMiddlewareRedirect)
|
||
|
|
require.NoError(t, err)
|
||
|
|
_, err = Instance.CreateSAMLClient(CTX, project.GetId(), spMiddlewarePost)
|
||
|
|
require.NoError(t, err)
|
||
|
|
|
||
|
|
sessionResp, err := Instance.Client.SessionV2.CreateSession(CTX, &session.CreateSessionRequest{
|
||
|
|
Checks: &session.Checks{
|
||
|
|
User: &session.CheckUser{
|
||
|
|
Search: &session.CheckUser_UserId{
|
||
|
|
UserId: Instance.Users[integration.UserTypeOrgOwner].ID,
|
||
|
|
},
|
||
|
|
},
|
||
|
|
},
|
||
|
|
})
|
||
|
|
require.NoError(t, err)
|
||
|
|
|
||
|
|
tests := []struct {
|
||
|
|
name string
|
||
|
|
req *saml_pb.CreateResponseRequest
|
||
|
|
AuthError string
|
||
|
|
want *saml_pb.CreateResponseResponse
|
||
|
|
wantURL *url.URL
|
||
|
|
wantErr bool
|
||
|
|
}{
|
||
|
|
{
|
||
|
|
name: "Not found",
|
||
|
|
req: &saml_pb.CreateResponseRequest{
|
||
|
|
SamlRequestId: "123",
|
||
|
|
ResponseKind: &saml_pb.CreateResponseRequest_Session{
|
||
|
|
Session: &saml_pb.Session{
|
||
|
|
SessionId: sessionResp.GetSessionId(),
|
||
|
|
SessionToken: sessionResp.GetSessionToken(),
|
||
|
|
},
|
||
|
|
},
|
||
|
|
},
|
||
|
|
wantErr: true,
|
||
|
|
},
|
||
|
|
{
|
||
|
|
name: "session not found",
|
||
|
|
req: &saml_pb.CreateResponseRequest{
|
||
|
|
SamlRequestId: func() string {
|
||
|
|
authRequestID, err := Instance.CreateSAMLAuthRequest(spMiddlewareRedirect, Instance.Users[integration.UserTypeOrgOwner].ID, acsRedirect, gofakeit.BitcoinAddress(), saml.HTTPRedirectBinding)
|
||
|
|
require.NoError(t, err)
|
||
|
|
return authRequestID
|
||
|
|
}(),
|
||
|
|
ResponseKind: &saml_pb.CreateResponseRequest_Session{
|
||
|
|
Session: &saml_pb.Session{
|
||
|
|
SessionId: "foo",
|
||
|
|
SessionToken: "bar",
|
||
|
|
},
|
||
|
|
},
|
||
|
|
},
|
||
|
|
wantErr: true,
|
||
|
|
},
|
||
|
|
{
|
||
|
|
name: "session token invalid",
|
||
|
|
req: &saml_pb.CreateResponseRequest{
|
||
|
|
SamlRequestId: func() string {
|
||
|
|
authRequestID, err := Instance.CreateSAMLAuthRequest(spMiddlewareRedirect, Instance.Users[integration.UserTypeOrgOwner].ID, acsRedirect, gofakeit.BitcoinAddress(), saml.HTTPRedirectBinding)
|
||
|
|
require.NoError(t, err)
|
||
|
|
return authRequestID
|
||
|
|
}(),
|
||
|
|
ResponseKind: &saml_pb.CreateResponseRequest_Session{
|
||
|
|
Session: &saml_pb.Session{
|
||
|
|
SessionId: sessionResp.GetSessionId(),
|
||
|
|
SessionToken: "bar",
|
||
|
|
},
|
||
|
|
},
|
||
|
|
},
|
||
|
|
wantErr: true,
|
||
|
|
},
|
||
|
|
{
|
||
|
|
name: "fail callback, post",
|
||
|
|
req: &saml_pb.CreateResponseRequest{
|
||
|
|
SamlRequestId: func() string {
|
||
|
|
authRequestID, err := Instance.CreateSAMLAuthRequest(spMiddlewarePost, Instance.Users[integration.UserTypeOrgOwner].ID, acsPost, gofakeit.BitcoinAddress(), saml.HTTPPostBinding)
|
||
|
|
require.NoError(t, err)
|
||
|
|
return authRequestID
|
||
|
|
}(),
|
||
|
|
ResponseKind: &saml_pb.CreateResponseRequest_Error{
|
||
|
|
Error: &saml_pb.AuthorizationError{
|
||
|
|
Error: saml_pb.ErrorReason_ERROR_REASON_REQUEST_DENIED,
|
||
|
|
ErrorDescription: gu.Ptr("nope"),
|
||
|
|
},
|
||
|
|
},
|
||
|
|
},
|
||
|
|
want: &saml_pb.CreateResponseResponse{
|
||
|
|
Url: regexp.QuoteMeta(`https://` + rootURLPost + `/saml/acs`),
|
||
|
|
Binding: &saml_pb.CreateResponseResponse_Post{Post: &saml_pb.PostResponse{
|
||
|
|
RelayState: "notempty",
|
||
|
|
SamlResponse: "notempty",
|
||
|
|
}},
|
||
|
|
Details: &object.Details{
|
||
|
|
ChangeDate: timestamppb.Now(),
|
||
|
|
ResourceOwner: Instance.ID(),
|
||
|
|
},
|
||
|
|
},
|
||
|
|
wantErr: false,
|
||
|
|
},
|
||
|
|
{
|
||
|
|
name: "fail callback, post, already failed",
|
||
|
|
req: &saml_pb.CreateResponseRequest{
|
||
|
|
SamlRequestId: func() string {
|
||
|
|
authRequestID, err := Instance.CreateSAMLAuthRequest(spMiddlewarePost, Instance.Users[integration.UserTypeOrgOwner].ID, acsPost, gofakeit.BitcoinAddress(), saml.HTTPPostBinding)
|
||
|
|
require.NoError(t, err)
|
||
|
|
Instance.FailSAMLAuthRequest(CTX, authRequestID, saml_pb.ErrorReason_ERROR_REASON_AUTH_N_FAILED)
|
||
|
|
return authRequestID
|
||
|
|
}(),
|
||
|
|
ResponseKind: &saml_pb.CreateResponseRequest_Error{
|
||
|
|
Error: &saml_pb.AuthorizationError{
|
||
|
|
Error: saml_pb.ErrorReason_ERROR_REASON_REQUEST_DENIED,
|
||
|
|
ErrorDescription: gu.Ptr("nope"),
|
||
|
|
},
|
||
|
|
},
|
||
|
|
},
|
||
|
|
wantErr: true,
|
||
|
|
},
|
||
|
|
{
|
||
|
|
name: "fail callback, redirect",
|
||
|
|
req: &saml_pb.CreateResponseRequest{
|
||
|
|
SamlRequestId: func() string {
|
||
|
|
authRequestID, err := Instance.CreateSAMLAuthRequest(spMiddlewareRedirect, Instance.Users[integration.UserTypeOrgOwner].ID, acsPost, gofakeit.BitcoinAddress(), saml.HTTPPostBinding)
|
||
|
|
require.NoError(t, err)
|
||
|
|
return authRequestID
|
||
|
|
}(),
|
||
|
|
ResponseKind: &saml_pb.CreateResponseRequest_Error{
|
||
|
|
Error: &saml_pb.AuthorizationError{
|
||
|
|
Error: saml_pb.ErrorReason_ERROR_REASON_REQUEST_DENIED,
|
||
|
|
ErrorDescription: gu.Ptr("nope"),
|
||
|
|
},
|
||
|
|
},
|
||
|
|
},
|
||
|
|
want: &saml_pb.CreateResponseResponse{
|
||
|
|
Url: `https:\/\/` + rootURLRedirect + `\/saml\/acs\?RelayState=(.*)&SAMLResponse=(.*)`,
|
||
|
|
Binding: &saml_pb.CreateResponseResponse_Redirect{Redirect: &saml_pb.RedirectResponse{}},
|
||
|
|
Details: &object.Details{
|
||
|
|
ChangeDate: timestamppb.Now(),
|
||
|
|
ResourceOwner: Instance.ID(),
|
||
|
|
},
|
||
|
|
},
|
||
|
|
wantErr: false,
|
||
|
|
},
|
||
|
|
{
|
||
|
|
name: "callback, redirect",
|
||
|
|
req: &saml_pb.CreateResponseRequest{
|
||
|
|
SamlRequestId: func() string {
|
||
|
|
authRequestID, err := Instance.CreateSAMLAuthRequest(spMiddlewareRedirect, Instance.Users[integration.UserTypeOrgOwner].ID, acsRedirect, gofakeit.BitcoinAddress(), saml.HTTPRedirectBinding)
|
||
|
|
require.NoError(t, err)
|
||
|
|
return authRequestID
|
||
|
|
}(),
|
||
|
|
ResponseKind: &saml_pb.CreateResponseRequest_Session{
|
||
|
|
Session: &saml_pb.Session{
|
||
|
|
SessionId: sessionResp.GetSessionId(),
|
||
|
|
SessionToken: sessionResp.GetSessionToken(),
|
||
|
|
},
|
||
|
|
},
|
||
|
|
},
|
||
|
|
want: &saml_pb.CreateResponseResponse{
|
||
|
|
Url: `https:\/\/` + rootURLRedirect + `\/saml\/acs\?RelayState=(.*)&SAMLResponse=(.*)&SigAlg=(.*)&Signature=(.*)`,
|
||
|
|
Binding: &saml_pb.CreateResponseResponse_Redirect{Redirect: &saml_pb.RedirectResponse{}},
|
||
|
|
Details: &object.Details{
|
||
|
|
ChangeDate: timestamppb.Now(),
|
||
|
|
ResourceOwner: Instance.ID(),
|
||
|
|
},
|
||
|
|
},
|
||
|
|
wantErr: false,
|
||
|
|
},
|
||
|
|
{
|
||
|
|
name: "callback, post",
|
||
|
|
req: &saml_pb.CreateResponseRequest{
|
||
|
|
SamlRequestId: func() string {
|
||
|
|
authRequestID, err := Instance.CreateSAMLAuthRequest(spMiddlewarePost, Instance.Users[integration.UserTypeOrgOwner].ID, acsPost, gofakeit.BitcoinAddress(), saml.HTTPPostBinding)
|
||
|
|
require.NoError(t, err)
|
||
|
|
return authRequestID
|
||
|
|
}(),
|
||
|
|
ResponseKind: &saml_pb.CreateResponseRequest_Session{
|
||
|
|
Session: &saml_pb.Session{
|
||
|
|
SessionId: sessionResp.GetSessionId(),
|
||
|
|
SessionToken: sessionResp.GetSessionToken(),
|
||
|
|
},
|
||
|
|
},
|
||
|
|
},
|
||
|
|
want: &saml_pb.CreateResponseResponse{
|
||
|
|
Url: regexp.QuoteMeta(`https://` + rootURLPost + `/saml/acs`),
|
||
|
|
Binding: &saml_pb.CreateResponseResponse_Post{Post: &saml_pb.PostResponse{
|
||
|
|
RelayState: "notempty",
|
||
|
|
SamlResponse: "notempty",
|
||
|
|
}},
|
||
|
|
Details: &object.Details{
|
||
|
|
ChangeDate: timestamppb.Now(),
|
||
|
|
ResourceOwner: Instance.ID(),
|
||
|
|
},
|
||
|
|
},
|
||
|
|
wantErr: false,
|
||
|
|
},
|
||
|
|
{
|
||
|
|
name: "callback, post",
|
||
|
|
req: &saml_pb.CreateResponseRequest{
|
||
|
|
SamlRequestId: func() string {
|
||
|
|
authRequestID, err := Instance.CreateSAMLAuthRequest(spMiddlewarePost, Instance.Users[integration.UserTypeOrgOwner].ID, acsPost, gofakeit.BitcoinAddress(), saml.HTTPPostBinding)
|
||
|
|
require.NoError(t, err)
|
||
|
|
Instance.SuccessfulSAMLAuthRequest(CTX, Instance.Users[integration.UserTypeOrgOwner].ID, authRequestID)
|
||
|
|
return authRequestID
|
||
|
|
}(),
|
||
|
|
ResponseKind: &saml_pb.CreateResponseRequest_Session{
|
||
|
|
Session: &saml_pb.Session{
|
||
|
|
SessionId: sessionResp.GetSessionId(),
|
||
|
|
SessionToken: sessionResp.GetSessionToken(),
|
||
|
|
},
|
||
|
|
},
|
||
|
|
},
|
||
|
|
wantErr: true,
|
||
|
|
},
|
||
|
|
}
|
||
|
|
for _, tt := range tests {
|
||
|
|
t.Run(tt.name, func(t *testing.T) {
|
||
|
|
got, err := Client.CreateResponse(CTX, tt.req)
|
||
|
|
if tt.wantErr {
|
||
|
|
require.Error(t, err)
|
||
|
|
return
|
||
|
|
}
|
||
|
|
require.NoError(t, err)
|
||
|
|
integration.AssertDetails(t, tt.want, got)
|
||
|
|
if tt.want != nil {
|
||
|
|
assert.Regexp(t, regexp.MustCompile(tt.want.Url), got.GetUrl())
|
||
|
|
if tt.want.GetPost() != nil {
|
||
|
|
assert.NotEmpty(t, got.GetPost().GetRelayState())
|
||
|
|
assert.NotEmpty(t, got.GetPost().GetSamlResponse())
|
||
|
|
}
|
||
|
|
if tt.want.GetRedirect() != nil {
|
||
|
|
assert.NotNil(t, got.GetRedirect())
|
||
|
|
}
|
||
|
|
}
|
||
|
|
})
|
||
|
|
}
|
||
|
|
}
|