1
0
Fork 0
forked from mirrors/homebox
homebox/backend/internal/data/repo/repo_users_test.go

176 lines
3.4 KiB
Go
Raw Normal View History

2022-08-30 02:30:36 +00:00
package repo
import (
"context"
"fmt"
"testing"
"github.com/stretchr/testify/assert"
)
func userFactory() UserCreate {
return UserCreate{
Name: fk.Str(10),
Email: fk.Email(),
Password: fk.Str(10),
IsSuperuser: fk.Bool(),
GroupID: tGroup.ID,
2022-08-30 02:30:36 +00:00
}
}
func TestUserRepo_GetOneEmail(t *testing.T) {
2022-08-30 02:30:36 +00:00
assert := assert.New(t)
user := userFactory()
2022-08-30 02:30:36 +00:00
ctx := context.Background()
_, err := tRepos.Users.Create(ctx, user)
2022-09-03 09:52:05 +00:00
assert.NoError(err)
2022-08-30 02:30:36 +00:00
foundUser, err := tRepos.Users.GetOneEmail(ctx, user.Email)
2022-08-30 02:30:36 +00:00
assert.NotNil(foundUser)
assert.Nil(err)
assert.Equal(user.Email, foundUser.Email)
assert.Equal(user.Name, foundUser.Name)
// Cleanup
err = tRepos.Users.DeleteAll(ctx)
2022-09-03 09:52:05 +00:00
assert.NoError(err)
2022-08-30 02:30:36 +00:00
}
func TestUserRepo_GetOneId(t *testing.T) {
2022-08-30 02:30:36 +00:00
assert := assert.New(t)
user := userFactory()
2022-08-30 02:30:36 +00:00
ctx := context.Background()
userOut, _ := tRepos.Users.Create(ctx, user)
foundUser, err := tRepos.Users.GetOneId(ctx, userOut.ID)
2022-08-30 02:30:36 +00:00
assert.NotNil(foundUser)
assert.Nil(err)
assert.Equal(user.Email, foundUser.Email)
assert.Equal(user.Name, foundUser.Name)
// Cleanup
err = tRepos.Users.DeleteAll(ctx)
2022-09-03 09:52:05 +00:00
assert.NoError(err)
2022-08-30 02:30:36 +00:00
}
func TestUserRepo_GetAll(t *testing.T) {
2022-08-30 02:30:36 +00:00
// Setup
toCreate := []UserCreate{
userFactory(),
userFactory(),
userFactory(),
userFactory(),
2022-08-30 02:30:36 +00:00
}
ctx := context.Background()
created := []UserOut{}
2022-08-30 02:30:36 +00:00
for _, usr := range toCreate {
usrOut, _ := tRepos.Users.Create(ctx, usr)
2022-08-30 02:30:36 +00:00
created = append(created, usrOut)
}
// Validate
allUsers, err := tRepos.Users.GetAll(ctx)
2022-08-30 02:30:36 +00:00
2022-08-31 03:21:18 +00:00
assert.NoError(t, err)
2022-08-30 02:30:36 +00:00
assert.Equal(t, len(created), len(allUsers))
for _, usr := range created {
fmt.Printf("%+v\n", usr)
2022-08-31 03:21:18 +00:00
for _, usr2 := range allUsers {
if usr.ID == usr2.ID {
assert.Equal(t, usr.Email, usr2.Email)
// Check groups are loaded
assert.NotNil(t, usr2.GroupID)
2022-08-31 03:21:18 +00:00
}
}
2022-08-30 02:30:36 +00:00
}
for _, usr := range created {
_ = tRepos.Users.Delete(ctx, usr.ID)
2022-08-30 02:30:36 +00:00
}
// Cleanup
err = tRepos.Users.DeleteAll(ctx)
2022-09-03 09:52:05 +00:00
assert.NoError(t, err)
2022-08-30 02:30:36 +00:00
}
func TestUserRepo_Update(t *testing.T) {
user, err := tRepos.Users.Create(context.Background(), userFactory())
assert.NoError(t, err)
updateData := UserUpdate{
Name: fk.Str(10),
Email: fk.Email(),
}
// Update
err = tRepos.Users.Update(context.Background(), user.ID, updateData)
assert.NoError(t, err)
// Validate
updated, err := tRepos.Users.GetOneId(context.Background(), user.ID)
assert.NoError(t, err)
assert.NotEqual(t, user.Name, updated.Name)
assert.NotEqual(t, user.Email, updated.Email)
2022-08-30 02:30:36 +00:00
}
func TestUserRepo_Delete(t *testing.T) {
2022-08-30 02:30:36 +00:00
// Create 10 Users
for i := 0; i < 10; i++ {
user := userFactory()
2022-08-30 02:30:36 +00:00
ctx := context.Background()
_, _ = tRepos.Users.Create(ctx, user)
2022-08-30 02:30:36 +00:00
}
// Delete all
ctx := context.Background()
allUsers, _ := tRepos.Users.GetAll(ctx)
2022-08-30 02:30:36 +00:00
assert.Greater(t, len(allUsers), 0)
err := tRepos.Users.DeleteAll(ctx)
2022-09-03 09:52:05 +00:00
assert.NoError(t, err)
2022-08-30 02:30:36 +00:00
allUsers, _ = tRepos.Users.GetAll(ctx)
2022-08-30 02:30:36 +00:00
assert.Equal(t, len(allUsers), 0)
}
func TestUserRepo_GetSuperusers(t *testing.T) {
2022-08-30 02:30:36 +00:00
// Create 10 Users
superuser := 0
users := 0
for i := 0; i < 10; i++ {
user := userFactory()
2022-08-30 02:30:36 +00:00
ctx := context.Background()
_, _ = tRepos.Users.Create(ctx, user)
2022-08-30 02:30:36 +00:00
if user.IsSuperuser {
superuser++
} else {
users++
}
}
// Delete all
ctx := context.Background()
superUsers, err := tRepos.Users.GetSuperusers(ctx)
2022-08-30 02:30:36 +00:00
assert.NoError(t, err)
for _, usr := range superUsers {
assert.True(t, usr.IsSuperuser)
}
// Cleanup
err = tRepos.Users.DeleteAll(ctx)
2022-09-03 09:52:05 +00:00
assert.NoError(t, err)
2022-08-30 02:30:36 +00:00
}