memos/test/store/user_setting_test.go

29 lines
703 B
Go
Raw Normal View History

2023-06-29 22:55:03 +08:00
package teststore
import (
"context"
"testing"
"github.com/stretchr/testify/require"
2023-09-17 22:55:13 +08:00
storepb "github.com/usememos/memos/proto/gen/store"
2023-06-29 22:55:03 +08:00
"github.com/usememos/memos/store"
)
func TestUserSettingStore(t *testing.T) {
ctx := context.Background()
ts := NewTestingStore(ctx, t)
user, err := createTestingHostUser(ctx, ts)
require.NoError(t, err)
2023-12-16 12:18:53 +08:00
_, err = ts.UpsertUserSetting(ctx, &storepb.UserSetting{
UserId: user.ID,
Key: storepb.UserSettingKey_USER_SETTING_LOCALE,
Value: &storepb.UserSetting_Locale{Locale: "en"},
2023-06-29 22:55:03 +08:00
})
require.NoError(t, err)
2023-12-16 12:18:53 +08:00
list, err := ts.ListUserSettings(ctx, &store.FindUserSetting{})
2023-06-29 22:55:03 +08:00
require.NoError(t, err)
require.Equal(t, 1, len(list))
ts.Close()
2023-06-29 22:55:03 +08:00
}