test(drive): retry duplicate-remote push in live E2E (#1403)

This commit is contained in:
Yuxuan Zhao
2026-06-12 13:48:19 +08:00
committed by GitHub
parent 76ba6fad4f
commit 2a7e9c7d0d
3 changed files with 9 additions and 6 deletions

View File

@@ -161,7 +161,7 @@ func TestDrive_DuplicateRemoteWorkflow(t *testing.T) {
time.Sleep(1200 * time.Millisecond)
_ = uploadNamedFile(t, workDir, folderToken, "_push_dup_second.txt", "dup.txt", "remote-second")
pushResult, err := clie2e.RunCmd(ctx, clie2e.Request{
pushResult, err := clie2e.RunCmdWithRetry(ctx, clie2e.Request{
Args: []string{
"drive", "+push",
"--local-dir", "local",
@@ -173,7 +173,7 @@ func TestDrive_DuplicateRemoteWorkflow(t *testing.T) {
},
WorkDir: workDir,
DefaultAs: "bot",
})
}, clie2e.RetryOptions{})
require.NoError(t, err)
pushResult.AssertExitCode(t, 0)
pushResult.AssertStdoutStatus(t, true)

View File

@@ -313,14 +313,16 @@ func cleanupFeedShortcuts(parentT *testing.T, defaultAs string, chatIDs ...strin
}
// TestIM_FeedShortcutDryRun covers all three shortcuts in --dry-run mode
// using fake credentials. This is the only test that runs without a real
// user token because dry-run short-circuits before any network call.
// using env-only identity hints. strict_mode/default_as lock the command to
// user identity without injecting a fake user token that would trigger
// user_info verification during startup.
func TestIM_FeedShortcutDryRun(t *testing.T) {
t.Setenv("LARKSUITE_CLI_CONFIG_DIR", t.TempDir())
t.Setenv("LARKSUITE_CLI_APP_ID", "app")
t.Setenv("LARKSUITE_CLI_APP_SECRET", "secret")
t.Setenv("LARKSUITE_CLI_USER_ACCESS_TOKEN", "fake_user_token")
t.Setenv("LARKSUITE_CLI_BRAND", "feishu")
t.Setenv("LARKSUITE_CLI_STRICT_MODE", "user")
t.Setenv("LARKSUITE_CLI_DEFAULT_AS", "user")
ctx, cancel := context.WithTimeout(context.Background(), 30*time.Second)
t.Cleanup(cancel)

View File

@@ -247,8 +247,9 @@ func TestIM_FlagDryRun(t *testing.T) {
t.Setenv("LARKSUITE_CLI_CONFIG_DIR", t.TempDir())
t.Setenv("LARKSUITE_CLI_APP_ID", "app")
t.Setenv("LARKSUITE_CLI_APP_SECRET", "secret")
t.Setenv("LARKSUITE_CLI_USER_ACCESS_TOKEN", "fake_user_token")
t.Setenv("LARKSUITE_CLI_BRAND", "feishu")
t.Setenv("LARKSUITE_CLI_STRICT_MODE", "user")
t.Setenv("LARKSUITE_CLI_DEFAULT_AS", "user")
ctx, cancel := context.WithTimeout(context.Background(), 30*time.Second)
t.Cleanup(cancel)