diff --git a/test/gateway-hook-concurrency.e2e.test.ts b/test/gateway-hook-concurrency.e2e.test.ts index 9daa8ab2a403..020e09f204f7 100644 --- a/test/gateway-hook-concurrency.e2e.test.ts +++ b/test/gateway-hook-concurrency.e2e.test.ts @@ -89,7 +89,10 @@ describe("Gateway hook concurrency", () => { error: "hook agent run did not start before admission timeout", runId: expect.any(String), }); - expect(modelServer.active(), instance.logs()).toBeGreaterThan(0); + await vi.waitFor(() => expect(modelServer.active(), instance.logs()).toBeGreaterThan(0), { + interval: 20, + timeout: 30_000, + }); expect(modelServer.peak(), instance.logs()).toBeLessThanOrEqual(SHARED_BUDGET); expect(modelServer.requestCount(), instance.logs()).toBeLessThanOrEqual(SHARED_BUDGET + 1);