mirror of
https://github.com/grafana/grafana.git
synced 2025-02-11 08:05:43 -06:00
c63ebc887b
* Chore: Run integration tests without grabpl * Add new step for integration tests in lib.star * Remove old integration test step from lib.star * Change drone signature * Fix: Edit starlark integration step to not affect enterprise * Remove all build tags & rename starlark integration test step * Resync .drone.yml with .drone.star * Fix lint errors * Fix lint errors * Fix lint errors * Fix more lint errors * Fix another lint error * Rename integration test step * Fix last lint error * Recomment enterprise step * Remove comment from Makefile Co-authored-by: Ida Furjesova <ida.furjesova@grafana.com>
93 lines
2.8 KiB
Go
93 lines
2.8 KiB
Go
package sqlstore
|
|
|
|
import (
|
|
"context"
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/grafana/grafana/pkg/models"
|
|
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
func TestIntegrationTempUserCommandsAndQueries(t *testing.T) {
|
|
ss := InitTestDB(t)
|
|
cmd := models.CreateTempUserCommand{
|
|
OrgId: 2256,
|
|
Name: "hello",
|
|
Code: "asd",
|
|
Email: "e@as.co",
|
|
Status: models.TmpUserInvitePending,
|
|
}
|
|
setup := func(t *testing.T) {
|
|
InitTestDB(t)
|
|
err := ss.CreateTempUser(context.Background(), &cmd)
|
|
require.Nil(t, err)
|
|
}
|
|
|
|
t.Run("Should be able to get temp users by org id", func(t *testing.T) {
|
|
setup(t)
|
|
query := models.GetTempUsersQuery{OrgId: 2256, Status: models.TmpUserInvitePending}
|
|
err := ss.GetTempUsersQuery(context.Background(), &query)
|
|
|
|
require.Nil(t, err)
|
|
require.Equal(t, 1, len(query.Result))
|
|
})
|
|
|
|
t.Run("Should be able to get temp users by email", func(t *testing.T) {
|
|
setup(t)
|
|
query := models.GetTempUsersQuery{Email: "e@as.co", Status: models.TmpUserInvitePending}
|
|
err := ss.GetTempUsersQuery(context.Background(), &query)
|
|
|
|
require.Nil(t, err)
|
|
require.Equal(t, 1, len(query.Result))
|
|
})
|
|
|
|
t.Run("Should be able to get temp users by code", func(t *testing.T) {
|
|
setup(t)
|
|
query := models.GetTempUserByCodeQuery{Code: "asd"}
|
|
err := ss.GetTempUserByCode(context.Background(), &query)
|
|
|
|
require.Nil(t, err)
|
|
require.Equal(t, "hello", query.Result.Name)
|
|
})
|
|
|
|
t.Run("Should be able update status", func(t *testing.T) {
|
|
setup(t)
|
|
cmd2 := models.UpdateTempUserStatusCommand{Code: "asd", Status: models.TmpUserRevoked}
|
|
err := ss.UpdateTempUserStatus(context.Background(), &cmd2)
|
|
require.Nil(t, err)
|
|
})
|
|
|
|
t.Run("Should be able update email sent and email sent on", func(t *testing.T) {
|
|
setup(t)
|
|
cmd2 := models.UpdateTempUserWithEmailSentCommand{Code: cmd.Result.Code}
|
|
err := ss.UpdateTempUserWithEmailSent(context.Background(), &cmd2)
|
|
require.Nil(t, err)
|
|
|
|
query := models.GetTempUsersQuery{OrgId: 2256, Status: models.TmpUserInvitePending}
|
|
err = ss.GetTempUsersQuery(context.Background(), &query)
|
|
|
|
require.Nil(t, err)
|
|
require.True(t, query.Result[0].EmailSent)
|
|
require.False(t, query.Result[0].EmailSentOn.UTC().Before(query.Result[0].Created.UTC()))
|
|
})
|
|
|
|
t.Run("Should be able expire temp user", func(t *testing.T) {
|
|
setup(t)
|
|
createdAt := time.Unix(cmd.Result.Created, 0)
|
|
cmd2 := models.ExpireTempUsersCommand{OlderThan: createdAt.Add(1 * time.Second)}
|
|
err := ss.ExpireOldUserInvites(context.Background(), &cmd2)
|
|
require.Nil(t, err)
|
|
require.Equal(t, int64(1), cmd2.NumExpired)
|
|
|
|
t.Run("Should do nothing when no temp users to expire", func(t *testing.T) {
|
|
createdAt := time.Unix(cmd.Result.Created, 0)
|
|
cmd2 := models.ExpireTempUsersCommand{OlderThan: createdAt.Add(1 * time.Second)}
|
|
err := ss.ExpireOldUserInvites(context.Background(), &cmd2)
|
|
require.Nil(t, err)
|
|
require.Equal(t, int64(0), cmd2.NumExpired)
|
|
})
|
|
})
|
|
}
|