mirror of
https://github.com/Rain-kl/OpenFlare.git
synced 2026-09-29 05:56:38 +08:00
ryan
103 lines
2.6 KiB
Go
103 lines
2.6 KiB
Go
// Copyright 2026 Arctel.net
|
|
// SPDX-License-Identifier: Apache-2.0
|
|
|
|
package cmd
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
"github.com/stretchr/testify/require"
|
|
|
|
"Wavelet/core"
|
|
"Wavelet/pkg/config"
|
|
)
|
|
|
|
func TestNewWaveletAppProfiles(t *testing.T) {
|
|
profiles := []core.Profile{
|
|
core.ProfileAPI,
|
|
core.ProfileWorker,
|
|
core.ProfileSchedule,
|
|
core.ProfileAll,
|
|
}
|
|
|
|
for _, prof := range profiles {
|
|
t.Run(string(prof), func(t *testing.T) {
|
|
app := newWaveletApp(prof)
|
|
require.NotNil(t, app)
|
|
assert.Equal(t, prof, app.Profile())
|
|
|
|
// 3 infra + (1 cache + 2 worker/cron) + 8 domain + 1 http driver = 15 plugins
|
|
plugins := app.Plugins()
|
|
assert.Len(t, plugins, 15)
|
|
|
|
// Verify standard infra plugins
|
|
_, ok := app.Plugin("database")
|
|
assert.True(t, ok, "database plugin missing")
|
|
|
|
_, ok = app.Plugin("logger")
|
|
assert.True(t, ok, "logger plugin missing")
|
|
|
|
_, ok = app.Plugin("storage")
|
|
assert.True(t, ok, "storage plugin missing")
|
|
|
|
// In zero-Redis mode (default in test)
|
|
_, ok = app.Plugin("cache_memory")
|
|
assert.True(t, ok, "cache_memory plugin missing")
|
|
|
|
_, ok = app.Plugin("driver_inproc_worker")
|
|
assert.True(t, ok, "inproc worker driver missing")
|
|
|
|
_, ok = app.Plugin("driver_inproc_cron")
|
|
assert.True(t, ok, "inproc scheduler driver missing")
|
|
|
|
// Verify domain plugins
|
|
_, ok = app.Plugin("auth")
|
|
assert.True(t, ok, "auth plugin missing")
|
|
|
|
_, ok = app.Plugin("user")
|
|
assert.True(t, ok, "user plugin missing")
|
|
|
|
_, ok = app.Plugin("message_gateway")
|
|
assert.True(t, ok, "message_gateway plugin missing")
|
|
|
|
_, ok = app.Plugin("risk_control")
|
|
assert.True(t, ok, "risk_control plugin missing")
|
|
|
|
_, ok = app.Plugin("admin")
|
|
assert.True(t, ok, "admin plugin missing")
|
|
|
|
_, ok = app.Plugin("upload")
|
|
assert.True(t, ok, "upload plugin missing")
|
|
|
|
_, ok = app.Plugin("cap")
|
|
assert.True(t, ok, "cap plugin missing")
|
|
|
|
_, ok = app.Plugin("system")
|
|
assert.True(t, ok, "system plugin missing")
|
|
|
|
// Verify driver plugins
|
|
_, ok = app.Plugin("driver_http")
|
|
assert.True(t, ok, "http driver missing")
|
|
})
|
|
}
|
|
}
|
|
|
|
func TestNewWaveletAppWithRedisEnabled(t *testing.T) {
|
|
orig := config.Config.Redis.Enabled
|
|
config.Config.Redis.Enabled = true
|
|
defer func() { config.Config.Redis.Enabled = orig }()
|
|
|
|
app := newWaveletApp(core.ProfileAll)
|
|
require.NotNil(t, app)
|
|
|
|
_, ok := app.Plugin("cache")
|
|
assert.True(t, ok, "cache plugin missing in Redis mode")
|
|
|
|
_, ok = app.Plugin("driver_asynq_worker")
|
|
assert.True(t, ok, "asynq worker driver missing in Redis mode")
|
|
|
|
_, ok = app.Plugin("driver_asynq_cron")
|
|
assert.True(t, ok, "asynq scheduler driver missing in Redis mode")
|
|
}
|