HomeDash/admin-server/pages_test.go

package main

import (
	"net/http"
	"net/http/httptest"
	"os"
	"path/filepath"
	"strings"
	"testing"
)

// 未配置任何文案时应回落到内置默认值（不配置也照旧）。
func TestPageStrFallbacks(t *testing.T) {
	empty := map[string]any{}
	if got := pageStr(empty, "authTitle", defaultAuthTitle); got != defaultAuthTitle {
		t.Fatalf("空配置应回落默认，得到 %q", got)
	}
	info := map[string]any{"authTitle": "  内网入口  ", "heroTitle": ""}
	if got := pageStr(info, "authTitle", defaultAuthTitle); got != "内网入口" {
		t.Fatalf("应取配置值并去空白，得到 %q", got)
	}
	// 空串视为未配置
	if got := pageStr(info, "heroTitle", defaultPortalHero); got != defaultPortalHero {
		t.Fatalf("空串应回落默认，得到 %q", got)
	}
}

// 认证页字段从 pageInfo 读取，缺失时回落。
func TestAuthPageInfoFallback(t *testing.T) {
	dir := t.TempDir()
	old := confFile
	confFile = filepath.Join(dir, "conf.yml")
	defer func() { confFile = old }()
	if err := os.WriteFile(confFile, []byte("pageInfo:\n  authTitle: 我的门户\n  authLogo: AB\n"), 0o644); err != nil {
		t.Fatal(err)
	}
	title, subtitle, logo, _, _ := authPageInfo()
	if title != "我的门户" || logo != "AB" {
		t.Fatalf("未读到配置值: %q %q", title, logo)
	}
	if subtitle != defaultAuthSubtitle {
		t.Fatalf("未配置的字段应回落默认，得到 %q", subtitle)
	}
}

// 数据卷里的整页覆盖优先于内置模板。
func TestPortalOverrideWins(t *testing.T) {
	dir := t.TempDir()
	old := confFile
	confFile = filepath.Join(dir, "conf.yml")
	defer func() { confFile = old }()
	if err := os.MkdirAll(filepath.Join(dir, "portal"), 0o755); err != nil {
		t.Fatal(err)
	}
	custom := "<!doctype html><title>我的门户</title>"
	if err := os.WriteFile(filepath.Join(dir, "portal", "index.html"), []byte(custom), 0o644); err != nil {
		t.Fatal(err)
	}
	if got := string(renderPortalPage()); !strings.Contains(got, "我的门户") {
		t.Fatalf("数据卷覆盖未生效: %q", got)
	}
	// 空白覆盖文件应被忽略（回落到内置渲染）
	if err := os.WriteFile(filepath.Join(dir, "portal", "index.html"), []byte("   \n"), 0o644); err != nil {
		t.Fatal(err)
	}
	if got := string(renderPortalPage()); strings.Contains(got, "我的门户") {
		t.Fatalf("空白覆盖文件应被忽略")
	}
}

// 数据卷里的卡片片段应替换标记之间的内容（改门户链接无需重建镜像）。
func TestCardsOverrideReplacesMarkedBlock(t *testing.T) {
	dir := t.TempDir()
	old := confFile
	defer func() { confFile = old }()
	confFile = filepath.Join(dir, "conf.yml")
	if err := os.MkdirAll(filepath.Join(dir, "portal"), 0o755); err != nil {
		t.Fatal(err)
	}
	if err := os.WriteFile(filepath.Join(dir, "portal", "cards.html"), []byte("<a href=\"https://mine.example.org\">我的服务</a>"), 0o644); err != nil {
		t.Fatal(err)
	}
	page := "<div>" + cardsStartMarker + "<a href=\"https://old.example.com\">旧</a>" + cardsEndMarker + "</div>"
	got := applyCardsOverride(page)
	if !strings.Contains(got, "mine.example.org") || strings.Contains(got, "old.example.com") {
		t.Fatalf("覆盖未生效: %s", got)
	}
	// 清空后回落内置卡片
	if err := os.WriteFile(filepath.Join(dir, "portal", "cards.html"), []byte("  \n"), 0o644); err != nil {
		t.Fatal(err)
	}
	if got := applyCardsOverride(page); !strings.Contains(got, "old.example.com") {
		t.Fatalf("空白片段应回落内置卡片: %s", got)
	}
}

// /api/schema 必须列出全部可配置项与写入口（agent 靠它按图配置）。
func TestSchemaListsConfigurableItems(t *testing.T) {
	rec := httptest.NewRecorder()
	handleSchema(rec, httptest.NewRequest(http.MethodGet, "/api/schema", nil))
	if rec.Code != 200 {
		t.Fatalf("期望 200，实际 %d", rec.Code)
	}
	body := rec.Body.String()
	for _, want := range []string{"pageInfo.heroTitle", "pageInfo.icp", "pageInfo.authLogo",
		"portalCards", "gatewayRoutes", "portalAuth", "wallpaper", "/api/gateway/apply"} {
		if !strings.Contains(body, want) {
			t.Fatalf("schema 缺少 %q", want)
		}
	}
}

// 卡片接口：写入后读得到，写空串等于重置。
func TestPortalCardsAPIRoundTrip(t *testing.T) {
	dir := t.TempDir()
	old := confFile
	defer func() { confFile = old }()
	confFile = filepath.Join(dir, "conf.yml")

	post := func(html string) *httptest.ResponseRecorder {
		req := httptest.NewRequest(http.MethodPost, "/api/portal-cards", strings.NewReader(`{"html":"`+html+`"}`))
		rec := httptest.NewRecorder()
		handlePortalCards(rec, req)
		return rec
	}
	if rec := post(`<a href='https://mine.example.org'>我的服务</a>`); rec.Code != 200 {
		t.Fatalf("写入失败 %d: %s", rec.Code, rec.Body.String())
	}
	if got := string(applyCardsOverride(cardsStartMarker + "old" + cardsEndMarker)); !strings.Contains(got, "mine.example.org") {
		t.Fatalf("写入后未生效: %s", got)
	}
	if rec := post(""); rec.Code != 200 || !strings.Contains(rec.Body.String(), "reset") {
		t.Fatalf("清空应返回 reset: %d %s", rec.Code, rec.Body.String())
	}
	if got := string(applyCardsOverride(cardsStartMarker + "old" + cardsEndMarker)); !strings.Contains(got, "old") {
		t.Fatalf("清空后应回落内置: %s", got)
	}
}