diff --git a/yarn-project/prover-node/src/prover-node.test.ts b/yarn-project/prover-node/src/prover-node.test.ts index 98714c2e1d0..19da19eadd3 100644 --- a/yarn-project/prover-node/src/prover-node.test.ts +++ b/yarn-project/prover-node/src/prover-node.test.ts @@ -15,6 +15,7 @@ import { import { type ContractDataSource, EthAddress, Fr } from '@aztec/circuits.js'; import { times } from '@aztec/foundation/collection'; import { Signature } from '@aztec/foundation/eth-signature'; +import { makeBackoff, retry } from '@aztec/foundation/retry'; import { sleep } from '@aztec/foundation/sleep'; import { openTmpStore } from '@aztec/kv-store/lmdb'; import { @@ -367,10 +368,15 @@ describe('prover-node', () => { await proverNode.handleEpochCompleted(10n); // Wait for message to be propagated - await sleep(1000); - - // Check the other node received a quote via p2p - expect(p2pEpochReceivedSpy).toHaveBeenCalledTimes(1); + await retry( + // eslint-disable-next-line require-await + async () => { + // Check the other node received a quote via p2p + expect(p2pEpochReceivedSpy).toHaveBeenCalledTimes(1); + }, + 'Waiting for quote to be received', + makeBackoff(times(20, () => 1)), + ); // We should be able to retreive the quote from the other node const peerFinalStateQuotes = await otherP2PClient.getEpochProofQuotes(10n);