migrate

Wavelet/internal/apps/openflare/option/logics_test.go

@@ -0,0 +1,119 @@
+// Copyright 2026 Arctel.net
+// SPDX-License-Identifier: Apache-2.0
+
+package option
+
+import (
+	"context"
+	"testing"
+
+	"github.com/Rain-kl/Wavelet/internal/db"
+	"github.com/Rain-kl/Wavelet/internal/model"
+	"github.com/glebarez/sqlite"
+	"github.com/stretchr/testify/assert"
+	"github.com/stretchr/testify/require"
+	"gorm.io/gorm"
+)
+
+func setupOptionTestDB(t *testing.T) func() {
+	t.Helper()
+
+	sqliteDB, err := gorm.Open(sqlite.Open(":memory:"), &gorm.Config{
+		DisableForeignKeyConstraintWhenMigrating: true,
+	})
+	require.NoError(t, err)
+	require.NoError(t, sqliteDB.AutoMigrate(&model.OpenFlareOption{}))
+
+	db.SetDB(sqliteDB)
+	ResetInitializationForTest()
+
+	return func() {
+		db.SetDB(nil)
+		ResetInitializationForTest()
+	}
+}
+
+func TestListOptionsFiltersSecretKeys(t *testing.T) {
+	cleanup := setupOptionTestDB(t)
+	defer cleanup()
+	ctx := context.Background()
+
+	require.NoError(t, model.UpdateOpenFlareOptions(ctx, []model.OpenFlareOption{
+		{Key: "SystemName", Value: "TestFlare"},
+		{Key: "SMTPToken", Value: "secret-token"},
+		{Key: "GitHubClientSecret", Value: "secret-id"},
+	}))
+
+	options, err := listOptions(ctx)
+	require.NoError(t, err)
+
+	keys := make(map[string]string, len(options))
+	for _, option := range options {
+		keys[option.Key] = option.Value
+	}
+
+	assert.Equal(t, "TestFlare", keys["SystemName"])
+	assert.NotContains(t, keys, "SMTPToken")
+	assert.NotContains(t, keys, "GitHubClientSecret")
+}
+
+func TestUpdateOptionHotReloadsOptionMap(t *testing.T) {
+	cleanup := setupOptionTestDB(t)
+	defer cleanup()
+	ctx := context.Background()
+
+	err := updateOption(ctx, model.OpenFlareOption{
+		Key:   "SystemName",
+		Value: "HotReloaded",
+	})
+	require.NoError(t, err)
+
+	assert.Equal(t, "HotReloaded", model.OptionValue("SystemName"))
+	assert.Equal(t, "HotReloaded", model.SystemName)
+}
+
+func TestGetNoticeAndAbout(t *testing.T) {
+	cleanup := setupOptionTestDB(t)
+	defer cleanup()
+	ctx := context.Background()
+
+	require.NoError(t, updateOption(ctx, model.OpenFlareOption{Key: "Notice", Value: "hello"}))
+	require.NoError(t, updateOption(ctx, model.OpenFlareOption{Key: "About", Value: "about-us"}))
+
+	notice, err := getNotice(ctx)
+	require.NoError(t, err)
+	assert.Equal(t, "hello", notice)
+
+	about, err := getAbout(ctx)
+	require.NoError(t, err)
+	assert.Equal(t, "about-us", about)
+}
+
+func TestLookupGeoIPDisabledProvider(t *testing.T) {
+	cleanup := setupOptionTestDB(t)
+	defer cleanup()
+	ctx := context.Background()
+
+	view, err := lookupGeoIP(ctx, "disabled", "8.8.8.8")
+	require.NoError(t, err)
+	assert.Equal(t, "disabled", view.Provider)
+	assert.Equal(t, "8.8.8.8", view.IP)
+}
+
+func TestCleanupDatabaseObservabilityStub(t *testing.T) {
+	cleanup := setupOptionTestDB(t)
+	defer cleanup()
+	ctx := context.Background()
+
+	retention := 7
+	result, err := cleanupDatabaseObservability(ctx, databaseCleanupInput{
+		Target:        "node_access_logs",
+		RetentionDays: &retention,
+	})
+	require.NoError(t, err)
+	assert.Equal(t, "node_access_logs", result.Target)
+	assert.Equal(t, int64(0), result.DeletedCount)
+	assert.False(t, result.DeleteAll)
+	require.NotNil(t, result.RetentionDays)
+	assert.Equal(t, 7, *result.RetentionDays)
+}