mirror of
https://github.com/zitadel/zitadel.git
synced 2025-01-10 22:53:42 +00:00
c3b97a91a2
# Which Problems Are Solved It is currently not possible to use SAML with the Session API. # How the Problems Are Solved Add SAML service, to get and resolve SAML requests. Add SAML session and SAML request aggregate, which can be linked to the Session to get back a SAMLResponse from the API directly. # Additional Changes Update of dependency zitadel/saml to provide all functionality for handling of SAML requests and responses. # Additional Context Closes #6053 --------- Co-authored-by: Livio Spring <livio.a@gmail.com>
128 lines
3.0 KiB
Go
128 lines
3.0 KiB
Go
package query
|
|
|
|
import (
|
|
"database/sql"
|
|
"database/sql/driver"
|
|
_ "embed"
|
|
"fmt"
|
|
"regexp"
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
"github.com/stretchr/testify/require"
|
|
|
|
"github.com/zitadel/zitadel/internal/api/authz"
|
|
"github.com/zitadel/zitadel/internal/database"
|
|
"github.com/zitadel/zitadel/internal/query/projection"
|
|
"github.com/zitadel/zitadel/internal/zerrors"
|
|
)
|
|
|
|
func TestQueries_SamlRequestByID(t *testing.T) {
|
|
expQuery := regexp.QuoteMeta(fmt.Sprintf(
|
|
samlRequestByIDQuery,
|
|
asOfSystemTime,
|
|
))
|
|
|
|
cols := []string{
|
|
projection.SamlRequestColumnID,
|
|
projection.SamlRequestColumnCreationDate,
|
|
projection.SamlRequestColumnLoginClient,
|
|
projection.SamlRequestColumnIssuer,
|
|
projection.SamlRequestColumnACS,
|
|
projection.SamlRequestColumnRelayState,
|
|
projection.SamlRequestColumnBinding,
|
|
}
|
|
type args struct {
|
|
shouldTriggerBulk bool
|
|
id string
|
|
checkLoginClient bool
|
|
}
|
|
tests := []struct {
|
|
name string
|
|
args args
|
|
expect sqlExpectation
|
|
want *SamlRequest
|
|
wantErr error
|
|
}{
|
|
{
|
|
name: "success, all values",
|
|
args: args{
|
|
shouldTriggerBulk: false,
|
|
id: "123",
|
|
checkLoginClient: true,
|
|
},
|
|
expect: mockQuery(expQuery, cols, []driver.Value{
|
|
"id",
|
|
testNow,
|
|
"loginClient",
|
|
"issuer",
|
|
"acs",
|
|
"relayState",
|
|
"binding",
|
|
}, "123", "instanceID"),
|
|
want: &SamlRequest{
|
|
ID: "id",
|
|
CreationDate: testNow,
|
|
LoginClient: "loginClient",
|
|
Issuer: "issuer",
|
|
ACS: "acs",
|
|
RelayState: "relayState",
|
|
Binding: "binding",
|
|
},
|
|
},
|
|
{
|
|
name: "no rows",
|
|
args: args{
|
|
shouldTriggerBulk: false,
|
|
id: "123",
|
|
},
|
|
expect: mockQueryScanErr(expQuery, cols, nil, "123", "instanceID"),
|
|
wantErr: zerrors.ThrowNotFound(sql.ErrNoRows, "QUERY-Thee9", "Errors.SamlRequest.NotExisting"),
|
|
},
|
|
{
|
|
name: "query error",
|
|
args: args{
|
|
shouldTriggerBulk: false,
|
|
id: "123",
|
|
},
|
|
expect: mockQueryErr(expQuery, sql.ErrConnDone, "123", "instanceID"),
|
|
wantErr: zerrors.ThrowInternal(sql.ErrConnDone, "QUERY-Ou8ue", "Errors.Internal"),
|
|
},
|
|
{
|
|
name: "wrong login client",
|
|
args: args{
|
|
shouldTriggerBulk: false,
|
|
id: "123",
|
|
checkLoginClient: true,
|
|
},
|
|
expect: mockQuery(expQuery, cols, []driver.Value{
|
|
"id",
|
|
testNow,
|
|
"wrongLoginClient",
|
|
"issuer",
|
|
"acs",
|
|
"relayState",
|
|
"binding",
|
|
}, "123", "instanceID"),
|
|
wantErr: zerrors.ThrowPermissionDeniedf(nil, "OIDCv2-aL0ag", "Errors.SamlRequest.WrongLoginClient"),
|
|
},
|
|
}
|
|
for _, tt := range tests {
|
|
t.Run(tt.name, func(t *testing.T) {
|
|
execMock(t, tt.expect, func(db *sql.DB) {
|
|
q := &Queries{
|
|
client: &database.DB{
|
|
DB: db,
|
|
Database: &prepareDB{},
|
|
},
|
|
}
|
|
ctx := authz.NewMockContext("instanceID", "orgID", "loginClient")
|
|
|
|
got, err := q.SamlRequestByID(ctx, tt.args.shouldTriggerBulk, tt.args.id, tt.args.checkLoginClient)
|
|
require.ErrorIs(t, err, tt.wantErr)
|
|
assert.Equal(t, tt.want, got)
|
|
})
|
|
})
|
|
}
|
|
}
|