Files
MyGoNavi/internal/webserver/server_test.go
Syngnat 7c8e8d8dd3 feat(sql-audit): 新增 SQL 审计中心并完善事务追踪
- 新增脱敏审计存储、筛选、保留策略、完整性校验与 JSON/CSV 导出
- 覆盖查询编辑器、事务、导入同步、对象操作、AI/MCP 与 Web 运行时入口
- 完善事务完整语句日志、Windows 快捷键映射及 SQL 分析布局
- 补充多语言、健康状态、数据目录迁移和回归测试
2026-07-12 15:24:27 +08:00

112 lines
3.4 KiB
Go

package webserver
import (
"encoding/json"
"reflect"
"strings"
"testing"
)
type webserverTestReceiver struct{}
func (webserverTestReceiver) Echo(value string) (map[string]any, error) {
return map[string]any{"value": value}, nil
}
func (webserverTestReceiver) Sum(left int, right int) int {
return left + right
}
func TestInjectRuntimeBridgeAddsScriptOnce(t *testing.T) {
indexHTML := "<html><head><title>GoNavi</title></head><body></body></html>"
injected := injectRuntimeBridge(indexHTML)
if !strings.Contains(injected, internalRoutePrefix+"/web-runtime.js") {
t.Fatalf("expected injected HTML to contain runtime bridge script, got: %s", injected)
}
reinjected := injectRuntimeBridge(injected)
if strings.Count(reinjected, internalRoutePrefix+"/web-runtime.js") != 1 {
t.Fatalf("expected runtime bridge script to be injected once, got: %s", reinjected)
}
}
func TestMethodInvokerInvokeDecodesArgumentsAndReturnsResult(t *testing.T) {
invoker := &methodInvoker{
targets: map[string]reflect.Value{
"test.receiver": reflect.ValueOf(webserverTestReceiver{}),
},
}
rawLeft, _ := json.Marshal(2)
rawRight, _ := json.Marshal(5)
result, err := invoker.Invoke(invokeRequest{
Namespace: "test",
Receiver: "receiver",
Method: "Sum",
Args: []json.RawMessage{rawLeft, rawRight},
})
if err != nil {
t.Fatalf("expected invoke success, got error: %v", err)
}
if result != 7 {
t.Fatalf("expected sum result 7, got %#v", result)
}
}
func TestMethodInvokerInvokeSupportsStructuredReturnValues(t *testing.T) {
invoker := &methodInvoker{
targets: map[string]reflect.Value{
"test.receiver": reflect.ValueOf(webserverTestReceiver{}),
},
}
rawValue, _ := json.Marshal("hello")
result, err := invoker.Invoke(invokeRequest{
Namespace: "test",
Receiver: "receiver",
Method: "Echo",
Args: []json.RawMessage{rawValue},
})
if err != nil {
t.Fatalf("expected invoke success, got error: %v", err)
}
payload, ok := result.(map[string]any)
if !ok {
t.Fatalf("expected structured result map, got %#v", result)
}
if payload["value"] != "hello" {
t.Fatalf("expected echoed value hello, got %#v", payload["value"])
}
}
func TestMethodInvokerRejectsDesktopOnlyAppMethodsBeforeReflection(t *testing.T) {
invoker := &methodInvoker{
targets: map[string]reflect.Value{
"app.app": reflect.ValueOf(webserverTestReceiver{}),
},
}
for _, method := range []string{
"Shutdown", "ExportSQLAuditFile", "OpenSQLFile", "ExecuteSQLFile", "ReadSQLFile",
"PreviewImportFile", "ImportDataWithProgress", "GetDataRootDirectoryInfo",
"ApplyDataRootDirectory", "OpenDataRootDirectory",
} {
_, err := invoker.Invoke(invokeRequest{Namespace: "app", Receiver: "app", Method: method})
if err == nil || !strings.Contains(err.Error(), "unavailable in web runtime") {
t.Fatalf("desktop-only method %s error = %v, want web runtime rejection", method, err)
}
}
}
func TestSQLAuditHeavyInvokeIncludesExportAndIntegrityVerification(t *testing.T) {
for _, method := range []string{"BuildSQLAuditExport", "VerifySQLAuditIntegrity"} {
if !isSQLAuditHeavyInvoke(invokeRequest{Namespace: "app", Receiver: "app", Method: method}) {
t.Fatalf("%s must share the SQL audit heavy-operation semaphore", method)
}
}
if isSQLAuditHeavyInvoke(invokeRequest{Namespace: "app", Receiver: "app", Method: "GetSQLAuditEvents"}) {
t.Fatal("ordinary paged audit reads must not use the heavy-operation semaphore")
}
}