forked from mirrors/homebox
164 lines
3.3 KiB
Go
164 lines
3.3 KiB
Go
package repo
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"testing"
|
|
|
|
"github.com/hay-kot/content/backend/ent"
|
|
"github.com/hay-kot/content/backend/internal/types"
|
|
"github.com/hay-kot/content/backend/pkgs/faker"
|
|
"github.com/stretchr/testify/assert"
|
|
)
|
|
|
|
func UserFactory() types.UserCreate {
|
|
f := faker.NewFaker()
|
|
|
|
return types.UserCreate{
|
|
Name: f.RandomString(10),
|
|
Email: f.RandomEmail(),
|
|
Password: f.RandomString(10),
|
|
IsSuperuser: f.RandomBool(),
|
|
GroupID: testGroup.ID,
|
|
}
|
|
}
|
|
|
|
func Test_EntUserRepo_GetOneEmail(t *testing.T) {
|
|
assert := assert.New(t)
|
|
user := UserFactory()
|
|
ctx := context.Background()
|
|
|
|
_, err := testRepos.Users.Create(ctx, user)
|
|
assert.NoError(err)
|
|
|
|
foundUser, err := testRepos.Users.GetOneEmail(ctx, user.Email)
|
|
|
|
assert.NotNil(foundUser)
|
|
assert.Nil(err)
|
|
assert.Equal(user.Email, foundUser.Email)
|
|
assert.Equal(user.Name, foundUser.Name)
|
|
|
|
// Cleanup
|
|
err = testRepos.Users.DeleteAll(ctx)
|
|
assert.NoError(err)
|
|
}
|
|
|
|
func Test_EntUserRepo_GetOneId(t *testing.T) {
|
|
assert := assert.New(t)
|
|
user := UserFactory()
|
|
ctx := context.Background()
|
|
|
|
userOut, _ := testRepos.Users.Create(ctx, user)
|
|
foundUser, err := testRepos.Users.GetOneId(ctx, userOut.ID)
|
|
|
|
assert.NotNil(foundUser)
|
|
assert.Nil(err)
|
|
assert.Equal(user.Email, foundUser.Email)
|
|
assert.Equal(user.Name, foundUser.Name)
|
|
|
|
// Cleanup
|
|
err = testRepos.Users.DeleteAll(ctx)
|
|
assert.NoError(err)
|
|
}
|
|
|
|
func Test_EntUserRepo_GetAll(t *testing.T) {
|
|
// Setup
|
|
toCreate := []types.UserCreate{
|
|
UserFactory(),
|
|
UserFactory(),
|
|
UserFactory(),
|
|
UserFactory(),
|
|
}
|
|
|
|
ctx := context.Background()
|
|
|
|
created := []*ent.User{}
|
|
|
|
for _, usr := range toCreate {
|
|
usrOut, _ := testRepos.Users.Create(ctx, usr)
|
|
created = append(created, usrOut)
|
|
}
|
|
|
|
// Validate
|
|
allUsers, err := testRepos.Users.GetAll(ctx)
|
|
|
|
assert.NoError(t, err)
|
|
assert.Equal(t, len(created), len(allUsers))
|
|
|
|
for _, usr := range created {
|
|
fmt.Printf("%+v\n", usr)
|
|
for _, usr2 := range allUsers {
|
|
if usr.ID == usr2.ID {
|
|
assert.Equal(t, usr.Email, usr2.Email)
|
|
|
|
// Check groups are loaded
|
|
assert.NotNil(t, usr2.Edges.Group)
|
|
}
|
|
}
|
|
}
|
|
|
|
for _, usr := range created {
|
|
_ = testRepos.Users.Delete(ctx, usr.ID)
|
|
}
|
|
|
|
// Cleanup
|
|
err = testRepos.Users.DeleteAll(ctx)
|
|
assert.NoError(t, err)
|
|
}
|
|
|
|
func Test_EntUserRepo_Update(t *testing.T) {
|
|
t.Skip()
|
|
}
|
|
|
|
func Test_EntUserRepo_Delete(t *testing.T) {
|
|
// Create 10 Users
|
|
for i := 0; i < 10; i++ {
|
|
user := UserFactory()
|
|
ctx := context.Background()
|
|
_, _ = testRepos.Users.Create(ctx, user)
|
|
}
|
|
|
|
// Delete all
|
|
ctx := context.Background()
|
|
allUsers, _ := testRepos.Users.GetAll(ctx)
|
|
|
|
assert.Greater(t, len(allUsers), 0)
|
|
err := testRepos.Users.DeleteAll(ctx)
|
|
assert.NoError(t, err)
|
|
|
|
allUsers, _ = testRepos.Users.GetAll(ctx)
|
|
assert.Equal(t, len(allUsers), 0)
|
|
|
|
}
|
|
|
|
func Test_EntUserRepo_GetSuperusers(t *testing.T) {
|
|
// Create 10 Users
|
|
superuser := 0
|
|
users := 0
|
|
|
|
for i := 0; i < 10; i++ {
|
|
user := UserFactory()
|
|
ctx := context.Background()
|
|
_, _ = testRepos.Users.Create(ctx, user)
|
|
|
|
if user.IsSuperuser {
|
|
superuser++
|
|
} else {
|
|
users++
|
|
}
|
|
}
|
|
|
|
// Delete all
|
|
ctx := context.Background()
|
|
|
|
superUsers, err := testRepos.Users.GetSuperusers(ctx)
|
|
assert.NoError(t, err)
|
|
|
|
for _, usr := range superUsers {
|
|
assert.True(t, usr.IsSuperuser)
|
|
}
|
|
|
|
// Cleanup
|
|
err = testRepos.Users.DeleteAll(ctx)
|
|
assert.NoError(t, err)
|
|
}
|