dingtalk-workspace-cli/internal/shortcut/smart/coverage_completion_test.go

// Copyright 2026 Alibaba Group
// Licensed under the Apache License, Version 2.0 (the "License");

package smart

import (
	"bytes"
	"context"
	"encoding/json"
	"errors"
	"strings"
	"testing"

	"github.com/DingTalk-Real-AI/dingtalk-workspace-cli/internal/helpers"
	"github.com/DingTalk-Real-AI/dingtalk-workspace-cli/pkg/edition"
)

type smartCoverageCaller struct {
	responses map[string][]string
	failAt    map[string]int
	counts    map[string]int
}

func (c *smartCoverageCaller) CallTool(
	_ context.Context,
	product, tool string,
	_ map[string]any,
) (*edition.ToolResult, error) {
	if c.counts == nil {
		c.counts = map[string]int{}
	}
	key := product + "/" + tool
	c.counts[key]++
	if c.failAt[key] == c.counts[key] {
		return nil, errors.New("fixture failure")
	}
	responses := c.responses[key]
	text := `{"result":[]}`
	if len(responses) > 0 {
		index := c.counts[key] - 1
		if index >= len(responses) {
			index = len(responses) - 1
		}
		text = responses[index]
	}
	return &edition.ToolResult{Content: []edition.ContentBlock{{Type: "text", Text: text}}}, nil
}

func (c *smartCoverageCaller) CallReadTool(ctx context.Context, product, tool string, args map[string]any) (*edition.ToolResult, error) {
	return c.CallTool(ctx, product, tool, args)
}

func (*smartCoverageCaller) Format() string { return "json" }
func (*smartCoverageCaller) DryRun() bool   { return false }
func (*smartCoverageCaller) Fields() string { return "" }
func (*smartCoverageCaller) JQ() string     { return "" }

func TestCrossPlatformCoverageRichMessageProjections(t *testing.T) {
	message := map[string]any{
		"openMessageId":      "msg",
		"openConversationId": "cid",
		"threadId":           "thread",
		"msgType":            "text",
		"createTime":         "1",
		"updateTime":         "2",
		"content":            `{"mediaId":"@image"}`,
		"emotionReplyList":   []any{map[string]any{"emojiName": "赞", "replyCount": 1}},
		"quotedMessage": map[string]any{
			"openMessageId": "quoted",
			"content":       `{"mediaId":"@quoted-image"}`,
		},
	}
	for name, row := range map[string]map[string]any{
		"at-me":  atMeProject(message),
		"chat":   projectChatMessage(message),
		"search": searchMsgProject(message),
	} {
		for _, key := range []string{"threadId", "messageType", "updateTime", "quotedMessage", "resourceRefs"} {
			if _, ok := row[key]; !ok {
				t.Errorf("%s projection missing %s: %#v", name, key, row)
			}
		}
		resources := row["resourceRefs"].([]map[string]any)
		if len(resources) != 2 {
			t.Fatalf("%s projected resources = %#v", name, resources)
		}
		quotedArgs := resources[1]["download"].(map[string]any)["arguments"].(map[string]any)
		if resources[1]["resourceId"] != "@quoted-image" ||
			quotedArgs["message-id"] != "quoted" ||
			quotedArgs["open-conversation-id"] != "cid" {
			t.Fatalf("%s quoted resource context = %#v", name, resources[1])
		}
	}
}

func TestCrossPlatformCoverageChatMessagesOpenIDRoute(t *testing.T) {
	caller := &smartCoverageCaller{responses: map[string][]string{
		"chat/list_individual_chat_message": {`{"result":{"messages":[]}}`},
	}}
	helpers.InitDeps(caller)
	root := newPlatformCoverageRoot()
	root.SetArgs([]string{"chat", "+chat-messages", "--open-dingtalk-id", "D-user", "--limit", "1", "--yes"})
	if err := root.Execute(); err != nil {
		t.Fatal(err)
	}
	if caller.counts["chat/list_individual_chat_message"] != 1 {
		t.Fatalf("calls = %#v", caller.counts)
	}
}

func TestCrossPlatformCoverageChatMembersListOutcomes(t *testing.T) {
	successResponses := map[string][]string{
		"chat/get_group_members": {`{"result":{"hasMore":false,"list":[{"memberEmpName":"成员","openDingtalkId":"D-user"}]}}`},
		"bot/list_group_bots":    {`{"result":{"bots":[{"robotName":"机器人","robotCode":"robot"}]}}`},
	}
	cases := []struct {
		name      string
		memberArg string
		failAt    map[string]int
		wantError bool
	}{
		{name: "both success"},
		{name: "user only failure", memberArg: "user", failAt: map[string]int{"chat/get_group_members": 1}, wantError: true},
		{name: "bot only failure", memberArg: "bot", failAt: map[string]int{"bot/list_group_bots": 1}, wantError: true},
		{name: "both failure", failAt: map[string]int{"chat/get_group_members": 1, "bot/list_group_bots": 1}, wantError: true},
		{name: "partial user failure", failAt: map[string]int{"chat/get_group_members": 1}},
		{name: "partial bot failure", failAt: map[string]int{"bot/list_group_bots": 1}},
	}
	for _, tc := range cases {
		t.Run(tc.name, func(t *testing.T) {
			caller := &smartCoverageCaller{responses: successResponses, failAt: tc.failAt}
			helpers.InitDeps(caller)
			root := newPlatformCoverageRoot()
			args := []string{"chat", "+chat-members-list", "--conversation-id", "cid"}
			if tc.memberArg != "" {
				args = append(args, "--member-types", tc.memberArg)
			}
			root.SetArgs(args)
			err := root.Execute()
			if (err != nil) != tc.wantError {
				t.Fatalf("error = %v, wantError=%v", err, tc.wantError)
			}
		})
	}

	for _, raw := range [][]string{{""}, {"unknown"}} {
		if _, _, err := resolveMemberTypes(raw); err == nil {
			t.Fatalf("resolveMemberTypes(%q) succeeded", raw)
		}
	}
	if users, bots, err := resolveMemberTypes(nil); err != nil || !users || !bots {
		t.Fatalf("default member types = %v %v %v", users, bots, err)
	}
}

