Test isCloseToInSync method

Consensys · Apr 5, 2024 · 5c18dcf · 5c18dcf
1 parent fe1f79b
commit 5c18dcf
Show file tree

Hide file tree

Showing 2 changed files with 18 additions and 1 deletion.
diff --git a/networking/eth2/src/main/java/tech/pegasys/teku/networking/eth2/ActiveEth2P2PNetwork.java b/networking/eth2/src/main/java/tech/pegasys/teku/networking/eth2/ActiveEth2P2PNetwork.java
@@ -179,7 +179,8 @@ public void onSyncStateChanged(final boolean isInSync, final boolean isOptimisti
     gossipForkManager.onOptimisticHeadChanged(isOptimistic);
   }
 
-  private boolean isCloseToInSync() {
+  @VisibleForTesting
+  boolean isCloseToInSync() {
     final UInt64 currentEpoch = recentChainData.getCurrentEpoch().orElseThrow();
     final int maxSeedLookahead = spec.getSpecConfig(currentEpoch).getMaxSeedLookahead();
     final int slotsPerEpoch = spec.slotsPerEpoch(currentEpoch);

diff --git a/...orking/eth2/src/test/java/tech/pegasys/teku/networking/eth2/ActiveEth2P2PNetworkTest.java b/...orking/eth2/src/test/java/tech/pegasys/teku/networking/eth2/ActiveEth2P2PNetworkTest.java
@@ -255,6 +255,22 @@ void onSyncStateChanged_shouldNotResultInMultipleSubscriptions() {
     verify(eventChannels, times(1)).subscribe(eq(BlockGossipChannel.class), any());
   }
 
+  @Test
+  void isCloseToInSync_shouldCalculateDistanceWithSeedWhenFar() {
+    // Current slot is a long way beyond the chain head
+    storageSystem.chainUpdater().setCurrentSlot(UInt64.valueOf(50));
+    final boolean closeToInSync = network.isCloseToInSync();
+    assertThat(closeToInSync).isFalse();
+  }
+
+  @Test
+  void isCloseToInSync_shouldCalculateDistanceWithSeedWhenClose() {
+    // Current slot is beyond the chain head, but less than maxSeedLookahead distance
+    storageSystem.chainUpdater().setCurrentSlot(UInt64.valueOf(30));
+    final boolean closeToInSync = network.isCloseToInSync();
+    assertThat(closeToInSync).isTrue();
+  }
+
   @SuppressWarnings("unchecked")
   private ArgumentCaptor<Iterable<Integer>> subnetIdCaptor() {
     return ArgumentCaptor.forClass(Iterable.class);