diff --git a/src/infra/net/ssrf.dispatcher.test.ts b/src/infra/net/ssrf.dispatcher.test.ts index 07231bf72dc..001ac72576a 100644 --- a/src/infra/net/ssrf.dispatcher.test.ts +++ b/src/infra/net/ssrf.dispatcher.test.ts @@ -247,9 +247,7 @@ describe("createPinnedDispatcher", () => { const callback = vi.fn(); lookup?.("example.com", callback); - const originalLookupCall = originalLookupMock.mock.calls[0]; - expect(originalLookupCall?.[0]).toBe("example.com"); - expect(typeof originalLookupCall?.[1]).toBe("function"); + expect(originalLookupMock).toHaveBeenCalledWith("example.com", expect.any(Function)); expect(callback).toHaveBeenCalledWith(null, "93.184.216.34", 4); });