func TestCrossPlatformCoverageChatMembersGroupResolutionAndProjection(t *testing.T) {
	cases := []struct {
		name      string
		search    string
		fail      bool
		wantError bool
	}{
		{name: "search error", fail: true, wantError: true},
		{name: "none", search: `{"result":[]}`, wantError: true},
		{name: "ambiguous", search: `{"result":[{"openConversationId":"c1","title":"群"},{"openConversationId":"c2","title":"群"}]}`, wantError: true},
		{name: "one", search: `{"result":[{"openConversationId":"c1","title":"群"}]}`},
	}
	for _, tc := range cases {
		t.Run(tc.name, func(t *testing.T) {
			failAt := map[string]int{}
			if tc.fail {
				failAt["im/search_groups"] = 1
			}
			caller := &smartCoverageCaller{
				responses: map[string][]string{
					"im/search_groups":       {tc.search},
					"chat/get_group_members": {`{"result":{"list":[]}}`},
					"bot/list_group_bots":    {`{"result":{"bots":[]}}`},
				},
				failAt: failAt,
			}
			helpers.InitDeps(caller)
			root := newPlatformCoverageRoot()
			root.SetArgs([]string{"chat", "+chat-members-list", "--group", "群"})
			err := root.Execute()
			if (err != nil) != tc.wantError {
				t.Fatalf("error = %v, wantError=%v", err, tc.wantError)
			}
		})
	}

	bots := groupBotProject(map[string]any{"bots": []any{"invalid", map[string]any{"name": "bot"}}})
	if len(bots) != 1 || bots[0]["name"] != "bot" {
		t.Fatalf("bots = %#v", bots)
	}
	if groupBotProject(map[string]any{"result": "invalid"}) == nil {
		t.Fatal("empty bot projection should be a non-nil empty slice")
	}
}

func TestCrossPlatformCoverageSearchValidationAndTimeErrors(t *testing.T) {
	cases := [][]string{
		{},
		{"--query", "x", "--start", "2026-07-01T00:00:00Z"},
		{"--query", "x", "--days", "0"},
		{"--query", "x", "--limit", "101"},
		{"--query", "x", "--page-limit", "0"},
		{"--query", "x", "--start", "bad", "--end", "2026-07-02T00:00:00Z"},
		{"--query", "x", "--start", "2026-07-01T00:00:00Z", "--end", "bad"},
		{"--query", "x", "--start", "2026-07-02T00:00:00Z", "--end", "2026-07-01T00:00:00Z"},
	}
	for _, tail := range cases {
		helpers.InitDeps(&smartCoverageCaller{})
		root := newPlatformCoverageRoot()
		root.SetArgs(append([]string{"chat", "+search-msg", "--yes"}, tail...))
		if err := root.Execute(); err == nil {
			t.Errorf("invalid search args succeeded: %v", tail)
		}
	}
}

func TestCrossPlatformCoverageSearchPaginationFailureModes(t *testing.T) {
	cases := []struct {
		name      string
		responses []string
		failAt    int
		args      []string
		wantError bool
	}{
		{name: "initial error", failAt: 1, args: []string{"--query", "x"}, wantError: true},
		{
			name: "duplicate and inferred cursor",
			responses: []string{
				`{"result":{"messages":[{"openMessageId":"m1"},{"openMessageId":"m1"}],"nextCursor":"c2"}}`,
				`{"result":{"messages":[],"hasMore":false}}`,
			},
			args: []string{"--query", "x", "--page-all", "--no-enrich"},
		},
		{
			name:      "stalled cursor",
			responses: []string{`{"result":{"messages":[],"hasMore":true,"nextCursor":"same"}}`},
			args:      []string{"--query", "x", "--cursor", "same", "--page-all", "--no-enrich"},
		},
		{
			name:      "page limit",
			responses: []string{`{"result":{"messages":[],"hasMore":true,"nextCursor":"next"}}`},
			args:      []string{"--query", "x", "--page-all", "--page-limit", "1", "--no-enrich"},
		},
	}
	for _, tc := range cases {
		t.Run(tc.name, func(t *testing.T) {
			caller := &smartCoverageCaller{
				responses: map[string][]string{"im/search_messages": tc.responses},
				failAt:    map[string]int{"im/search_messages": tc.failAt},
			}
			helpers.InitDeps(caller)
			root := newPlatformCoverageRoot()
			var output bytes.Buffer
			root.SetOut(&output)
			root.SetArgs(append([]string{"chat", "+search-msg", "--yes"}, tc.args...))
			err := root.Execute()
			if (err != nil) != tc.wantError {
				t.Fatalf("error = %v, wantError=%v", err, tc.wantError)
			}
			if err == nil {
				var payload map[string]any
				if decodeErr := json.Unmarshal(output.Bytes(), &payload); decodeErr != nil {
					t.Fatalf("decode %q: %v", strings.TrimSpace(output.String()), decodeErr)
				}
				if payload["complete"] != false && tc.name != "duplicate and inferred cursor" {
					t.Fatalf("payload = %#v", payload)
				}
			}
		})
	}
}