mirror of
https://github.com/Rain-kl/OpenFlare.git
synced 2026-09-29 05:56:38 +08:00
ryan
455e2f8be5
Public config now comes from admin as a flat visibility=1 map instead of a cross-plugin query that compared an integer column to "visible". Login and register resolve CaptchaService per request so CAP is not skipped when user applies before cap.
93 lines
2.7 KiB
Go
93 lines
2.7 KiB
Go
// Copyright 2026 Arctel.net
|
|
// SPDX-License-Identifier: Apache-2.0
|
|
|
|
package admin_test
|
|
|
|
import (
|
|
"Wavelet/core"
|
|
"Wavelet/core/contracts"
|
|
"Wavelet/plugins/domain/admin"
|
|
"context"
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
func TestAdminPluginUnit(t *testing.T) {
|
|
ctx := core.NewContext(context.Background())
|
|
p := admin.New()
|
|
assert.Equal(t, "admin", p.Name())
|
|
assert.Equal(t, "1.0.0", p.Manifest().Version)
|
|
require.NoError(t, p.Apply(ctx))
|
|
|
|
// Verify routes
|
|
routes := ctx.Router().Routes()
|
|
assert.NotEmpty(t, routes)
|
|
var hasRobots bool
|
|
for _, r := range routes {
|
|
if r.Path == "/robots.txt" && r.Method == "GET" {
|
|
hasRobots = true
|
|
break
|
|
}
|
|
}
|
|
assert.True(t, hasRobots, "admin plugin must register /robots.txt")
|
|
|
|
// Verify tasks
|
|
_, ok := ctx.Tasks().Get("logs:db_switch")
|
|
require.True(t, ok)
|
|
|
|
// Verify settings
|
|
setting, ok := ctx.Settings().Get("admin.system_cleanup_cron")
|
|
require.True(t, ok)
|
|
assert.Equal(t, "0 4 * * *", setting.Default)
|
|
|
|
provider, err := core.Inject[contracts.PublicConfigProvider](ctx)
|
|
require.NoError(t, err)
|
|
require.NotNil(t, provider)
|
|
}
|
|
|
|
func TestAdminMigrationsIncludeTaskExecutionsAndSchedules(t *testing.T) {
|
|
ctx := core.NewContext(context.Background())
|
|
p := admin.New()
|
|
require.NoError(t, p.Apply(ctx))
|
|
|
|
entry, ok := ctx.Migrations().Get("admin")
|
|
require.True(t, ok, "admin plugin must register migrations")
|
|
assert.Equal(t, "admin", entry.PluginID)
|
|
|
|
// Verify sqlite migration files include w_schedules and w_task_executions
|
|
sqliteDir, err := entry.FS.Open("migrations/sqlite/00001_initial.sql")
|
|
require.NoError(t, err)
|
|
defer sqliteDir.Close()
|
|
|
|
stat, err := sqliteDir.Stat()
|
|
require.NoError(t, err)
|
|
buf := make([]byte, stat.Size())
|
|
_, err = sqliteDir.Read(buf)
|
|
require.NoError(t, err)
|
|
content := string(buf)
|
|
|
|
assert.Contains(t, content, "CREATE TABLE IF NOT EXISTS w_task_executions")
|
|
assert.Contains(t, content, "CREATE TABLE IF NOT EXISTS w_schedules")
|
|
assert.Contains(t, content, "CREATE TABLE IF NOT EXISTS w_system_configs")
|
|
assert.Contains(t, content, "CREATE TABLE IF NOT EXISTS w_templates")
|
|
|
|
// Verify postgres migration files include w_schedules and w_task_executions
|
|
pgDir, err := entry.FS.Open("migrations/postgres/00001_initial.sql")
|
|
require.NoError(t, err)
|
|
defer pgDir.Close()
|
|
|
|
stat, err = pgDir.Stat()
|
|
require.NoError(t, err)
|
|
buf = make([]byte, stat.Size())
|
|
_, err = pgDir.Read(buf)
|
|
require.NoError(t, err)
|
|
pgContent := string(buf)
|
|
|
|
assert.Contains(t, pgContent, "CREATE TABLE IF NOT EXISTS w_task_executions")
|
|
assert.Contains(t, pgContent, "CREATE TABLE IF NOT EXISTS w_schedules")
|
|
assert.Contains(t, pgContent, "CREATE TABLE IF NOT EXISTS w_system_configs")
|
|
assert.Contains(t, pgContent, "CREATE TABLE IF NOT EXISTS w_templates")
|
|
}
|