2026-05-08 07:27:06 +08:00
|
|
|
package service
|
|
|
|
|
|
|
|
|
|
import (
|
2026-06-05 11:31:33 +08:00
|
|
|
"apigo.cc/go/watch"
|
2026-05-08 07:27:06 +08:00
|
|
|
"github.com/gorilla/websocket"
|
|
|
|
|
"net/http/httptest"
|
2026-06-05 11:31:33 +08:00
|
|
|
"os"
|
|
|
|
|
"path/filepath"
|
2026-05-08 07:27:06 +08:00
|
|
|
"strings"
|
|
|
|
|
"testing"
|
|
|
|
|
)
|
|
|
|
|
|
|
|
|
|
func TestWebSocketService(t *testing.T) {
|
|
|
|
|
// 注册 WebSocket 服务
|
2026-05-09 16:39:20 +08:00
|
|
|
Host("*").WebSocket("/ws", func(conn *websocket.Conn) {
|
2026-05-08 22:41:01 +08:00
|
|
|
for {
|
2026-05-09 16:39:20 +08:00
|
|
|
var msg map[string]any
|
2026-05-08 22:41:01 +08:00
|
|
|
if err := conn.ReadJSON(&msg); err != nil {
|
|
|
|
|
break
|
|
|
|
|
}
|
2026-05-09 16:39:20 +08:00
|
|
|
_ = conn.WriteJSON(map[string]any{"reply": msg["msg"]})
|
2026-05-08 22:41:01 +08:00
|
|
|
}
|
2026-05-09 16:39:20 +08:00
|
|
|
}).Auth(0).Memo("test websocket")
|
2026-05-08 07:27:06 +08:00
|
|
|
|
|
|
|
|
// 启动测试服务器
|
2026-06-04 18:16:46 +08:00
|
|
|
server := httptest.NewServer(&RouteHandler{ws: DefaultServer})
|
2026-05-08 07:27:06 +08:00
|
|
|
defer server.Close()
|
|
|
|
|
|
|
|
|
|
// 建立连接
|
|
|
|
|
wsURL := "ws" + strings.TrimPrefix(server.URL, "http") + "/ws"
|
|
|
|
|
conn, _, err := websocket.DefaultDialer.Dial(wsURL, nil)
|
|
|
|
|
if err != nil {
|
|
|
|
|
t.Fatalf("Dial failed: %v", err)
|
|
|
|
|
}
|
|
|
|
|
defer conn.Close()
|
|
|
|
|
|
|
|
|
|
// 发送消息
|
2026-05-09 16:39:20 +08:00
|
|
|
msg := map[string]any{"action": "echo", "msg": "hello"}
|
2026-05-08 07:27:06 +08:00
|
|
|
if err := conn.WriteJSON(msg); err != nil {
|
|
|
|
|
t.Fatalf("WriteJSON failed: %v", err)
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
// 接收响应
|
2026-05-09 16:39:20 +08:00
|
|
|
var reply map[string]any
|
2026-05-08 07:27:06 +08:00
|
|
|
if err := conn.ReadJSON(&reply); err != nil {
|
|
|
|
|
t.Fatalf("ReadJSON failed: %v", err)
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
if reply["reply"] != "hello" {
|
|
|
|
|
t.Errorf("Reply mismatch: %v", reply)
|
|
|
|
|
}
|
|
|
|
|
}
|
2026-06-05 11:31:33 +08:00
|
|
|
|
|
|
|
|
func TestEnableWebDev(t *testing.T) {
|
|
|
|
|
// 1. 初始化 EnableWebDev
|
|
|
|
|
EnableWebDev(watch.Config{
|
|
|
|
|
Paths: []string{"."},
|
|
|
|
|
})
|
|
|
|
|
|
|
|
|
|
// 2. 准备一个真实的静态 HTML 文件
|
|
|
|
|
staticDir := "test_static"
|
|
|
|
|
_ = os.MkdirAll(staticDir, 0755)
|
|
|
|
|
htmlFile := filepath.Join(staticDir, "index.html")
|
|
|
|
|
_ = os.WriteFile(htmlFile, []byte("<html><head></head><body>Static Content</body></html>"), 0644)
|
|
|
|
|
defer os.RemoveAll(staticDir)
|
|
|
|
|
|
|
|
|
|
// 注册静态服务
|
|
|
|
|
Static("/static/", staticDir)
|
|
|
|
|
|
|
|
|
|
handler := &RouteHandler{ws: DefaultServer}
|
|
|
|
|
|
|
|
|
|
// 3. 测试静态文件注入
|
|
|
|
|
req := httptest.NewRequest("GET", "/static/index.html", nil)
|
|
|
|
|
w := httptest.NewRecorder()
|
|
|
|
|
handler.ServeHTTP(w, req)
|
|
|
|
|
|
|
|
|
|
body := w.Body.String()
|
|
|
|
|
if !strings.Contains(body, "let _watchWS = null") {
|
|
|
|
|
t.Errorf("Static HTML injection failed, code not found in body: %s", body)
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
// 4. 测试普通服务注入
|
|
|
|
|
Register("GET", "/test-dev", func() string {
|
|
|
|
|
return "<html><head></head><body>Hello</body></html>"
|
|
|
|
|
})
|
|
|
|
|
|
|
|
|
|
req2 := httptest.NewRequest("GET", "/test-dev", nil)
|
|
|
|
|
w2 := httptest.NewRecorder()
|
|
|
|
|
handler.ServeHTTP(w2, req2)
|
|
|
|
|
|
|
|
|
|
body2 := w2.Body.String()
|
|
|
|
|
if !strings.Contains(body2, "let _watchWS = null") {
|
|
|
|
|
t.Errorf("Dynamic HTML injection failed")
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
// 5. 验证非 HTML 不注入
|
|
|
|
|
Register("GET", "/test-json", func() map[string]string {
|
|
|
|
|
return map[string]string{"foo": "bar"}
|
|
|
|
|
})
|
|
|
|
|
req3 := httptest.NewRequest("GET", "/test-json", nil)
|
|
|
|
|
w3 := httptest.NewRecorder()
|
|
|
|
|
handler.ServeHTTP(w3, req3)
|
|
|
|
|
|
|
|
|
|
body3 := w3.Body.String()
|
|
|
|
|
if strings.Contains(body3, "let _watchWS = null") {
|
|
|
|
|
t.Errorf("JSON should not be injected")
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|
