52 lines
1.4 KiB
Go
52 lines
1.4 KiB
Go
package server
|
|
|
|
import (
|
|
"net/http"
|
|
"net/http/httptest"
|
|
"testing"
|
|
|
|
"chat-agent-system/internal/config"
|
|
)
|
|
|
|
func TestServer_WebAssetsHostingAndFallback(t *testing.T) {
|
|
cfg, err := config.Load("")
|
|
if err != nil {
|
|
t.Fatalf("加载配置失败: %v", err)
|
|
}
|
|
cfg.HTTPPort = 8089
|
|
|
|
srv := NewServer(cfg, nil)
|
|
handler := srv.Routes()
|
|
|
|
// 1. 测试访问根路径 "/" 应返回前端 SPA index.html
|
|
reqRoot := httptest.NewRequest(http.MethodGet, "/", nil)
|
|
wRoot := httptest.NewRecorder()
|
|
handler.ServeHTTP(wRoot, reqRoot)
|
|
|
|
if wRoot.Code != http.StatusOK {
|
|
t.Errorf("expected status 200 for root SPA, got %d", wRoot.Code)
|
|
}
|
|
bodyStr := wRoot.Body.String()
|
|
if bodyStr == "" {
|
|
t.Errorf("expected non-empty SPA body")
|
|
}
|
|
|
|
// 2. 测试 SPA 客户端路由回退 (如访问 /sessions/sess-001 应返回 index.html 而不是 404)
|
|
reqClientRoute := httptest.NewRequest(http.MethodGet, "/sessions/sess-001", nil)
|
|
wClient := httptest.NewRecorder()
|
|
handler.ServeHTTP(wClient, reqClientRoute)
|
|
|
|
if wClient.Code != http.StatusOK {
|
|
t.Errorf("expected status 200 for client route fallback, got %d", wClient.Code)
|
|
}
|
|
|
|
// 3. 测试 API 路由不受静态路由影响 (GET /api/v1/chat/sessions 依然正常工作)
|
|
reqAPI := httptest.NewRequest(http.MethodGet, "/api/v1/chat/sessions", nil)
|
|
wAPI := httptest.NewRecorder()
|
|
handler.ServeHTTP(wAPI, reqAPI)
|
|
|
|
if wAPI.Code != http.StatusOK {
|
|
t.Errorf("expected status 200 for /api/v1/chat/sessions, got %d", wAPI.Code)
|
|
}
|
|
}
|