first commit

store/test/instance_setting_test.go

@@ -0,0 +1,300 @@
+package test
+
+import (
+	"context"
+	"testing"
+
+	"github.com/stretchr/testify/require"
+
+	storepb "github.com/usememos/memos/proto/gen/store"
+	"github.com/usememos/memos/store"
+)
+
+func TestInstanceSettingV1Store(t *testing.T) {
+	t.Parallel()
+	ctx := context.Background()
+	ts := NewTestingStore(ctx, t)
+	instanceSetting, err := ts.UpsertInstanceSetting(ctx, &storepb.InstanceSetting{
+		Key: storepb.InstanceSettingKey_GENERAL,
+		Value: &storepb.InstanceSetting_GeneralSetting{
+			GeneralSetting: &storepb.InstanceGeneralSetting{
+				AdditionalScript: "",
+			},
+		},
+	})
+	require.NoError(t, err)
+	setting, err := ts.GetInstanceSetting(ctx, &store.FindInstanceSetting{
+		Name: storepb.InstanceSettingKey_GENERAL.String(),
+	})
+	require.NoError(t, err)
+	require.Equal(t, instanceSetting, setting)
+	ts.Close()
+}
+
+func TestInstanceSettingGetNonExistent(t *testing.T) {
+	t.Parallel()
+	ctx := context.Background()
+	ts := NewTestingStore(ctx, t)
+
+	// Get non-existent setting
+	setting, err := ts.GetInstanceSetting(ctx, &store.FindInstanceSetting{
+		Name: storepb.InstanceSettingKey_STORAGE.String(),
+	})
+	require.NoError(t, err)
+	require.Nil(t, setting)
+
+	ts.Close()
+}
+
+func TestInstanceSettingUpsertUpdate(t *testing.T) {
+	t.Parallel()
+	ctx := context.Background()
+	ts := NewTestingStore(ctx, t)
+
+	// Create setting
+	_, err := ts.UpsertInstanceSetting(ctx, &storepb.InstanceSetting{
+		Key: storepb.InstanceSettingKey_GENERAL,
+		Value: &storepb.InstanceSetting_GeneralSetting{
+			GeneralSetting: &storepb.InstanceGeneralSetting{
+				AdditionalScript: "console.log('v1')",
+			},
+		},
+	})
+	require.NoError(t, err)
+
+	// Update setting
+	_, err = ts.UpsertInstanceSetting(ctx, &storepb.InstanceSetting{
+		Key: storepb.InstanceSettingKey_GENERAL,
+		Value: &storepb.InstanceSetting_GeneralSetting{
+			GeneralSetting: &storepb.InstanceGeneralSetting{
+				AdditionalScript: "console.log('v2')",
+			},
+		},
+	})
+	require.NoError(t, err)
+
+	// Verify update
+	setting, err := ts.GetInstanceSetting(ctx, &store.FindInstanceSetting{
+		Name: storepb.InstanceSettingKey_GENERAL.String(),
+	})
+	require.NoError(t, err)
+	require.Equal(t, "console.log('v2')", setting.GetGeneralSetting().AdditionalScript)
+
+	// Verify only one setting exists
+	list, err := ts.ListInstanceSettings(ctx, &store.FindInstanceSetting{
+		Name: storepb.InstanceSettingKey_GENERAL.String(),
+	})
+	require.NoError(t, err)
+	require.Equal(t, 1, len(list))
+
+	ts.Close()
+}
+
+func TestInstanceSettingBasicSetting(t *testing.T) {
+	t.Parallel()
+	ctx := context.Background()
+	ts := NewTestingStore(ctx, t)
+
+	// Get default basic setting (should return empty defaults)
+	basicSetting, err := ts.GetInstanceBasicSetting(ctx)
+	require.NoError(t, err)
+	require.NotNil(t, basicSetting)
+
+	// Set basic setting
+	_, err = ts.UpsertInstanceSetting(ctx, &storepb.InstanceSetting{
+		Key: storepb.InstanceSettingKey_BASIC,
+		Value: &storepb.InstanceSetting_BasicSetting{
+			BasicSetting: &storepb.InstanceBasicSetting{
+				SecretKey: "my-secret-key",
+			},
+		},
+	})
+	require.NoError(t, err)
+
+	// Verify
+	basicSetting, err = ts.GetInstanceBasicSetting(ctx)
+	require.NoError(t, err)
+	require.Equal(t, "my-secret-key", basicSetting.SecretKey)
+
+	ts.Close()
+}
+
+func TestInstanceSettingGeneralSetting(t *testing.T) {
+	t.Parallel()
+	ctx := context.Background()
+	ts := NewTestingStore(ctx, t)
+
+	// Get default general setting
+	generalSetting, err := ts.GetInstanceGeneralSetting(ctx)
+	require.NoError(t, err)
+	require.NotNil(t, generalSetting)
+
+	// Set general setting
+	_, err = ts.UpsertInstanceSetting(ctx, &storepb.InstanceSetting{
+		Key: storepb.InstanceSettingKey_GENERAL,
+		Value: &storepb.InstanceSetting_GeneralSetting{
+			GeneralSetting: &storepb.InstanceGeneralSetting{
+				AdditionalScript: "console.log('test')",
+				AdditionalStyle:  "body { color: red; }",
+			},
+		},
+	})
+	require.NoError(t, err)
+
+	// Verify
+	generalSetting, err = ts.GetInstanceGeneralSetting(ctx)
+	require.NoError(t, err)
+	require.Equal(t, "console.log('test')", generalSetting.AdditionalScript)
+	require.Equal(t, "body { color: red; }", generalSetting.AdditionalStyle)
+
+	ts.Close()
+}
+
+func TestInstanceSettingMemoRelatedSetting(t *testing.T) {
+	t.Parallel()
+	ctx := context.Background()
+	ts := NewTestingStore(ctx, t)
+
+	// Get default memo related setting (should have defaults)
+	memoSetting, err := ts.GetInstanceMemoRelatedSetting(ctx)
+	require.NoError(t, err)
+	require.NotNil(t, memoSetting)
+	require.GreaterOrEqual(t, memoSetting.ContentLengthLimit, int32(store.DefaultContentLengthLimit))
+	require.NotEmpty(t, memoSetting.Reactions)
+
+	// Set custom memo related setting
+	customReactions := []string{"👍", "👎", "🚀"}
+	_, err = ts.UpsertInstanceSetting(ctx, &storepb.InstanceSetting{
+		Key: storepb.InstanceSettingKey_MEMO_RELATED,
+		Value: &storepb.InstanceSetting_MemoRelatedSetting{
+			MemoRelatedSetting: &storepb.InstanceMemoRelatedSetting{
+				ContentLengthLimit: 16384,
+				Reactions:          customReactions,
+			},
+		},
+	})
+	require.NoError(t, err)
+
+	// Verify
+	memoSetting, err = ts.GetInstanceMemoRelatedSetting(ctx)
+	require.NoError(t, err)
+	require.Equal(t, int32(16384), memoSetting.ContentLengthLimit)
+	require.Equal(t, customReactions, memoSetting.Reactions)
+
+	ts.Close()
+}
+
+func TestInstanceSettingStorageSetting(t *testing.T) {
+	t.Parallel()
+	ctx := context.Background()
+	ts := NewTestingStore(ctx, t)
+
+	// Get default storage setting (should have defaults)
+	storageSetting, err := ts.GetInstanceStorageSetting(ctx)
+	require.NoError(t, err)
+	require.NotNil(t, storageSetting)
+	require.Equal(t, storepb.InstanceStorageSetting_LOCAL, storageSetting.StorageType)
+	require.Equal(t, int64(30), storageSetting.UploadSizeLimitMb)
+	require.Equal(t, "assets/{timestamp}_{filename}", storageSetting.FilepathTemplate)
+
+	// Set custom storage setting
+	_, err = ts.UpsertInstanceSetting(ctx, &storepb.InstanceSetting{
+		Key: storepb.InstanceSettingKey_STORAGE,
+		Value: &storepb.InstanceSetting_StorageSetting{
+			StorageSetting: &storepb.InstanceStorageSetting{
+				StorageType:       storepb.InstanceStorageSetting_LOCAL,
+				UploadSizeLimitMb: 100,
+				FilepathTemplate:  "uploads/{date}/{filename}",
+			},
+		},
+	})
+	require.NoError(t, err)
+
+	// Verify
+	storageSetting, err = ts.GetInstanceStorageSetting(ctx)
+	require.NoError(t, err)
+	require.Equal(t, storepb.InstanceStorageSetting_LOCAL, storageSetting.StorageType)
+	require.Equal(t, int64(100), storageSetting.UploadSizeLimitMb)
+	require.Equal(t, "uploads/{date}/{filename}", storageSetting.FilepathTemplate)
+
+	ts.Close()
+}
+
+func TestInstanceSettingListAll(t *testing.T) {
+	t.Parallel()
+	ctx := context.Background()
+	ts := NewTestingStore(ctx, t)
+
+	// Count initial settings
+	initialList, err := ts.ListInstanceSettings(ctx, &store.FindInstanceSetting{})
+	require.NoError(t, err)
+	initialCount := len(initialList)
+
+	// Create multiple settings
+	_, err = ts.UpsertInstanceSetting(ctx, &storepb.InstanceSetting{
+		Key: storepb.InstanceSettingKey_GENERAL,
+		Value: &storepb.InstanceSetting_GeneralSetting{
+			GeneralSetting: &storepb.InstanceGeneralSetting{},
+		},
+	})
+	require.NoError(t, err)
+
+	_, err = ts.UpsertInstanceSetting(ctx, &storepb.InstanceSetting{
+		Key: storepb.InstanceSettingKey_STORAGE,
+		Value: &storepb.InstanceSetting_StorageSetting{
+			StorageSetting: &storepb.InstanceStorageSetting{},
+		},
+	})
+	require.NoError(t, err)
+
+	// List all - should have 2 more than initial
+	list, err := ts.ListInstanceSettings(ctx, &store.FindInstanceSetting{})
+	require.NoError(t, err)
+	require.Equal(t, initialCount+2, len(list))
+
+	ts.Close()
+}
+
+func TestInstanceSettingEdgeCases(t *testing.T) {
+	t.Parallel()
+	ctx := context.Background()
+	ts := NewTestingStore(ctx, t)
+
+	// Case 1: General Setting with special characters and Unicode
+	specialScript := `<script>alert("你好"); var x = 'test\'s';</script>`
+	specialStyle := `body { font-family: "Noto Sans SC", sans-serif; content: "\u2764"; }`
+	_, err := ts.UpsertInstanceSetting(ctx, &storepb.InstanceSetting{
+		Key: storepb.InstanceSettingKey_GENERAL,
+		Value: &storepb.InstanceSetting_GeneralSetting{
+			GeneralSetting: &storepb.InstanceGeneralSetting{
+				AdditionalScript: specialScript,
+				AdditionalStyle:  specialStyle,
+			},
+		},
+	})
+	require.NoError(t, err)
+
+	generalSetting, err := ts.GetInstanceGeneralSetting(ctx)
+	require.NoError(t, err)
+	require.Equal(t, specialScript, generalSetting.AdditionalScript)
+	require.Equal(t, specialStyle, generalSetting.AdditionalStyle)
+
+	// Case 2: Memo Related Setting with Unicode reactions
+	unicodeReactions := []string{"🐱", "🐶", "🦊", "🦄"}
+	_, err = ts.UpsertInstanceSetting(ctx, &storepb.InstanceSetting{
+		Key: storepb.InstanceSettingKey_MEMO_RELATED,
+		Value: &storepb.InstanceSetting_MemoRelatedSetting{
+			MemoRelatedSetting: &storepb.InstanceMemoRelatedSetting{
+				ContentLengthLimit: 1000,
+				Reactions:          unicodeReactions,
+			},
+		},
+	})
+	require.NoError(t, err)
+
+	memoSetting, err := ts.GetInstanceMemoRelatedSetting(ctx)
+	require.NoError(t, err)
+	require.Equal(t, unicodeReactions, memoSetting.Reactions)
+
+	ts.Close()
+}