user_setting_test.go 821 B

12345678910111213141516171819202122232425262728293031323334
  1. package teststore
  2. import (
  3. "context"
  4. "testing"
  5. "github.com/stretchr/testify/require"
  6. "github.com/usememos/memos/store"
  7. )
  8. func TestUserSettingStore(t *testing.T) {
  9. ctx := context.Background()
  10. ts := NewTestingStore(ctx, t)
  11. user, err := createTestingHostUser(ctx, ts)
  12. require.NoError(t, err)
  13. testSetting, err := ts.UpsertUserSetting(ctx, &store.UserSetting{
  14. UserID: user.ID,
  15. Key: "test_key",
  16. Value: "test_value",
  17. })
  18. require.NoError(t, err)
  19. localeSetting, err := ts.UpsertUserSetting(ctx, &store.UserSetting{
  20. UserID: user.ID,
  21. Key: "locale",
  22. Value: "zh",
  23. })
  24. require.NoError(t, err)
  25. list, err := ts.ListUserSettings(ctx, &store.FindUserSetting{})
  26. require.NoError(t, err)
  27. require.Equal(t, 2, len(list))
  28. require.Equal(t, testSetting, list[0])
  29. require.Equal(t, localeSetting, list[1])
  30. }