mirror of
https://github.com/google-gemini/gemini-cli.git
synced 2026-05-02 16:04:38 -07:00
fix(infra) - Fix flake for file interactive system (#11019)
Co-authored-by: gemini-cli-robot <gemini-cli-robot@google.com>
This commit is contained in:
@@ -585,6 +585,34 @@ export class TestRig {
|
||||
);
|
||||
}
|
||||
|
||||
async expectToolCallSuccess(toolNames: string[], timeout?: number) {
|
||||
// Use environment-specific timeout
|
||||
if (!timeout) {
|
||||
timeout = getDefaultTimeout();
|
||||
}
|
||||
|
||||
// Wait for telemetry to be ready before polling for tool calls
|
||||
await this.waitForTelemetryReady();
|
||||
|
||||
const success = await poll(
|
||||
() => {
|
||||
const toolLogs = this.readToolLogs();
|
||||
return toolNames.some((name) =>
|
||||
toolLogs.some(
|
||||
(log) => log.toolRequest.name === name && log.toolRequest.success,
|
||||
),
|
||||
);
|
||||
},
|
||||
timeout,
|
||||
100,
|
||||
);
|
||||
|
||||
expect(
|
||||
success,
|
||||
`Expected to find successful toolCalls for ${JSON.stringify(toolNames)}`,
|
||||
).toBe(true);
|
||||
}
|
||||
|
||||
async waitForAnyToolCall(toolNames: string[], timeout?: number) {
|
||||
// Use environment-specific timeout
|
||||
if (!timeout) {
|
||||
|
||||
Reference in New Issue
Block a user