84 lines
2.2 KiB
Go
84 lines
2.2 KiB
Go
package main
|
||
|
||
import (
|
||
"bytes"
|
||
"fmt"
|
||
"strings"
|
||
"testing"
|
||
"time"
|
||
|
||
"github.com/gofiber/fiber/v3"
|
||
|
||
"github.com/any-hub/any-hub/internal/hubmodule"
|
||
"github.com/any-hub/any-hub/internal/proxy/hooks"
|
||
"github.com/any-hub/any-hub/internal/server"
|
||
)
|
||
|
||
func TestParseCLIFlagsPriority(t *testing.T) {
|
||
t.Setenv("ANY_HUB_CONFIG", "/tmp/env.toml")
|
||
|
||
opts, err := parseCLIFlags([]string{})
|
||
if err != nil {
|
||
t.Fatalf("解析失败: %v", err)
|
||
}
|
||
if opts.configPath != "/tmp/env.toml" {
|
||
t.Fatalf("应优先使用环境变量，得到 %s", opts.configPath)
|
||
}
|
||
|
||
opts, err = parseCLIFlags([]string{"--config", "/tmp/flag.toml"})
|
||
if err != nil {
|
||
t.Fatalf("解析失败: %v", err)
|
||
}
|
||
if opts.configPath != "/tmp/flag.toml" {
|
||
t.Fatalf("flag 应高于环境变量，得到 %s", opts.configPath)
|
||
}
|
||
}
|
||
|
||
func TestRunCheckConfigSuccess(t *testing.T) {
|
||
useBufferWriters(t)
|
||
code := run(cliOptions{configPath: configFixture(t, "valid.toml"), checkOnly: true})
|
||
if code != 0 {
|
||
t.Fatalf("期望退出码 0，得到 %d", code)
|
||
}
|
||
}
|
||
|
||
func TestRunCheckConfigFailure(t *testing.T) {
|
||
useBufferWriters(t)
|
||
code := run(cliOptions{configPath: configFixture(t, "missing.toml"), checkOnly: true})
|
||
if code == 0 {
|
||
t.Fatalf("无效配置应返回非零退出码")
|
||
}
|
||
}
|
||
|
||
func TestRunVersionOutput(t *testing.T) {
|
||
useBufferWriters(t)
|
||
code := run(cliOptions{showVersion: true})
|
||
if code != 0 {
|
||
t.Fatalf("version 模式应成功退出，得到 %d", code)
|
||
}
|
||
if !strings.Contains(stdOut.(*bytes.Buffer).String(), "any-hub") {
|
||
t.Fatalf("version 输出应包含 any-hub 标识")
|
||
}
|
||
}
|
||
|
||
func TestRegisterModuleHandlersFailsWithoutHook(t *testing.T) {
|
||
useBufferWriters(t)
|
||
key := fmt.Sprintf("missing-hook-%d", time.Now().UnixNano())
|
||
hubmodule.MustRegister(hubmodule.ModuleMetadata{
|
||
Key: key,
|
||
Description: "test module without hooks",
|
||
MigrationState: hubmodule.MigrationStateBeta,
|
||
SupportedProtocols: []string{
|
||
"custom",
|
||
},
|
||
})
|
||
defer hooks.MustRegister(key, hooks.Hooks{})
|
||
|
||
err := registerModuleHandlers(server.ProxyHandlerFunc(func(ctx fiber.Ctx, route *server.HubRoute) error {
|
||
return nil
|
||
}))
|
||
if err == nil || !strings.Contains(err.Error(), "missing hook") {
|
||
t.Fatalf("expected missing hook error, got %v", err)
|
||
}
|
||
}
|