Files
90CNServer/pvf/internal/mcpserver/server_test.go
T
hanfuit 645b64fe21 DNF90 服务端与启动器完整源码(含 GM 工具功能增强)
服务端(Go):
- GM 塞背包/带属性装备发放:在线角色推送 op14 增量物品行,背包实时刷新,无需回角色选择界面
- GM 改金币/改等级:去掉在线拒绝,落库后推送 op14 钱包/op37 等级状态,在线实时刷新
- 等级端点统一切换到新版 GMSetCharacterLevel(自带在线刷新)
- 无限/锁死疲劳模式 + 燃烧疲劳:每房间消耗 2 点、打怪与通关经验双倍
- 燃烧疲劳开关时给所有在线角色推送系统提示(op12 mode 0)
- 深渊史诗掉率:GM 全局倍率 + 城镇增益药剂窗口倍率,真实接入掉落结算

启动器(Python/JS):
- 标题栏自绘、窗口 1204x717、禁止拖拽改大小、仅最小化/关闭
- GM 工具:疲劳与燃烧、深渊史诗掉率、拍卖行上架(单一定价对齐)
- 拍卖行上架前端对齐服务端单一价格语义,修复 400 报错
2026-11-04 14:31:55 +08:00

47 lines
1.8 KiB
Go

package mcpserver
import (
"context"
"encoding/json"
"net/http/httptest"
"testing"
"github.com/modelcontextprotocol/go-sdk/mcp"
)
func TestTransportOriginAndUnauthenticatedDiscovery(t *testing.T) {
h := Handler([]Tool{{Name: "pvf_test", Description: "test", ReadOnly: true, Schema: map[string]any{"type": "object", "properties": map[string]any{"value": map[string]any{"type": "string"}}, "required": []string{"value"}, "additionalProperties": false}, Call: func(_ context.Context, raw json.RawMessage) (any, error) { return map[string]bool{"ok": true}, nil }}})
for _, test := range []struct {
host, origin string
status int
}{{"evil.example", "", 403}, {"127.0.0.1:1234", "https://evil.example", 403}} {
r := httptest.NewRequest("POST", "http://"+test.host+"/mcp", nil)
r.Header.Set("Origin", test.origin)
w := httptest.NewRecorder()
h.ServeHTTP(w, r)
if w.Code != test.status {
t.Fatalf("unexpected HTTP status %d", w.Code)
}
}
server := httptest.NewServer(h)
defer server.Close()
client := mcp.NewClient(&mcp.Implementation{Name: "test", Version: "1"}, nil)
session, err := client.Connect(context.Background(), &mcp.StreamableClientTransport{Endpoint: server.URL + "/mcp"}, nil)
if err != nil {
t.Fatal(err)
}
defer session.Close()
listed, err := session.ListTools(context.Background(), nil)
if err != nil || len(listed.Tools) != 1 {
t.Fatal("tool discovery failed")
}
result, err := session.CallTool(context.Background(), &mcp.CallToolParams{Name: "pvf_test", Arguments: map[string]any{"value": "hello"}})
if err != nil || result.IsError {
t.Fatal("valid tool call failed")
}
result, err = session.CallTool(context.Background(), &mcp.CallToolParams{Name: "pvf_test", Arguments: map[string]any{"value": 5}})
if err != nil || !result.IsError {
t.Fatal("invalid schema accepted")
}
}