add test

Consensys · Apr 4, 2024 · b8ec8fe · b8ec8fe
1 parent 78edae4
commit b8ec8fe
Showing 1 changed file with 17 additions and 0 deletions.
diff --git a/...ch/pegasys/teku/spec/datastructures/state/beaconstate/common/ValidatorIndexCacheTest.java b/...ch/pegasys/teku/spec/datastructures/state/beaconstate/common/ValidatorIndexCacheTest.java
@@ -43,6 +43,23 @@ public class ValidatorIndexCacheTest {
   @SuppressWarnings("unchecked")
   final Cache<BLSPublicKey, Integer> cache = mock(Cache.class);
 
+  @Test
+  public void shouldReturnEmptyIfValidatorIndexIsNotConsistentWithNumberOfValidatorsInState() {
+    final SszList<Validator> validators = state.getValidators();
+    final int latestFinalizedIndex = NUMBER_OF_VALIDATORS - 1;
+    final ValidatorIndexCache validatorIndexCache = new ValidatorIndexCache();
+    validatorIndexCache.updateLatestFinalizedIndex(state);
+
+    final BLSPublicKey publicKey = validators.get(latestFinalizedIndex).getPublicKey();
+    // cache eagerly the last validator public key
+    validatorIndexCache.invalidateWithNewValue(publicKey, latestFinalizedIndex);
+
+    // state with one less validator
+    final BeaconState state = dataStructureUtil.randomBeaconState(NUMBER_OF_VALIDATORS - 1);
+
+    assertThat(validatorIndexCache.getValidatorIndex(state, publicKey)).isEmpty();
+  }
+
   @Test
   public void shouldScanFinalizedStateAndCache() {
     final SszList<Validator> validators = state.getValidators();