Skip to content

Commit

Permalink
chore: Add retries for prover node p2p test (#10699)
Browse files Browse the repository at this point in the history
  • Loading branch information
spalladino authored Dec 13, 2024
1 parent 4cc0a6d commit 4115bf9
Showing 1 changed file with 10 additions and 4 deletions.
14 changes: 10 additions & 4 deletions yarn-project/prover-node/src/prover-node.test.ts
Original file line number Diff line number Diff line change
Expand Up @@ -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 {
Expand Down Expand Up @@ -368,10 +369,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);
Expand Down

0 comments on commit 4115bf9

Please sign in to comment.