fix test

Consensys · Dec 13, 2022 · 3971758 · 3971758
1 parent 261d922
commit 3971758
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/.../teku/networking/eth2/rpc/beaconchain/methods/BlobsSidecarsByRangeMessageHandlerTest.java b/.../teku/networking/eth2/rpc/beaconchain/methods/BlobsSidecarsByRangeMessageHandlerTest.java
@@ -106,13 +106,15 @@ public void shouldNotSendBlobsSidecarsIfPeerIsRateLimited() {
 
   @Test
   public void shouldSendToPeerRequestedNumberOfBlobsSidecars() {
-    final UInt64 count = UInt64.valueOf(5);
 
     final Bytes32 headBlockRoot = dataStructureUtil.randomBytes32();
 
+    when(peer.wantToMakeRequest()).thenReturn(true);
+    when(peer.wantToReceiveBlobsSidecars(listener, 5)).thenReturn(true);
     when(combinedChainDataClient.getBestBlockRoot()).thenReturn(Optional.of(headBlockRoot));
     when(listener.respond(any())).thenReturn(SafeFuture.COMPLETE);
 
+    final UInt64 count = UInt64.valueOf(5);
     final BlobsSidecarsByRangeRequestMessage request =
         new BlobsSidecarsByRangeRequestMessage(ZERO, count);