mirror of
https://github.com/larksuite/cli.git
synced 2026-07-05 07:31:22 +08:00
4422265d5f
`im chats link` is registered as a regular service method (no `risk: high-risk-write` annotation), so the framework does not register the `--yes` flag on it. Setting `Yes: true` on the e2e Request makes the runner append `--yes`, which cobra rejects with `unknown flag: --yes` before the request is ever issued — the rest of the assertions then fall through with empty stdout. The flag was added in #633 alongside the risk-tiering rollout that covered other workflows that genuinely flipped to high-risk-write. For chats link the API call (creating a chat share link with a configurable validity period) is not destructive and was never re-classified, so the line is just leftover from that pass. Drop it to restore the e2e green; if we ever decide to gate share-link creation behind confirmation we can re-add it together with the metadata flip. Change-Id: Ieb094407a7f0fa18cd130a9d80c7146274b5ecc7
130 lines
3.7 KiB
Go
130 lines
3.7 KiB
Go
// Copyright (c) 2026 Lark Technologies Pte. Ltd.
|
|
// SPDX-License-Identifier: MIT
|
|
|
|
package im
|
|
|
|
import (
|
|
"context"
|
|
"testing"
|
|
"time"
|
|
|
|
clie2e "github.com/larksuite/cli/tests/cli_e2e"
|
|
"github.com/stretchr/testify/assert"
|
|
"github.com/stretchr/testify/require"
|
|
"github.com/tidwall/gjson"
|
|
)
|
|
|
|
// TestIM_ChatUpdateWorkflow tests the +chat-update shortcut.
|
|
func TestIM_ChatUpdateWorkflow(t *testing.T) {
|
|
parentT := t
|
|
ctx, cancel := context.WithTimeout(context.Background(), 2*time.Minute)
|
|
t.Cleanup(cancel)
|
|
|
|
suffix := clie2e.GenerateSuffix()
|
|
originalName := "lark-cli-e2e-im-update-" + suffix
|
|
updatedName := originalName + "-updated"
|
|
updatedDescription := "Updated description for e2e test"
|
|
|
|
chatID := createChat(t, parentT, ctx, originalName)
|
|
|
|
t.Run("update chat name as bot", func(t *testing.T) {
|
|
result, err := clie2e.RunCmd(ctx, clie2e.Request{
|
|
Args: []string{"im", "+chat-update",
|
|
"--chat-id", chatID,
|
|
"--name", updatedName,
|
|
},
|
|
DefaultAs: "bot",
|
|
})
|
|
require.NoError(t, err)
|
|
result.AssertExitCode(t, 0)
|
|
result.AssertStdoutStatus(t, true)
|
|
})
|
|
|
|
t.Run("update chat description as bot", func(t *testing.T) {
|
|
result, err := clie2e.RunCmd(ctx, clie2e.Request{
|
|
Args: []string{"im", "+chat-update",
|
|
"--chat-id", chatID,
|
|
"--description", updatedDescription,
|
|
},
|
|
DefaultAs: "bot",
|
|
})
|
|
require.NoError(t, err)
|
|
result.AssertExitCode(t, 0)
|
|
result.AssertStdoutStatus(t, true)
|
|
})
|
|
|
|
t.Run("get updated chat as bot", func(t *testing.T) {
|
|
result, err := clie2e.RunCmd(ctx, clie2e.Request{
|
|
Args: []string{"im", "chats", "get"},
|
|
DefaultAs: "bot",
|
|
Params: map[string]any{"chat_id": chatID},
|
|
})
|
|
require.NoError(t, err)
|
|
result.AssertExitCode(t, 0)
|
|
result.AssertStdoutStatus(t, 0)
|
|
|
|
assert.Equal(t, updatedName, gjson.Get(result.Stdout, "data.name").String())
|
|
assert.Equal(t, updatedDescription, gjson.Get(result.Stdout, "data.description").String())
|
|
})
|
|
}
|
|
|
|
// TestIM_ChatsGetWorkflow tests the im chats get command.
|
|
func TestIM_ChatsGetWorkflow(t *testing.T) {
|
|
parentT := t
|
|
ctx, cancel := context.WithTimeout(context.Background(), 2*time.Minute)
|
|
t.Cleanup(cancel)
|
|
|
|
suffix := clie2e.GenerateSuffix()
|
|
chatName := "lark-cli-e2e-chats-get-" + suffix
|
|
|
|
chatID := createChat(t, parentT, ctx, chatName)
|
|
|
|
t.Run("get chat info as bot", func(t *testing.T) {
|
|
result, err := clie2e.RunCmd(ctx, clie2e.Request{
|
|
Args: []string{"im", "chats", "get"},
|
|
DefaultAs: "bot",
|
|
Params: map[string]any{"chat_id": chatID},
|
|
})
|
|
require.NoError(t, err)
|
|
t.Logf("chats get result: %s", result.Stdout)
|
|
result.AssertExitCode(t, 0)
|
|
result.AssertStdoutStatus(t, 0)
|
|
|
|
dataExists := gjson.Get(result.Stdout, "data").Exists()
|
|
require.True(t, dataExists, "data object should exist")
|
|
|
|
chatNameGot := gjson.Get(result.Stdout, "data.name").String()
|
|
require.Equal(t, chatName, chatNameGot)
|
|
})
|
|
}
|
|
|
|
// TestIM_ChatsLinkWorkflow tests the im chats link command.
|
|
func TestIM_ChatsLinkWorkflow(t *testing.T) {
|
|
parentT := t
|
|
ctx, cancel := context.WithTimeout(context.Background(), 2*time.Minute)
|
|
t.Cleanup(cancel)
|
|
|
|
suffix := clie2e.GenerateSuffix()
|
|
chatName := "lark-cli-e2e-chats-link-" + suffix
|
|
|
|
chatID := createChat(t, parentT, ctx, chatName)
|
|
|
|
t.Run("get chat share link as bot", func(t *testing.T) {
|
|
result, err := clie2e.RunCmd(ctx, clie2e.Request{
|
|
Args: []string{"im", "chats", "link"},
|
|
DefaultAs: "bot",
|
|
Params: map[string]any{"chat_id": chatID},
|
|
Data: map[string]any{
|
|
"validity_period": "week",
|
|
},
|
|
})
|
|
require.NoError(t, err)
|
|
result.AssertExitCode(t, 0)
|
|
result.AssertStdoutStatus(t, 0)
|
|
|
|
shareLink := gjson.Get(result.Stdout, "data.share_link").String()
|
|
require.NotEmpty(t, shareLink, "share_link should not be empty")
|
|
t.Logf("Generated share link: %s", shareLink)
|
|
})
|
|
}
|