forked from Consensys/teku
-
Notifications
You must be signed in to change notification settings - Fork 0
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
7390 deserialisation (Consensys#7549)
* add RemoteValidatorApiHandler Block V3 implementation
- Loading branch information
1 parent
0ade8dc
commit 30d8007
Showing
18 changed files
with
537 additions
and
44 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
276 changes: 276 additions & 0 deletions
276
...est/java/tech/pegasys/teku/validator/remote/typedef/handlers/ProduceBlockRequestTest.java
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,276 @@ | ||
| /* | ||
| * Copyright Consensys Software Inc., 2023 | ||
| * | ||
| * Licensed under the Apache License, Version 2.0 (the "License"); you may not use this file except in compliance with | ||
| * the License. You may obtain a copy of the License at | ||
| * | ||
| * http://www.apache.org/licenses/LICENSE-2.0 | ||
| * | ||
| * Unless required by applicable law or agreed to in writing, software distributed under the License is distributed on | ||
| * an "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the License for the | ||
| * specific language governing permissions and limitations under the License. | ||
| */ | ||
|
|
||
| package tech.pegasys.teku.validator.remote.typedef.handlers; | ||
|
|
||
| import static org.assertj.core.api.Assertions.assertThat; | ||
| import static org.assertj.core.api.Assumptions.assumeThat; | ||
| import static tech.pegasys.teku.infrastructure.http.HttpStatusCodes.SC_OK; | ||
| import static tech.pegasys.teku.infrastructure.http.RestApiConstants.HEADER_EXECUTION_PAYLOAD_BLINDED; | ||
| import static tech.pegasys.teku.infrastructure.unsigned.UInt64.ONE; | ||
| import static tech.pegasys.teku.spec.SpecMilestone.BELLATRIX; | ||
| import static tech.pegasys.teku.spec.SpecMilestone.CAPELLA; | ||
| import static tech.pegasys.teku.spec.SpecMilestone.DENEB; | ||
|
|
||
| import com.google.common.net.MediaType; | ||
| import java.util.Optional; | ||
| import okhttp3.mockwebserver.MockResponse; | ||
| import okio.Buffer; | ||
| import org.junit.jupiter.api.AfterEach; | ||
| import org.junit.jupiter.api.BeforeEach; | ||
| import org.junit.jupiter.api.TestTemplate; | ||
| import tech.pegasys.teku.bls.BLSSignature; | ||
| import tech.pegasys.teku.infrastructure.unsigned.UInt64; | ||
| import tech.pegasys.teku.spec.SpecMilestone; | ||
| import tech.pegasys.teku.spec.TestSpecContext; | ||
| import tech.pegasys.teku.spec.datastructures.blocks.BeaconBlock; | ||
| import tech.pegasys.teku.spec.datastructures.blocks.BlockContainer; | ||
| import tech.pegasys.teku.spec.datastructures.blocks.versions.deneb.BlindedBlockContents; | ||
| import tech.pegasys.teku.spec.datastructures.blocks.versions.deneb.BlockContents; | ||
| import tech.pegasys.teku.spec.networks.Eth2Network; | ||
| import tech.pegasys.teku.validator.remote.typedef.AbstractTypeDefRequestTestBase; | ||
|
|
||
| @TestSpecContext(allMilestones = true, network = Eth2Network.MINIMAL) | ||
| public class ProduceBlockRequestTest extends AbstractTypeDefRequestTestBase { | ||
|
|
||
| private ProduceBlockRequest request; | ||
| private Buffer responseBodyBuffer; | ||
|
|
||
| @BeforeEach | ||
| void setupRequest() { | ||
| request = | ||
| new ProduceBlockRequest(mockWebServer.url("/"), okHttpClient, spec, UInt64.ONE, false); | ||
| responseBodyBuffer = new Buffer(); | ||
| } | ||
|
|
||
| @AfterEach | ||
| void reset() { | ||
| responseBodyBuffer.clear(); | ||
| responseBodyBuffer.close(); | ||
| } | ||
|
|
||
| @TestTemplate | ||
| public void shouldGetUnblindedBeaconBlockAsJson() { | ||
| assumeThat(specMilestone).isLessThan(DENEB); | ||
| final BeaconBlock beaconBlock = dataStructureUtil.randomBeaconBlock(ONE); | ||
| final ProduceBlockRequest.ProduceBlockResponse blockResponse = | ||
| new ProduceBlockRequest.ProduceBlockResponse(beaconBlock); | ||
|
|
||
| final String mockResponse = readExpectedJsonResource(specMilestone, false, false); | ||
|
|
||
| mockWebServer.enqueue(new MockResponse().setResponseCode(SC_OK).setBody(mockResponse)); | ||
|
|
||
| final BLSSignature signature = beaconBlock.getBlock().getBody().getRandaoReveal(); | ||
|
|
||
| final Optional<BlockContainer> maybeBlockContainer = | ||
| request.createUnsignedBlock(signature, Optional.empty()); | ||
|
|
||
| assertThat(maybeBlockContainer).isPresent(); | ||
|
|
||
| assertThat(maybeBlockContainer.get()).isEqualTo(blockResponse.getData()); | ||
| } | ||
|
|
||
| @TestTemplate | ||
| public void shouldGetUnblindedBeaconBlockAsSsz() { | ||
| assumeThat(specMilestone).isLessThan(DENEB); | ||
| final BeaconBlock beaconBlock = dataStructureUtil.randomBeaconBlock(ONE); | ||
| final ProduceBlockRequest.ProduceBlockResponse blockResponse = | ||
| new ProduceBlockRequest.ProduceBlockResponse(beaconBlock); | ||
|
|
||
| responseBodyBuffer.write( | ||
| spec.getGenesisSchemaDefinitions() | ||
| .getBlockContainerSchema() | ||
| .sszSerialize(beaconBlock) | ||
| .toArrayUnsafe()); | ||
|
|
||
| mockWebServer.enqueue( | ||
| new MockResponse() | ||
| .setResponseCode(SC_OK) | ||
| .setHeader("Content-Type", MediaType.OCTET_STREAM) | ||
| .setBody(responseBodyBuffer)); | ||
|
|
||
| final BLSSignature signature = beaconBlock.getBlock().getBody().getRandaoReveal(); | ||
|
|
||
| final Optional<BlockContainer> maybeBlockContainer = | ||
| request.createUnsignedBlock(signature, Optional.empty()); | ||
|
|
||
| assertThat(maybeBlockContainer).isPresent(); | ||
|
|
||
| assertThat(maybeBlockContainer.get()).isEqualTo(blockResponse.getData()); | ||
| } | ||
|
|
||
| @TestTemplate | ||
| public void shouldGetBlindedBeaconBlockAsJson() { | ||
| assumeThat(specMilestone).isGreaterThanOrEqualTo(BELLATRIX).isLessThanOrEqualTo(CAPELLA); | ||
| final BeaconBlock blindedBeaconBlock = dataStructureUtil.randomBlindedBeaconBlock(ONE); | ||
| final ProduceBlockRequest.ProduceBlockResponse blockResponse = | ||
| new ProduceBlockRequest.ProduceBlockResponse(blindedBeaconBlock); | ||
|
|
||
| final String mockResponse = readExpectedJsonResource(specMilestone, true, false); | ||
|
|
||
| mockWebServer.enqueue(new MockResponse().setResponseCode(SC_OK).setBody(mockResponse)); | ||
|
|
||
| final BLSSignature signature = blindedBeaconBlock.getBlock().getBody().getRandaoReveal(); | ||
|
|
||
| final Optional<BlockContainer> maybeBlockContainer = | ||
| request.createUnsignedBlock(signature, Optional.empty()); | ||
|
|
||
| assertThat(maybeBlockContainer).isPresent(); | ||
|
|
||
| assertThat(maybeBlockContainer.get()).isEqualTo(blockResponse.getData()); | ||
| } | ||
|
|
||
| @TestTemplate | ||
| public void shouldGetBlindedBeaconBlockAsSsz() { | ||
| assumeThat(specMilestone).isLessThan(DENEB); | ||
| final BeaconBlock blindedBeaconBlock = dataStructureUtil.randomBlindedBeaconBlock(ONE); | ||
| final ProduceBlockRequest.ProduceBlockResponse blockResponse = | ||
| new ProduceBlockRequest.ProduceBlockResponse(blindedBeaconBlock); | ||
|
|
||
| responseBodyBuffer.write( | ||
| spec.getGenesisSchemaDefinitions() | ||
| .getBlindedBlockContainerSchema() | ||
| .sszSerialize(blindedBeaconBlock) | ||
| .toArrayUnsafe()); | ||
|
|
||
| mockWebServer.enqueue( | ||
| new MockResponse() | ||
| .setResponseCode(SC_OK) | ||
| .setHeader(HEADER_EXECUTION_PAYLOAD_BLINDED, "true") | ||
| .setHeader("Content-Type", MediaType.OCTET_STREAM) | ||
| .setBody(responseBodyBuffer)); | ||
|
|
||
| final BLSSignature signature = blindedBeaconBlock.getBlock().getBody().getRandaoReveal(); | ||
|
|
||
| final Optional<BlockContainer> maybeBlockContainer = | ||
| request.createUnsignedBlock(signature, Optional.empty()); | ||
|
|
||
| assertThat(maybeBlockContainer).isPresent(); | ||
|
|
||
| assertThat(maybeBlockContainer.get()).isEqualTo(blockResponse.getData()); | ||
| } | ||
|
|
||
| @TestTemplate | ||
| public void shouldGetUnblindedBlockContentsPostDenebAsJson() { | ||
| assumeThat(specMilestone).isEqualTo(DENEB); | ||
| final BlockContents blockContents = dataStructureUtil.randomBlockContents(ONE); | ||
| final ProduceBlockRequest.ProduceBlockResponse blockResponse = | ||
| new ProduceBlockRequest.ProduceBlockResponse(blockContents); | ||
|
|
||
| final String mockResponse = readExpectedJsonResource(specMilestone, false, true); | ||
|
|
||
| mockWebServer.enqueue(new MockResponse().setResponseCode(SC_OK).setBody(mockResponse)); | ||
|
|
||
| final BLSSignature signature = blockContents.getBlock().getBody().getRandaoReveal(); | ||
|
|
||
| final Optional<BlockContainer> maybeBlockContainer = | ||
| request.createUnsignedBlock(signature, Optional.empty()); | ||
|
|
||
| assertThat(maybeBlockContainer).isPresent(); | ||
|
|
||
| assertThat(maybeBlockContainer.get()).isEqualTo(blockResponse.getData()); | ||
| } | ||
|
|
||
| @TestTemplate | ||
| public void shouldGetUnblindedBlockContentsPostDenebAsSsz() { | ||
| assumeThat(specMilestone).isEqualTo(DENEB); | ||
| final BlockContents blockContents = dataStructureUtil.randomBlockContents(ONE); | ||
| final ProduceBlockRequest.ProduceBlockResponse blockResponse = | ||
| new ProduceBlockRequest.ProduceBlockResponse(blockContents); | ||
|
|
||
| responseBodyBuffer.write( | ||
| spec.getGenesisSchemaDefinitions() | ||
| .getBlockContainerSchema() | ||
| .sszSerialize(blockContents) | ||
| .toArray()); | ||
|
|
||
| mockWebServer.enqueue( | ||
| new MockResponse() | ||
| .setResponseCode(SC_OK) | ||
| .setHeader("Content-Type", MediaType.OCTET_STREAM) | ||
| .setBody(responseBodyBuffer)); | ||
|
|
||
| final BLSSignature signature = blockContents.getBlock().getBody().getRandaoReveal(); | ||
|
|
||
| final Optional<BlockContainer> maybeBlockContainer = | ||
| request.createUnsignedBlock(signature, Optional.empty()); | ||
|
|
||
| assertThat(maybeBlockContainer).isPresent(); | ||
|
|
||
| assertThat(maybeBlockContainer.get()).isEqualTo(blockResponse.getData()); | ||
| } | ||
|
|
||
| @TestTemplate | ||
| public void shouldGetBlindedBlockContentsPostDenebAsJson() { | ||
| assumeThat(specMilestone).isEqualTo(DENEB); | ||
| final BlindedBlockContents blindedBlockContents = | ||
| dataStructureUtil.randomBlindedBlockContents(ONE); | ||
| final ProduceBlockRequest.ProduceBlockResponse blockResponse = | ||
| new ProduceBlockRequest.ProduceBlockResponse(blindedBlockContents); | ||
|
|
||
| final String mockResponse = readExpectedJsonResource(specMilestone, true, true); | ||
|
|
||
| mockWebServer.enqueue(new MockResponse().setResponseCode(SC_OK).setBody(mockResponse)); | ||
|
|
||
| final BLSSignature signature = blindedBlockContents.getBlock().getBody().getRandaoReveal(); | ||
|
|
||
| final Optional<BlockContainer> maybeBlockContainer = | ||
| request.createUnsignedBlock(signature, Optional.empty()); | ||
|
|
||
| assertThat(maybeBlockContainer).isPresent(); | ||
|
|
||
| assertThat(maybeBlockContainer.get()).isEqualTo(blockResponse.getData()); | ||
| } | ||
|
|
||
| @TestTemplate | ||
| public void shouldGetBlindedBlockContentsPostDenebAsSsz() { | ||
| assumeThat(specMilestone).isEqualTo(DENEB); | ||
| final BlindedBlockContents blindedBlockContents = | ||
| dataStructureUtil.randomBlindedBlockContents(ONE); | ||
| final ProduceBlockRequest.ProduceBlockResponse blockResponse = | ||
| new ProduceBlockRequest.ProduceBlockResponse(blindedBlockContents); | ||
|
|
||
| responseBodyBuffer.write( | ||
| spec.getGenesisSchemaDefinitions() | ||
| .getBlindedBlockContainerSchema() | ||
| .sszSerialize(blindedBlockContents) | ||
| .toArray()); | ||
|
|
||
| mockWebServer.enqueue( | ||
| new MockResponse() | ||
| .setResponseCode(SC_OK) | ||
| .setHeader(HEADER_EXECUTION_PAYLOAD_BLINDED, "true") | ||
| .setHeader("Content-Type", MediaType.OCTET_STREAM) | ||
| .setBody(responseBodyBuffer)); | ||
|
|
||
| final BLSSignature signature = blindedBlockContents.getBlock().getBody().getRandaoReveal(); | ||
|
|
||
| final Optional<BlockContainer> maybeBlockContainer = | ||
| request.createUnsignedBlock(signature, Optional.empty()); | ||
|
|
||
| assertThat(maybeBlockContainer).isPresent(); | ||
|
|
||
| assertThat(maybeBlockContainer.get()).isEqualTo(blockResponse.getData()); | ||
| } | ||
|
|
||
| private String readExpectedJsonResource( | ||
| final SpecMilestone specMilestone, final boolean blinded, final boolean blockContents) { | ||
| final String fileName = | ||
| String.format( | ||
| "new%s%s%s.json", | ||
| blinded ? "Blinded" : "", | ||
| blockContents ? "BlockContents" : "Block", | ||
| specMilestone.name()); | ||
| return readResource("responses/produce_block_responses/" + fileName); | ||
| } | ||
| } |
Oops, something went wrong.