mirror of
https://github.com/usememos/memos.git
synced 2024-11-17 04:07:30 +08:00
56 lines
1.5 KiB
Go
56 lines
1.5 KiB
Go
package teststore
|
|
|
|
import (
|
|
"context"
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/require"
|
|
"golang.org/x/crypto/bcrypt"
|
|
|
|
"github.com/usememos/memos/store"
|
|
)
|
|
|
|
func TestUserStore(t *testing.T) {
|
|
ctx := context.Background()
|
|
ts := NewTestingStore(ctx, t)
|
|
user, err := createTestingHostUser(ctx, ts)
|
|
require.NoError(t, err)
|
|
users, err := ts.ListUsers(ctx, &store.FindUser{})
|
|
require.NoError(t, err)
|
|
require.Equal(t, 1, len(users))
|
|
require.Equal(t, store.RoleHost, users[0].Role)
|
|
require.Equal(t, user, users[0])
|
|
userPatchNickname := "test_nickname_2"
|
|
userPatch := &store.UpdateUser{
|
|
ID: user.ID,
|
|
Nickname: &userPatchNickname,
|
|
}
|
|
user, err = ts.UpdateUser(ctx, userPatch)
|
|
require.NoError(t, err)
|
|
require.Equal(t, userPatchNickname, user.Nickname)
|
|
err = ts.DeleteUser(ctx, &store.DeleteUser{
|
|
ID: user.ID,
|
|
})
|
|
require.NoError(t, err)
|
|
users, err = ts.ListUsers(ctx, &store.FindUser{})
|
|
require.NoError(t, err)
|
|
require.Equal(t, 0, len(users))
|
|
ts.Close()
|
|
}
|
|
|
|
func createTestingHostUser(ctx context.Context, ts *store.Store) (*store.User, error) {
|
|
userCreate := &store.User{
|
|
Username: "test",
|
|
Role: store.RoleHost,
|
|
Email: "test@test.com",
|
|
Nickname: "test_nickname",
|
|
Description: "test_description",
|
|
}
|
|
passwordHash, err := bcrypt.GenerateFromPassword([]byte("test_password"), bcrypt.DefaultCost)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
userCreate.PasswordHash = string(passwordHash)
|
|
user, err := ts.CreateUser(ctx, userCreate)
|
|
return user, err
|
|
}
|