test(chat): add download-media CLI integration coverage
This commit is contained in:
修雨 committed
1 parent
bf33ab622f
commit
b1f4a5d62a
1 file changed
+187
@@ -163,6 +163,193 @@ func TestMockMCPSmoke_CLIRoutesSerializedArgumentsAndPrintsJSON(t *testing.T) {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
func TestMockMCPSmoke_ChatDownloadMediaJSONCLI(t *testing.T) {
|
||||||
|
type mediaRequest struct {
|
||||||
|
method string
|
||||||
|
path string
|
||||||
|
query string
|
||||||
|
header string
|
||||||
|
}
|
||||||
|
|
||||||
|
mediaPayload := []byte("synthetic media payload")
|
||||||
|
var mediaRequestsMu sync.Mutex
|
||||||
|
var mediaRequests []mediaRequest
|
||||||
|
mediaServer := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
|
||||||
|
mediaRequestsMu.Lock()
|
||||||
|
mediaRequests = append(mediaRequests, mediaRequest{
|
||||||
|
method: r.Method,
|
||||||
|
path: r.URL.Path,
|
||||||
|
query: r.URL.RawQuery,
|
||||||
|
header: r.Header.Get("X-Download-Fixture"),
|
||||||
|
})
|
||||||
|
mediaRequestsMu.Unlock()
|
||||||
|
w.WriteHeader(http.StatusOK)
|
||||||
|
_, _ = w.Write(mediaPayload)
|
||||||
|
}))
|
||||||
|
defer mediaServer.Close()
|
||||||
|
|
||||||
|
downloadURL := mediaServer.URL + "/photo.jpg?fixture=one&part=two"
|
||||||
|
downloadInfo, err := json.Marshal(map[string]any{
|
||||||
|
"resourceUrl": downloadURL,
|
||||||
|
"headers": map[string]string{
|
||||||
|
"X-Download-Fixture": "synthetic",
|
||||||
|
},
|
||||||
|
})
|
||||||
|
if err != nil {
|
||||||
|
t.Fatal(err)
|
||||||
|
}
|
||||||
|
|
||||||
|
var requestsMu sync.Mutex
|
||||||
|
var requests []recordedToolCall
|
||||||
|
mcpServer := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
|
||||||
|
call := recordedToolCall{
|
||||||
|
path: r.URL.Path,
|
||||||
|
method: r.Method,
|
||||||
|
authorization: r.Header.Get("Authorization"),
|
||||||
|
}
|
||||||
|
|
||||||
|
var envelope struct {
|
||||||
|
JSONRPC string `json:"jsonrpc"`
|
||||||
|
ID int `json:"id"`
|
||||||
|
Method string `json:"method"`
|
||||||
|
Params struct {
|
||||||
|
Name string `json:"name"`
|
||||||
|
Arguments map[string]any `json:"arguments"`
|
||||||
|
} `json:"params"`
|
||||||
|
}
|
||||||
|
if err := json.NewDecoder(r.Body).Decode(&envelope); err != nil {
|
||||||
|
call.err = err
|
||||||
|
} else {
|
||||||
|
call.jsonrpc = envelope.JSONRPC
|
||||||
|
call.tool = envelope.Params.Name
|
||||||
|
call.arguments = envelope.Params.Arguments
|
||||||
|
if envelope.Method != "tools/call" {
|
||||||
|
call.err = fmt.Errorf("JSON-RPC method = %q, want tools/call", envelope.Method)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
requestsMu.Lock()
|
||||||
|
requests = append(requests, call)
|
||||||
|
requestsMu.Unlock()
|
||||||
|
|
||||||
|
w.Header().Set("Content-Type", "application/json")
|
||||||
|
_ = json.NewEncoder(w).Encode(map[string]any{
|
||||||
|
"jsonrpc": "2.0",
|
||||||
|
"id": envelope.ID,
|
||||||
|
"result": map[string]any{
|
||||||
|
"content": []map[string]any{{
|
||||||
|
"type": "text",
|
||||||
|
"text": string(downloadInfo),
|
||||||
|
}},
|
||||||
|
},
|
||||||
|
})
|
||||||
|
}))
|
||||||
|
defer mcpServer.Close()
|
||||||
|
|
||||||
|
outputDir := t.TempDir()
|
||||||
|
env := isolatedCLIEnv(t, map[string]string{
|
||||||
|
"DINGTALK_IM_MCP_URL": mcpServer.URL + "/mcp/im",
|
||||||
|
})
|
||||||
|
args := []string{
|
||||||
|
"--token", "ci-smoke-token",
|
||||||
|
"--format", "json",
|
||||||
|
"chat", "message", "download-media",
|
||||||
|
"--type", "mediaId",
|
||||||
|
"--resource-id", "resource-001",
|
||||||
|
"--message-id", "message-001",
|
||||||
|
"--open-conversation-id", "conversation-001",
|
||||||
|
"--output", outputDir,
|
||||||
|
}
|
||||||
|
stdout, stderr, err := runCLI(t, env, args...)
|
||||||
|
if err != nil {
|
||||||
|
t.Fatalf("dws chat message download-media failed: %v\nstdout:\n%s\nstderr:\n%s", err, stdout, stderr)
|
||||||
|
}
|
||||||
|
|
||||||
|
requestsMu.Lock()
|
||||||
|
recorded := append([]recordedToolCall(nil), requests...)
|
||||||
|
requestsMu.Unlock()
|
||||||
|
if len(recorded) != 1 {
|
||||||
|
t.Fatalf("local fake MCP server received %d requests, want exactly one tools/call: %#v", len(recorded), recorded)
|
||||||
|
}
|
||||||
|
call := recorded[0]
|
||||||
|
if call.err != nil {
|
||||||
|
t.Fatal(call.err)
|
||||||
|
}
|
||||||
|
if call.path != "/mcp/im" || call.method != http.MethodPost || call.jsonrpc != "2.0" {
|
||||||
|
t.Fatalf("MCP request = path %q method %q jsonrpc %q", call.path, call.method, call.jsonrpc)
|
||||||
|
}
|
||||||
|
if call.authorization != "Bearer ci-smoke-token" {
|
||||||
|
t.Fatalf("Authorization = %q, want synthetic smoke token", call.authorization)
|
||||||
|
}
|
||||||
|
if call.tool != "get_resource_download_url" {
|
||||||
|
t.Fatalf("tool = %q, want get_resource_download_url", call.tool)
|
||||||
|
}
|
||||||
|
wantArgs := map[string]any{
|
||||||
|
"resourceType": "mediaId",
|
||||||
|
"resourceId": "resource-001",
|
||||||
|
"openMessageId": "message-001",
|
||||||
|
"openConversationId": "conversation-001",
|
||||||
|
}
|
||||||
|
if !reflect.DeepEqual(call.arguments, wantArgs) {
|
||||||
|
t.Fatalf("arguments = %#v, want %#v", call.arguments, wantArgs)
|
||||||
|
}
|
||||||
|
|
||||||
|
mediaRequestsMu.Lock()
|
||||||
|
recordedMedia := append([]mediaRequest(nil), mediaRequests...)
|
||||||
|
mediaRequestsMu.Unlock()
|
||||||
|
if len(recordedMedia) != 1 {
|
||||||
|
t.Fatalf("media server received %d requests, want exactly one: %#v", len(recordedMedia), recordedMedia)
|
||||||
|
}
|
||||||
|
gotMedia := recordedMedia[0]
|
||||||
|
if gotMedia.method != http.MethodGet || gotMedia.path != "/photo.jpg" {
|
||||||
|
t.Fatalf("media request = method %q path %q", gotMedia.method, gotMedia.path)
|
||||||
|
}
|
||||||
|
if gotMedia.query != "fixture=one&part=two" {
|
||||||
|
t.Fatalf("media query = %q, want fixture=one&part=two", gotMedia.query)
|
||||||
|
}
|
||||||
|
if gotMedia.header != "synthetic" {
|
||||||
|
t.Fatalf("media header = %q, want synthetic", gotMedia.header)
|
||||||
|
}
|
||||||
|
|
||||||
|
wantOutput := filepath.Join(outputDir, "photo.jpg")
|
||||||
|
gotPayload, err := os.ReadFile(wantOutput)
|
||||||
|
if err != nil {
|
||||||
|
t.Fatalf("read downloaded file: %v", err)
|
||||||
|
}
|
||||||
|
if !bytes.Equal(gotPayload, mediaPayload) {
|
||||||
|
t.Fatalf("downloaded payload = %q, want %q", gotPayload, mediaPayload)
|
||||||
|
}
|
||||||
|
|
||||||
|
var result map[string]any
|
||||||
|
if err := json.Unmarshal([]byte(stdout), &result); err != nil {
|
||||||
|
t.Fatalf("CLI returned non-JSON stdout: %v\nstdout:\n%s\nstderr:\n%s", err, stdout, stderr)
|
||||||
|
}
|
||||||
|
if len(result) != 3 || result["success"] != true || result["downloadUrl"] != downloadURL || result["output"] != wantOutput {
|
||||||
|
t.Fatalf("CLI result = %#v, want exact success/downloadUrl/output contract", result)
|
||||||
|
}
|
||||||
|
if strings.Contains(stdout, "[INFO]") {
|
||||||
|
t.Fatalf("JSON stdout contains progress text: %s", stdout)
|
||||||
|
}
|
||||||
|
if !strings.Contains(stdout, "?fixture=one&part=two") {
|
||||||
|
t.Fatalf("downloadUrl was escaped or changed: %s", stdout)
|
||||||
|
}
|
||||||
|
|
||||||
|
publicResult := map[string]any{
|
||||||
|
"success": true,
|
||||||
|
"downloadUrl": "http://127.0.0.1:<fixture-port>/photo.jpg?fixture=one&part=two",
|
||||||
|
"output": "./downloads/photo.jpg",
|
||||||
|
}
|
||||||
|
var publicOutput bytes.Buffer
|
||||||
|
publicEncoder := json.NewEncoder(&publicOutput)
|
||||||
|
publicEncoder.SetEscapeHTML(false)
|
||||||
|
publicEncoder.SetIndent("", " ")
|
||||||
|
if err := publicEncoder.Encode(publicResult); err != nil {
|
||||||
|
t.Fatal(err)
|
||||||
|
}
|
||||||
|
t.Log("CLI (public-safe): dws chat message download-media --type mediaId --resource-id resource-001 --message-id message-001 --open-conversation-id conversation-001 --output ./downloads/ --format json")
|
||||||
|
t.Logf("stdout (public-safe):\n%s", strings.TrimSpace(publicOutput.String()))
|
||||||
|
t.Logf("downloaded file: photo.jpg (%d bytes, content verified)", len(gotPayload))
|
||||||
|
}
|
||||||
|
|
||||||
func isolatedCLIEnv(t *testing.T, extra map[string]string) []string {
|
func isolatedCLIEnv(t *testing.T, extra map[string]string) []string {
|
||||||
t.Helper()
|
t.Helper()
|
||||||
|
|
||||||
|
|||||||
Reference in new issue
Block a user