mirror of https://github.com/hyperledger/besu
Signed-off-by: Matilda Clerke <matilda.clerke@consensys.net>pull/7638/head
Matilda Clerke
2 months ago
parent
c03bffbc93
commit
b2a45c201e
@ -0,0 +1,89 @@ |
||||
/* |
||||
* Copyright contributors to Hyperledger Besu. |
||||
* |
||||
* 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. |
||||
* |
||||
* SPDX-License-Identifier: Apache-2.0 |
||||
*/ |
||||
package org.hyperledger.besu.ethereum.mainnet; |
||||
|
||||
import org.hyperledger.besu.datatypes.Hash; |
||||
import org.hyperledger.besu.ethereum.core.BlockHeader; |
||||
import org.hyperledger.besu.ethereum.core.Request; |
||||
import org.hyperledger.besu.ethereum.core.Transaction; |
||||
import org.hyperledger.besu.ethereum.core.TransactionReceipt; |
||||
import org.hyperledger.besu.ethereum.core.Withdrawal; |
||||
import org.hyperledger.besu.evm.log.LogsBloomFilter; |
||||
|
||||
import java.util.List; |
||||
|
||||
/** A utility class for body validation tasks. Implemented utilising BodyValidation */ |
||||
public class BodyValidator { |
||||
|
||||
/** |
||||
* Generates the transaction root for a list of transactions |
||||
* |
||||
* @param transactions the transactions |
||||
* @return the transaction root |
||||
*/ |
||||
public Hash transactionsRoot(final List<Transaction> transactions) { |
||||
return BodyValidation.transactionsRoot(transactions); |
||||
} |
||||
|
||||
/** |
||||
* Generates the withdrawals root for a list of withdrawals |
||||
* |
||||
* @param withdrawals the transactions |
||||
* @return the transaction root |
||||
*/ |
||||
public Hash withdrawalsRoot(final List<Withdrawal> withdrawals) { |
||||
return BodyValidation.withdrawalsRoot(withdrawals); |
||||
} |
||||
|
||||
/** |
||||
* Generates the requests root for a list of requests |
||||
* |
||||
* @param requests list of request |
||||
* @return the requests root |
||||
*/ |
||||
public Hash requestsRoot(final List<Request> requests) { |
||||
return BodyValidation.requestsRoot(requests); |
||||
} |
||||
|
||||
/** |
||||
* Generates the receipt root for a list of receipts |
||||
* |
||||
* @param receipts the receipts |
||||
* @return the receipt root |
||||
*/ |
||||
public Hash receiptsRoot(final List<TransactionReceipt> receipts) { |
||||
return BodyValidation.receiptsRoot(receipts); |
||||
} |
||||
|
||||
/** |
||||
* Generates the ommers hash for a list of ommer block headers |
||||
* |
||||
* @param ommers the ommer block headers |
||||
* @return the ommers hash |
||||
*/ |
||||
public Hash ommersHash(final List<BlockHeader> ommers) { |
||||
return BodyValidation.ommersHash(ommers); |
||||
} |
||||
|
||||
/** |
||||
* Generates the logs bloom filter for a list of transaction receipts |
||||
* |
||||
* @param receipts the transaction receipts |
||||
* @return the logs bloom filter |
||||
*/ |
||||
public LogsBloomFilter logsBloom(final List<TransactionReceipt> receipts) { |
||||
return BodyValidation.logsBloom(receipts); |
||||
} |
||||
} |
||||
@ -0,0 +1,34 @@ |
||||
/* |
||||
* Copyright contributors to Hyperledger Besu. |
||||
* |
||||
* 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. |
||||
* |
||||
* SPDX-License-Identifier: Apache-2.0 |
||||
*/ |
||||
package org.hyperledger.besu.ethereum.eth.manager.peertask; |
||||
|
||||
import java.lang.reflect.Field; |
||||
|
||||
import org.junit.platform.commons.util.ReflectionUtils; |
||||
|
||||
public class PeerTaskFeatureToggleTestHelper { |
||||
|
||||
public static void setPeerTaskFeatureToggle(final boolean usePeerTaskSystem) |
||||
throws IllegalAccessException { |
||||
Field usePeerTaskSystemField = |
||||
ReflectionUtils.findFields( |
||||
PeerTaskFeatureToggle.class, |
||||
(f) -> f.getName().equals("USE_PEER_TASK_SYSTEM"), |
||||
ReflectionUtils.HierarchyTraversalMode.TOP_DOWN) |
||||
.getFirst(); |
||||
usePeerTaskSystemField.setAccessible(true); |
||||
usePeerTaskSystemField.set(null, usePeerTaskSystem); |
||||
} |
||||
} |
||||
@ -0,0 +1,152 @@ |
||||
/* |
||||
* Copyright contributors to Hyperledger Besu. |
||||
* |
||||
* 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. |
||||
* |
||||
* SPDX-License-Identifier: Apache-2.0 |
||||
*/ |
||||
package org.hyperledger.besu.ethereum.eth.manager.peertask.task; |
||||
|
||||
import org.hyperledger.besu.datatypes.Hash; |
||||
import org.hyperledger.besu.ethereum.core.BlockHeader; |
||||
import org.hyperledger.besu.ethereum.core.TransactionReceipt; |
||||
import org.hyperledger.besu.ethereum.eth.EthProtocol; |
||||
import org.hyperledger.besu.ethereum.eth.manager.peertask.InvalidPeerTaskResponseException; |
||||
import org.hyperledger.besu.ethereum.eth.messages.EthPV63; |
||||
import org.hyperledger.besu.ethereum.eth.messages.GetReceiptsMessage; |
||||
import org.hyperledger.besu.ethereum.eth.messages.ReceiptsMessage; |
||||
import org.hyperledger.besu.ethereum.mainnet.BodyValidator; |
||||
import org.hyperledger.besu.ethereum.p2p.rlpx.wire.MessageData; |
||||
|
||||
import java.util.ArrayList; |
||||
import java.util.Collections; |
||||
import java.util.List; |
||||
import java.util.Map; |
||||
import java.util.Optional; |
||||
|
||||
import org.apache.commons.lang3.StringUtils; |
||||
import org.junit.jupiter.api.Assertions; |
||||
import org.junit.jupiter.api.Test; |
||||
import org.mockito.Mockito; |
||||
|
||||
public class GetReceiptsFromPeerTaskTest { |
||||
|
||||
@Test |
||||
public void testGetSubProtocol() { |
||||
GetReceiptsFromPeerTask task = new GetReceiptsFromPeerTask(Collections.emptyList(), null); |
||||
Assertions.assertEquals(EthProtocol.NAME, task.getSubProtocol()); |
||||
} |
||||
|
||||
@Test |
||||
public void testGetRequiredBlockNumber() { |
||||
GetReceiptsFromPeerTask task = |
||||
new GetReceiptsFromPeerTask( |
||||
List.of(mockBlockHeader(1), mockBlockHeader(2), mockBlockHeader(3)), null); |
||||
Assertions.assertEquals(3, task.getRequiredBlockNumber()); |
||||
} |
||||
|
||||
@Test |
||||
public void testGetRequestMessage() { |
||||
GetReceiptsFromPeerTask task = |
||||
new GetReceiptsFromPeerTask( |
||||
List.of(mockBlockHeader(1), mockBlockHeader(2), mockBlockHeader(3)), null); |
||||
|
||||
MessageData messageData = task.getRequestMessage(); |
||||
GetReceiptsMessage getReceiptsMessage = GetReceiptsMessage.readFrom(messageData); |
||||
|
||||
Assertions.assertEquals(EthPV63.GET_RECEIPTS, getReceiptsMessage.getCode()); |
||||
Iterable<Hash> hashesInMessage = getReceiptsMessage.hashes(); |
||||
List<Hash> expectedHashes = |
||||
List.of( |
||||
Hash.fromHexString(StringUtils.repeat("00", 31) + "11"), |
||||
Hash.fromHexString(StringUtils.repeat("00", 31) + "21"), |
||||
Hash.fromHexString(StringUtils.repeat("00", 31) + "31")); |
||||
List<Hash> actualHashes = new ArrayList<>(); |
||||
hashesInMessage.forEach(actualHashes::add); |
||||
|
||||
Assertions.assertEquals(3, actualHashes.size()); |
||||
Assertions.assertEquals( |
||||
expectedHashes.stream().sorted().toList(), actualHashes.stream().sorted().toList()); |
||||
} |
||||
|
||||
@Test |
||||
public void testParseResponseWithNullResponseMessage() { |
||||
GetReceiptsFromPeerTask task = new GetReceiptsFromPeerTask(Collections.emptyList(), null); |
||||
Assertions.assertThrows(InvalidPeerTaskResponseException.class, () -> task.parseResponse(null)); |
||||
} |
||||
|
||||
@Test |
||||
public void testParseResponseForInvalidResponse() throws InvalidPeerTaskResponseException { |
||||
GetReceiptsFromPeerTask task = |
||||
new GetReceiptsFromPeerTask( |
||||
List.of(mockBlockHeader(1), mockBlockHeader(2), mockBlockHeader(3)), null); |
||||
ReceiptsMessage receiptsMessage = |
||||
ReceiptsMessage.create( |
||||
List.of( |
||||
List.of(new TransactionReceipt(1, 123, Collections.emptyList(), Optional.empty())), |
||||
List.of(new TransactionReceipt(1, 456, Collections.emptyList(), Optional.empty())), |
||||
List.of(new TransactionReceipt(1, 789, Collections.emptyList(), Optional.empty())), |
||||
List.of( |
||||
new TransactionReceipt(1, 101112, Collections.emptyList(), Optional.empty())))); |
||||
|
||||
Assertions.assertThrows( |
||||
InvalidPeerTaskResponseException.class, () -> task.parseResponse(receiptsMessage)); |
||||
} |
||||
|
||||
@Test |
||||
public void testParseResponse() throws InvalidPeerTaskResponseException { |
||||
BodyValidator bodyValidator = Mockito.mock(BodyValidator.class); |
||||
BlockHeader blockHeader1 = mockBlockHeader(1); |
||||
BlockHeader blockHeader2 = mockBlockHeader(2); |
||||
BlockHeader blockHeader3 = mockBlockHeader(3); |
||||
|
||||
GetReceiptsFromPeerTask task = |
||||
new GetReceiptsFromPeerTask( |
||||
List.of(blockHeader1, blockHeader2, blockHeader3), bodyValidator); |
||||
|
||||
TransactionReceipt receiptForBlock1 = |
||||
new TransactionReceipt(1, 123, Collections.emptyList(), Optional.empty()); |
||||
TransactionReceipt receiptForBlock2 = |
||||
new TransactionReceipt(1, 456, Collections.emptyList(), Optional.empty()); |
||||
TransactionReceipt receiptForBlock3 = |
||||
new TransactionReceipt(1, 789, Collections.emptyList(), Optional.empty()); |
||||
ReceiptsMessage receiptsMessage = |
||||
ReceiptsMessage.create( |
||||
List.of( |
||||
List.of(receiptForBlock1), List.of(receiptForBlock2), List.of(receiptForBlock3))); |
||||
|
||||
Mockito.when(bodyValidator.receiptsRoot(List.of(receiptForBlock1))) |
||||
.thenReturn(Hash.fromHexString(StringUtils.repeat("00", 31) + "12")); |
||||
Mockito.when(bodyValidator.receiptsRoot(List.of(receiptForBlock2))) |
||||
.thenReturn(Hash.fromHexString(StringUtils.repeat("00", 31) + "22")); |
||||
Mockito.when(bodyValidator.receiptsRoot(List.of(receiptForBlock3))) |
||||
.thenReturn(Hash.fromHexString(StringUtils.repeat("00", 31) + "32")); |
||||
|
||||
Map<BlockHeader, List<TransactionReceipt>> resultMap = task.parseResponse(receiptsMessage); |
||||
|
||||
Assertions.assertEquals(3, resultMap.size()); |
||||
Assertions.assertEquals(List.of(receiptForBlock1), resultMap.get(blockHeader1)); |
||||
Assertions.assertEquals(List.of(receiptForBlock2), resultMap.get(blockHeader2)); |
||||
Assertions.assertEquals(List.of(receiptForBlock3), resultMap.get(blockHeader3)); |
||||
} |
||||
|
||||
private BlockHeader mockBlockHeader(final long blockNumber) { |
||||
BlockHeader blockHeader = Mockito.mock(BlockHeader.class); |
||||
Mockito.when(blockHeader.getNumber()).thenReturn(blockNumber); |
||||
// second to last hex digit indicates the blockNumber, last hex digit indicates the usage of the
|
||||
// hash
|
||||
Mockito.when(blockHeader.getHash()) |
||||
.thenReturn(Hash.fromHexString(StringUtils.repeat("00", 31) + blockNumber + "1")); |
||||
Mockito.when(blockHeader.getReceiptsRoot()) |
||||
.thenReturn(Hash.fromHexString(StringUtils.repeat("00", 31) + blockNumber + "2")); |
||||
|
||||
return blockHeader; |
||||
} |
||||
} |
||||
Loading…
Reference in new issue