mirror of
https://github.com/larksuite/cli.git
synced 2026-07-06 00:06:28 +08:00
112 lines
3.4 KiB
Go
112 lines
3.4 KiB
Go
// Copyright (c) 2026 Lark Technologies Pte. Ltd.
|
|
// SPDX-License-Identifier: MIT
|
|
|
|
package im
|
|
|
|
import (
|
|
"context"
|
|
"strings"
|
|
"testing"
|
|
"time"
|
|
|
|
clie2e "github.com/larksuite/cli/tests/cli_e2e"
|
|
"github.com/stretchr/testify/assert"
|
|
"github.com/stretchr/testify/require"
|
|
"github.com/tidwall/gjson"
|
|
)
|
|
|
|
func TestIM_MessageReplyWorkflowAsBot(t *testing.T) {
|
|
parentT := t
|
|
ctx, cancel := context.WithTimeout(context.Background(), 2*time.Minute)
|
|
t.Cleanup(cancel)
|
|
|
|
suffix := clie2e.GenerateSuffix()
|
|
chatName := "lark-cli-e2e-im-reply-" + suffix
|
|
originalMessage := "lark-cli-e2e-original-message-" + suffix
|
|
replyText := "lark-cli-e2e-reply-text-" + suffix
|
|
|
|
chatID := createChat(t, parentT, ctx, chatName)
|
|
messageID := sendMessage(t, ctx, chatID, originalMessage)
|
|
|
|
t.Run("reply to message in thread as bot", func(t *testing.T) {
|
|
result, err := clie2e.RunCmd(ctx, clie2e.Request{
|
|
Args: []string{"im", "+messages-reply",
|
|
"--message-id", messageID,
|
|
"--text", replyText,
|
|
"--reply-in-thread",
|
|
},
|
|
DefaultAs: "bot",
|
|
})
|
|
require.NoError(t, err)
|
|
result.AssertExitCode(t, 0)
|
|
result.AssertStdoutStatus(t, true)
|
|
assert.NotEmpty(t, gjson.Get(result.Stdout, "data.message_id").String(), "stdout:\n%s", result.Stdout)
|
|
assert.Equal(t, chatID, gjson.Get(result.Stdout, "data.chat_id").String(), "stdout:\n%s", result.Stdout)
|
|
})
|
|
|
|
t.Run("list thread replies as bot", func(t *testing.T) {
|
|
listResult, err := clie2e.RunCmdWithRetry(ctx, clie2e.Request{
|
|
Args: []string{
|
|
"im", "+chat-messages-list",
|
|
"--chat-id", chatID,
|
|
"--start", time.Now().UTC().Add(-10 * time.Minute).Format(time.RFC3339),
|
|
"--end", time.Now().UTC().Add(10 * time.Minute).Format(time.RFC3339),
|
|
},
|
|
DefaultAs: "bot",
|
|
}, clie2e.RetryOptions{
|
|
ShouldRetry: func(result *clie2e.Result) bool {
|
|
if result == nil || result.ExitCode != 0 {
|
|
return true
|
|
}
|
|
for _, item := range gjson.Get(result.Stdout, "data.messages").Array() {
|
|
if item.Get("message_id").String() == messageID && item.Get("thread_id").String() != "" {
|
|
return false
|
|
}
|
|
}
|
|
return true
|
|
},
|
|
})
|
|
require.NoError(t, err)
|
|
listResult.AssertExitCode(t, 0)
|
|
listResult.AssertStdoutStatus(t, true)
|
|
|
|
var threadID string
|
|
for _, item := range gjson.Get(listResult.Stdout, "data.messages").Array() {
|
|
if item.Get("message_id").String() == messageID {
|
|
threadID = item.Get("thread_id").String()
|
|
break
|
|
}
|
|
}
|
|
require.NotEmpty(t, threadID, "expected thread_id for message %s in stdout:\n%s", messageID, listResult.Stdout)
|
|
|
|
threadResult, err := clie2e.RunCmdWithRetry(ctx, clie2e.Request{
|
|
Args: []string{"im", "+threads-messages-list", "--thread", threadID},
|
|
DefaultAs: "bot",
|
|
}, clie2e.RetryOptions{
|
|
ShouldRetry: func(result *clie2e.Result) bool {
|
|
if result == nil || result.ExitCode != 0 {
|
|
return true
|
|
}
|
|
for _, item := range gjson.Get(result.Stdout, "data.messages").Array() {
|
|
if strings.Contains(item.Get("content").String(), replyText) {
|
|
return false
|
|
}
|
|
}
|
|
return true
|
|
},
|
|
})
|
|
require.NoError(t, err)
|
|
threadResult.AssertExitCode(t, 0)
|
|
threadResult.AssertStdoutStatus(t, true)
|
|
|
|
var found bool
|
|
for _, item := range gjson.Get(threadResult.Stdout, "data.messages").Array() {
|
|
if strings.Contains(item.Get("content").String(), replyText) {
|
|
found = true
|
|
break
|
|
}
|
|
}
|
|
require.True(t, found, "expected reply content in stdout:\n%s", threadResult.Stdout)
|
|
})
|
|
}
|