apply suggestions

Consensys · Oct 14, 2024 · e69e1d6 · e69e1d6
1 parent 9361b6b
commit e69e1d6
Show file tree

Hide file tree

Showing 3 changed files with 5 additions and 21 deletions.
diff --git a/ethereum/spec/src/main/java/tech/pegasys/teku/spec/schemas/registry/BaseSchemaProvider.java b/ethereum/spec/src/main/java/tech/pegasys/teku/spec/schemas/registry/BaseSchemaProvider.java
@@ -134,8 +134,8 @@ public Builder<T> until(final SpecMilestone untilMilestone) {
       return this;
     }
 
-    public Builder<T> constant(final boolean isConstant) {
-      this.isConstant = isConstant;
+    public Builder<T> constant() {
+      this.isConstant = true;
       return this;
     }
 

diff --git a/...eum/spec/src/main/java/tech/pegasys/teku/spec/schemas/registry/SchemaRegistryBuilder.java b/...eum/spec/src/main/java/tech/pegasys/teku/spec/schemas/registry/SchemaRegistryBuilder.java
@@ -45,7 +45,7 @@ public static SchemaRegistryBuilder create() {
   private static SchemaProvider<SszBitvectorSchema<SszBitvector>>
       createAttnetsENRFieldSchemaProvider() {
     return providerBuilder(ATTNETS_ENR_FIELD_SCHEMA)
-        .constant(true)
+        .constant()
         .withCreator(
             SpecMilestone.PHASE0,
             (registry, specConfig) ->
@@ -55,7 +55,7 @@ public static SchemaRegistryBuilder create() {
 
   private static SchemaProvider<AttestationSchema<Attestation>> createAttestationSchemaProvider() {
     return providerBuilder(ATTESTATION_SCHEMA)
-        .constant(true)
+        .constant()
         .withCreator(
             SpecMilestone.PHASE0,
             (registry, specConfig) ->

diff --git a/...um/spec/src/test/java/tech/pegasys/teku/spec/schemas/registry/BaseSchemaProviderTest.java b/...um/spec/src/test/java/tech/pegasys/teku/spec/schemas/registry/BaseSchemaProviderTest.java
@@ -60,7 +60,7 @@ void shouldSupportContinuousUntilHighestMilestone() {
   void shouldSupportContinuousConstantWithUntil() {
     final SchemaProvider<?> provider =
         providerBuilder(STRING_SCHEMA_ID)
-            .constant(true)
+            .constant()
             .withCreator(PHASE0, (r, c) -> "TestSchemaPhase0")
             .withCreator(BELLATRIX, (r, c) -> "TestSchemaBellatrix")
             .until(CAPELLA)
@@ -115,22 +115,6 @@ void shouldSupportContinuousDefaultVariable() {
         .containsExactly(PHASE0, ALTAIR, BELLATRIX, CAPELLA);
   }
 
-  @Test
-  void shouldAllowSingleMilestone() {
-    final SchemaProvider<?> provider =
-        providerBuilder(STRING_SCHEMA_ID)
-            .withCreator(ALTAIR, (r, c) -> "TestSchemaAltair")
-            .withCreator(BELLATRIX, (r, c) -> "TestSchemaBellatrix")
-            .until(CAPELLA)
-            .build();
-
-    when(mockRegistry.getMilestone()).thenReturn(DENEB);
-
-    assertThatThrownBy(() -> provider.getSchema(mockRegistry))
-        .isInstanceOf(IllegalArgumentException.class)
-        .hasMessage("It is not supposed to create a specific version for DENEB");
-  }
-
   @Test
   void shouldThrowWhenNoCreators() {
     assertThatThrownBy(() -> providerBuilder(STRING_SCHEMA_ID).build())