package test import ( "fmt" "testing" "github.com/authorizerdev/authorizer/server/constants" "github.com/authorizerdev/authorizer/server/crypto" "github.com/authorizerdev/authorizer/server/db" "github.com/authorizerdev/authorizer/server/graph/model" "github.com/authorizerdev/authorizer/server/memorystore" "github.com/authorizerdev/authorizer/server/resolvers" "github.com/stretchr/testify/assert" ) func enableAccessTest(t *testing.T, s TestSetup) { t.Helper() t.Run(`should enable access`, func(t *testing.T) { req, ctx := createContext(s) email := "enable_access." + s.TestInfo.Email _, err := resolvers.MagicLinkLoginResolver(ctx, model.MagicLinkLoginInput{ Email: email, }) assert.NoError(t, err) verificationRequest, err := db.Provider.GetVerificationRequestByEmail(email, constants.VerificationTypeMagicLinkLogin) verifyRes, err := resolvers.VerifyEmailResolver(ctx, model.VerifyEmailInput{ Token: verificationRequest.Token, }) assert.NoError(t, err) assert.NotNil(t, verifyRes.AccessToken) adminSecret, err := memorystore.Provider.GetStringStoreEnvVariable(constants.EnvKeyAdminSecret) assert.Nil(t, err) h, err := crypto.EncryptPassword(adminSecret) assert.Nil(t, err) req.Header.Set("Cookie", fmt.Sprintf("%s=%s", constants.AdminCookieName, h)) res, err := resolvers.RevokeAccessResolver(ctx, model.UpdateAccessInput{ UserID: verifyRes.User.ID, }) assert.NoError(t, err) assert.NotEmpty(t, res.Message) res, err = resolvers.EnableAccessResolver(ctx, model.UpdateAccessInput{ UserID: verifyRes.User.ID, }) assert.NoError(t, err) assert.NotEmpty(t, res.Message) // it should allow login with enabled access res, err = resolvers.MagicLinkLoginResolver(ctx, model.MagicLinkLoginInput{ Email: email, }) assert.Nil(t, err) assert.NotEmpty(t, res.Message) cleanData(email) }) }