mirror of
https://github.com/zitadel/zitadel.git
synced 2025-12-23 17:18:26 +00:00
99d3d955b9
# Which Problems Are Solved
Flakiness in integration tests because of eventual consistentcy.
# How the Problems Are Solved
Split tests related to feature flags and other eventual consistent
resources.
# Additional Changes
None
# Additional Context
None
Co-authored-by: Marco A. <marco@zitadel.com>
(cherry picked from commit a8bbac37d9)
95 lines
3.2 KiB
Go
95 lines
3.2 KiB
Go
//go:build integration
|
|
|
|
package session_test
|
|
|
|
import (
|
|
"context"
|
|
"os"
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
"github.com/stretchr/testify/require"
|
|
"github.com/zitadel/logging"
|
|
|
|
"github.com/zitadel/zitadel/internal/integration"
|
|
"github.com/zitadel/zitadel/pkg/grpc/session/v2"
|
|
"github.com/zitadel/zitadel/pkg/grpc/user/v2"
|
|
)
|
|
|
|
var (
|
|
CTX context.Context
|
|
IAMOwnerCTX context.Context
|
|
UserCTX context.Context
|
|
LoginCTX context.Context
|
|
Instance *integration.Instance
|
|
Client session.SessionServiceClient
|
|
User *user.AddHumanUserResponse
|
|
)
|
|
|
|
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.SessionV2
|
|
|
|
CTX = Instance.WithAuthorization(ctx, integration.UserTypeOrgOwner)
|
|
IAMOwnerCTX = Instance.WithAuthorization(ctx, integration.UserTypeIAMOwner)
|
|
UserCTX = Instance.WithAuthorization(ctx, integration.UserTypeNoPermission)
|
|
LoginCTX = Instance.WithAuthorization(ctx, integration.UserTypeLogin)
|
|
User = createFullUser(CTX)
|
|
return m.Run()
|
|
}())
|
|
}
|
|
|
|
func createFullUser(ctx context.Context) *user.AddHumanUserResponse {
|
|
userResp := Instance.CreateHumanUser(ctx)
|
|
Instance.Client.UserV2.VerifyEmail(ctx, &user.VerifyEmailRequest{
|
|
UserId: userResp.GetUserId(),
|
|
VerificationCode: userResp.GetEmailCode(),
|
|
})
|
|
Instance.Client.UserV2.VerifyPhone(ctx, &user.VerifyPhoneRequest{
|
|
UserId: userResp.GetUserId(),
|
|
VerificationCode: userResp.GetPhoneCode(),
|
|
})
|
|
Instance.SetUserPassword(ctx, userResp.GetUserId(), integration.UserPassword, false)
|
|
Instance.RegisterUserPasskey(ctx, userResp.GetUserId())
|
|
return userResp
|
|
}
|
|
|
|
func createDeactivatedUser(ctx context.Context, t *testing.T) *user.AddHumanUserResponse {
|
|
userResp := Instance.CreateHumanUser(ctx)
|
|
_, err := Instance.Client.UserV2.DeactivateUser(ctx, &user.DeactivateUserRequest{UserId: userResp.GetUserId()})
|
|
logging.OnError(err).Fatal("deactivate human user")
|
|
|
|
retryDuration, tick := integration.WaitForAndTickWithMaxDuration(ctx, time.Minute)
|
|
require.EventuallyWithT(t, func(collect *assert.CollectT) {
|
|
result, err := Instance.Client.UserV2.GetUserByID(ctx, &user.GetUserByIDRequest{
|
|
UserId: userResp.GetUserId(),
|
|
})
|
|
assert.NoError(collect, err)
|
|
assert.Equal(t, user.UserState_USER_STATE_INACTIVE, result.GetUser().GetState())
|
|
}, retryDuration, tick, "user not locked in time")
|
|
|
|
return userResp
|
|
}
|
|
|
|
func createLockedUser(ctx context.Context, t *testing.T) *user.AddHumanUserResponse {
|
|
userResp := Instance.CreateHumanUser(ctx)
|
|
_, err := Instance.Client.UserV2.LockUser(ctx, &user.LockUserRequest{UserId: userResp.GetUserId()})
|
|
logging.OnError(err).Fatal("lock human user")
|
|
|
|
retryDuration, tick := integration.WaitForAndTickWithMaxDuration(ctx, time.Minute)
|
|
require.EventuallyWithT(t, func(collect *assert.CollectT) {
|
|
result, err := Instance.Client.UserV2.GetUserByID(ctx, &user.GetUserByIDRequest{
|
|
UserId: userResp.GetUserId(),
|
|
})
|
|
assert.NoError(collect, err)
|
|
assert.Equal(t, user.UserState_USER_STATE_LOCKED, result.GetUser().GetState())
|
|
}, retryDuration, tick, "user not locked in time")
|
|
|
|
return userResp
|
|
}
|