Add test for compute_proposer_index electra

Consensys · Jun 26, 2024 · a806f97 · a806f97
1 parent bbc319b
commit a806f97
Showing 1 changed file with 27 additions and 0 deletions.
diff --git a/...st/java/tech/pegasys/teku/spec/logic/versions/electra/helpers/MiscHelpersElectraTest.java b/...st/java/tech/pegasys/teku/spec/logic/versions/electra/helpers/MiscHelpersElectraTest.java
@@ -14,12 +14,18 @@
 package tech.pegasys.teku.spec.logic.versions.electra.helpers;
 
 import static org.assertj.core.api.Assertions.assertThat;
+import static org.mockito.Mockito.never;
+import static org.mockito.Mockito.spy;
+import static org.mockito.Mockito.verify;
 
+import it.unimi.dsi.fastutil.ints.IntList;
+import org.apache.tuweni.bytes.Bytes32;
 import org.junit.jupiter.api.Test;
 import tech.pegasys.teku.infrastructure.unsigned.UInt64;
 import tech.pegasys.teku.spec.Spec;
 import tech.pegasys.teku.spec.TestSpecFactory;
 import tech.pegasys.teku.spec.config.SpecConfigElectra;
+import tech.pegasys.teku.spec.datastructures.state.BeaconStateTestBuilder;
 import tech.pegasys.teku.spec.datastructures.state.beaconstate.BeaconState;
 import tech.pegasys.teku.spec.datastructures.state.beaconstate.versions.electra.BeaconStateElectra;
 import tech.pegasys.teku.spec.datastructures.state.beaconstate.versions.electra.MutableBeaconStateElectra;
@@ -73,4 +79,25 @@ public void isFormerDepositMechanismDisabled_returnsFalseIfNotDisabled() {
 
     assertThat(miscHelpersElectra.isFormerDepositMechanismDisabled(state)).isFalse();
   }
+
+  @Test
+  public void computeProposerIndexShouldUseMaxEffectiveBalanceElectra() {
+    final SpecConfigElectra specConfigElectra =
+        spy(SpecConfigElectra.required(spec.getGenesisSpecConfig()));
+    final MiscHelpersElectra miscHelpersElectra =
+        new MiscHelpersElectra(specConfigElectra, predicates, schemaDefinitionsElectra);
+
+    final BeaconState state =
+        new BeaconStateTestBuilder(dataStructureUtil)
+            .forkVersion(spec.getGenesisSpecConfig().getGenesisForkVersion())
+            .activeValidator(UInt64.THIRTY_TWO_ETH)
+            .activeValidator(UInt64.THIRTY_TWO_ETH)
+            .activeValidator(UInt64.THIRTY_TWO_ETH)
+            .build();
+
+    miscHelpersElectra.computeProposerIndex(state, IntList.of(0, 1, 2), Bytes32.ZERO);
+
+    verify(specConfigElectra).getMaxEffectiveBalanceElectra();
+    verify(specConfigElectra, never()).getMaxEffectiveBalance();
+  }
 }