// Copyright 2026 Arctel.net
// SPDX-License-Identifier: Apache-2.0

package cmd

import (
	"Wavelet/core"
	"Wavelet/pkg/config"
	"testing"

	"github.com/stretchr/testify/assert"
	"github.com/stretchr/testify/require"
)

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")
}