fix test

AztecProtocol · Jan 18, 2025 · 2a8ca8f · 2a8ca8f
1 parent d8fb737
commit 2a8ca8f
Show file tree

Hide file tree

Showing 3 changed files with 2 additions and 2 deletions.
diff --git a/yarn-project/p2p/src/services/peer_manager.test.ts b/yarn-project/p2p/src/services/peer_manager.test.ts
@@ -292,6 +292,7 @@ describe('PeerManager', () => {
       // Set the peer scores to trigger different states
       peerManager.penalizePeer(bannedPeerId, PeerErrorSeverity.LowToleranceError); // Will set score below -100
       peerManager.penalizePeer(bannedPeerId, PeerErrorSeverity.LowToleranceError); // Additional penalty to ensure banned state
+      peerManager.penalizePeer(bannedPeerId, PeerErrorSeverity.HighToleranceError);
 
       peerManager.penalizePeer(disconnectPeerId, PeerErrorSeverity.LowToleranceError); // Will set score between -100 and -50
       peerManager.penalizePeer(disconnectPeerId, PeerErrorSeverity.HighToleranceError);
@@ -315,6 +316,7 @@ describe('PeerManager', () => {
         ReqRespSubProtocol.GOODBYE,
         Buffer.from([GoodByeReason.DISCONNECTED]),
       );
+
       // Verify that hangUp was not called for the healthy peer
       expect(mockLibP2PNode.hangUp).not.toHaveBeenCalledWith(healthyPeerId);
 

diff --git a/yarn-project/p2p/src/services/reqresp/rate_limiter/rate_limiter.test.ts b/yarn-project/p2p/src/services/reqresp/rate_limiter/rate_limiter.test.ts
@@ -5,7 +5,6 @@ import { type PeerId } from '@libp2p/interface';
 import { type MockProxy, mock } from 'jest-mock-extended';
 
 import { type PeerScoring } from '../../peer-scoring/peer_scoring.js';
-import { type PeerManager } from '../../peer_manager.js';
 import { ReqRespSubProtocol, type ReqRespSubProtocolRateLimits } from '../interface.js';
 import { RequestResponseRateLimiter } from './rate_limiter.js';
 

diff --git a/yarn-project/p2p/src/services/reqresp/reqresp.test.ts b/yarn-project/p2p/src/services/reqresp/reqresp.test.ts
@@ -16,7 +16,6 @@ import {
   stopNodes,
 } from '../../mocks/index.js';
 import { type PeerScoring } from '../peer-scoring/peer_scoring.js';
-import { type PeerManager } from '../peer_manager.js';
 import { ReqRespSubProtocol, RequestableBuffer } from './interface.js';
 import { GoodByeReason } from './protocols/goodbye.js';