mirror of
https://github.com/grafana/grafana.git
synced 2025-09-25 00:05:39 +08:00
4937f0daab
* sqlstore: Run tests as integration tests * Truncate database instead of re-creating it on each test * Fix test description See https://github.com/grafana/grafana/pull/12129 * Fix lint issues * Fix postgres dialect after review suggestion * Rename and document functions after review suggestion * Add periods * Fix auto-increment value for mysql dialect Co-authored-by: Emil Tullstedt <emil.tullstedt@grafana.com>
86 lines
2.5 KiB
Go
86 lines
2.5 KiB
Go
// +build integration
|
|
|
|
package sqlstore
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"github.com/grafana/grafana/pkg/models"
|
|
. "github.com/smartystreets/goconvey/convey"
|
|
)
|
|
|
|
func TestTempUserCommandsAndQueries(t *testing.T) {
|
|
Convey("Testing Temp User commands & queries", t, func() {
|
|
InitTestDB(t)
|
|
|
|
Convey("Given saved api key", func() {
|
|
cmd := models.CreateTempUserCommand{
|
|
OrgId: 2256,
|
|
Name: "hello",
|
|
Code: "asd",
|
|
Email: "e@as.co",
|
|
Status: models.TmpUserInvitePending,
|
|
}
|
|
err := CreateTempUser(&cmd)
|
|
So(err, ShouldBeNil)
|
|
|
|
Convey("Should be able to get temp users by org id", func() {
|
|
query := models.GetTempUsersQuery{OrgId: 2256, Status: models.TmpUserInvitePending}
|
|
err = GetTempUsersQuery(&query)
|
|
|
|
So(err, ShouldBeNil)
|
|
So(len(query.Result), ShouldEqual, 1)
|
|
})
|
|
|
|
Convey("Should be able to get temp users by email", func() {
|
|
query := models.GetTempUsersQuery{Email: "e@as.co", Status: models.TmpUserInvitePending}
|
|
err = GetTempUsersQuery(&query)
|
|
|
|
So(err, ShouldBeNil)
|
|
So(len(query.Result), ShouldEqual, 1)
|
|
})
|
|
|
|
Convey("Should be able to get temp users by code", func() {
|
|
query := models.GetTempUserByCodeQuery{Code: "asd"}
|
|
err = GetTempUserByCode(&query)
|
|
|
|
So(err, ShouldBeNil)
|
|
So(query.Result.Name, ShouldEqual, "hello")
|
|
})
|
|
|
|
Convey("Should be able update status", func() {
|
|
cmd2 := models.UpdateTempUserStatusCommand{Code: "asd", Status: models.TmpUserRevoked}
|
|
err := UpdateTempUserStatus(&cmd2)
|
|
So(err, ShouldBeNil)
|
|
})
|
|
|
|
Convey("Should be able update email sent and email sent on", func() {
|
|
cmd2 := models.UpdateTempUserWithEmailSentCommand{Code: cmd.Result.Code}
|
|
err := UpdateTempUserWithEmailSent(&cmd2)
|
|
So(err, ShouldBeNil)
|
|
|
|
query := models.GetTempUsersQuery{OrgId: 2256, Status: models.TmpUserInvitePending}
|
|
err = GetTempUsersQuery(&query)
|
|
|
|
So(err, ShouldBeNil)
|
|
So(query.Result[0].EmailSent, ShouldBeTrue)
|
|
So(query.Result[0].EmailSentOn.UTC(), ShouldHappenOnOrAfter, query.Result[0].Created.UTC())
|
|
})
|
|
|
|
Convey("Should be able expire temp user", func() {
|
|
cmd2 := models.ExpireTempUsersCommand{OlderThan: timeNow()}
|
|
err := ExpireOldUserInvites(&cmd2)
|
|
So(err, ShouldBeNil)
|
|
So(cmd2.NumExpired, ShouldEqual, 1)
|
|
|
|
Convey("Should do nothing when no temp users to expire", func() {
|
|
cmd2 = models.ExpireTempUsersCommand{OlderThan: timeNow()}
|
|
err := ExpireOldUserInvites(&cmd2)
|
|
So(err, ShouldBeNil)
|
|
So(cmd2.NumExpired, ShouldEqual, 0)
|
|
})
|
|
})
|
|
})
|
|
})
|
|
}
|