fix: mock

AztecProtocol · Dec 20, 2024 · fa2d595 · fa2d595
1 parent d7d23d3
commit fa2d595
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 4 deletions.
diff --git a/yarn-project/p2p/src/services/peer-scoring/peer_scoring.ts b/yarn-project/p2p/src/services/peer-scoring/peer_scoring.ts
@@ -88,11 +88,12 @@ export class PeerScoring {
   }
 
   public getScoreState(peerId: string): PeerScoreState {
+    // TODO: permanently store banned peers?
     const score = this.getScore(peerId);
-    if (score <= MIN_SCORE_BEFORE_BAN) {
+    if (score < MIN_SCORE_BEFORE_BAN) {
       return PeerScoreState.Banned;
     }
-    if (score <= MIN_SCORE_BEFORE_DISCONNECT) {
+    if (score < MIN_SCORE_BEFORE_DISCONNECT) {
       return PeerScoreState.Disconnect;
     }
     return PeerScoreState.Healthy;

diff --git a/yarn-project/p2p/src/services/peer_manager.test.ts b/yarn-project/p2p/src/services/peer_manager.test.ts
@@ -323,8 +323,7 @@ describe('PeerManager', () => {
     });
 
     it('should properly clean up peers on stop', async () => {
-      const enr = await createMockENR();
-      await discoveredPeerCallback(enr);
+      mockLibP2PNode.getPeers.mockReturnValue([await createSecp256k1PeerId(), await createSecp256k1PeerId()]);
 
       await peerManager.stop();
 

diff --git a/yarn-project/p2p/src/services/reqresp/protocols/goodbye_protocol.ts b/yarn-project/p2p/src/services/reqresp/protocols/goodbye_protocol.ts