mirror of
https://github.com/Rain-kl/OpenFlare.git
synced 2026-09-28 05:46:36 +08:00
ryan
30bbe965bf
Asynq func handlers now go through ProcessTask so admin execution rows leave pending. The in-process worker resolves admin type identifiers and writes the same w_task_executions table. Remove the no-op admin system_cleanup that shadowed the upload handler.
88 lines
2.6 KiB
Go
88 lines
2.6 KiB
Go
// Copyright 2026 Arctel.net
|
|
// SPDX-License-Identifier: Apache-2.0
|
|
|
|
package admin_test
|
|
|
|
import (
|
|
"Wavelet/core"
|
|
"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)
|
|
}
|
|
|
|
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")
|
|
}
|