mirror of https://github.com/hyperledger/besu
Add eth protocol manager test builder (#7954)
* 7311: Add EthProtocolManagerTestBuilder Signed-off-by: Matilda Clerke <matilda.clerke@consensys.net> * 7311: Add EthProtocolManagerTestBuilder.builder method Signed-off-by: Matilda Clerke <matilda.clerke@consensys.net> * 7311: Replace EthProtocolManagerTestUtil.create() usages with builder Signed-off-by: Matilda Clerke <matilda.clerke@consensys.net> * 7311: Replace EthProtocolManagerTestUtil.create(TimeoutPolicy) Signed-off-by: Matilda Clerke <matilda.clerke@consensys.net> * 7311: Replace EthProtocolManagerTestUtil.create(Blockchain) Signed-off-by: Matilda Clerke <matilda.clerke@consensys.net> * 7311: Replace EthProtocolManagerTestUtil.create(EthScheduler) Signed-off-by: Matilda Clerke <matilda.clerke@consensys.net> * Replace EthProtocolManagerTestUtil.create(ProtocolSchedule, Blockchain, TimeoutPolicy, WorldStateArchive, TransactionPool, EthProtocolConfiguration) Signed-off-by: Matilda Clerke <matilda.clerke@consensys.net> * Replace EthProtocolManagerTestUtil.create(ProtocolSchedule, Blockchain, EthScheduler) Signed-off-by: Matilda Clerke <matilda.clerke@consensys.net> * Replace EthProtocolManagerTestUtil.create(ProtocolSchedule, Blockchain, EthScheduler) Signed-off-by: Matilda Clerke <matilda.clerke@consensys.net> * Replace EthProtocolManagerTestUtil.create(ProtocolSchedule, Blockchain, WorldStateArchive, TransactionPool, EthProtocolConfiguration) Signed-off-by: Matilda Clerke <matilda.clerke@consensys.net> * Replace EthProtocolManagerTestUtil.create(ProtocolSchedule, Blockchain, EthScheduler, WorldStateArchive, TransactionPool, EthProtocolConfiguration) Signed-off-by: Matilda Clerke <matilda.clerke@consensys.net> * Replace EthProtocolManagerTestUtil.create(ProtocolSchedule, Blockchain, EthScheduler, WorldStateArchive, TransactionPool, EthProtocolConfiguration) Signed-off-by: Matilda Clerke <matilda.clerke@consensys.net> * Remove remaining create methods and apply spotless Signed-off-by: Matilda Clerke <matilda.clerke@consensys.net> --------- Signed-off-by: Matilda Clerke <matilda.clerke@consensys.net>pull/7871/merge
Matilda-Clerke
2 days ago
committed by
GitHub
parent
c320724d29
commit
747a378017
@ -0,0 +1,245 @@ |
|||||||
|
/* |
||||||
|
* Copyright contributors to 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; |
||||||
|
|
||||||
|
import static org.hyperledger.besu.ethereum.core.InMemoryKeyValueStorageProvider.createInMemoryBlockchain; |
||||||
|
import static org.mockito.Mockito.mock; |
||||||
|
|
||||||
|
import org.hyperledger.besu.config.GenesisConfigFile; |
||||||
|
import org.hyperledger.besu.ethereum.chain.Blockchain; |
||||||
|
import org.hyperledger.besu.ethereum.chain.GenesisState; |
||||||
|
import org.hyperledger.besu.ethereum.core.BlockchainSetupUtil; |
||||||
|
import org.hyperledger.besu.ethereum.core.ProtocolScheduleFixture; |
||||||
|
import org.hyperledger.besu.ethereum.eth.EthProtocolConfiguration; |
||||||
|
import org.hyperledger.besu.ethereum.eth.peervalidation.PeerValidator; |
||||||
|
import org.hyperledger.besu.ethereum.eth.sync.SyncMode; |
||||||
|
import org.hyperledger.besu.ethereum.eth.sync.SynchronizerConfiguration; |
||||||
|
import org.hyperledger.besu.ethereum.eth.transactions.TransactionPool; |
||||||
|
import org.hyperledger.besu.ethereum.forkid.ForkIdManager; |
||||||
|
import org.hyperledger.besu.ethereum.mainnet.ProtocolSchedule; |
||||||
|
import org.hyperledger.besu.ethereum.worldstate.WorldStateArchive; |
||||||
|
import org.hyperledger.besu.metrics.noop.NoOpMetricsSystem; |
||||||
|
import org.hyperledger.besu.plugin.services.storage.DataStorageFormat; |
||||||
|
import org.hyperledger.besu.testutil.DeterministicEthScheduler; |
||||||
|
import org.hyperledger.besu.testutil.TestClock; |
||||||
|
|
||||||
|
import java.math.BigInteger; |
||||||
|
import java.util.Collections; |
||||||
|
import java.util.List; |
||||||
|
import java.util.Optional; |
||||||
|
|
||||||
|
import org.apache.tuweni.bytes.Bytes; |
||||||
|
|
||||||
|
public class EthProtocolManagerTestBuilder { |
||||||
|
private static final BigInteger DEFAULT_NETWORK_ID = BigInteger.ONE; |
||||||
|
private static final ProtocolSchedule DEFAULT_PROTOCOL_SCHEDULE = ProtocolScheduleFixture.MAINNET; |
||||||
|
|
||||||
|
private ProtocolSchedule protocolSchedule; |
||||||
|
private GenesisConfigFile genesisConfigFile; |
||||||
|
private GenesisState genesisState; |
||||||
|
private Blockchain blockchain; |
||||||
|
private BigInteger networkId; |
||||||
|
private WorldStateArchive worldStateArchive; |
||||||
|
private TransactionPool transactionPool; |
||||||
|
private EthProtocolConfiguration ethereumWireProtocolConfiguration; |
||||||
|
private ForkIdManager forkIdManager; |
||||||
|
private EthPeers ethPeers; |
||||||
|
private EthMessages ethMessages; |
||||||
|
private EthMessages snapMessages; |
||||||
|
private EthScheduler ethScheduler; |
||||||
|
private EthContext ethContext; |
||||||
|
private List<PeerValidator> peerValidators; |
||||||
|
private Optional<MergePeerFilter> mergePeerFilter; |
||||||
|
private SynchronizerConfiguration synchronizerConfiguration; |
||||||
|
|
||||||
|
public static EthProtocolManagerTestBuilder builder() { |
||||||
|
return new EthProtocolManagerTestBuilder(); |
||||||
|
} |
||||||
|
|
||||||
|
public EthProtocolManagerTestBuilder setProtocolSchedule( |
||||||
|
final ProtocolSchedule protocolSchedule) { |
||||||
|
this.protocolSchedule = protocolSchedule; |
||||||
|
return this; |
||||||
|
} |
||||||
|
|
||||||
|
public EthProtocolManagerTestBuilder setGenesisConfigFile( |
||||||
|
final GenesisConfigFile genesisConfigFile) { |
||||||
|
this.genesisConfigFile = genesisConfigFile; |
||||||
|
return this; |
||||||
|
} |
||||||
|
|
||||||
|
public EthProtocolManagerTestBuilder setGenesisState(final GenesisState genesisState) { |
||||||
|
this.genesisState = genesisState; |
||||||
|
return this; |
||||||
|
} |
||||||
|
|
||||||
|
public EthProtocolManagerTestBuilder setBlockchain(final Blockchain blockchain) { |
||||||
|
this.blockchain = blockchain; |
||||||
|
return this; |
||||||
|
} |
||||||
|
|
||||||
|
public EthProtocolManagerTestBuilder setNetworkId(final BigInteger networkId) { |
||||||
|
this.networkId = networkId; |
||||||
|
return this; |
||||||
|
} |
||||||
|
|
||||||
|
public EthProtocolManagerTestBuilder setWorldStateArchive( |
||||||
|
final WorldStateArchive worldStateArchive) { |
||||||
|
this.worldStateArchive = worldStateArchive; |
||||||
|
return this; |
||||||
|
} |
||||||
|
|
||||||
|
public EthProtocolManagerTestBuilder setTransactionPool(final TransactionPool transactionPool) { |
||||||
|
this.transactionPool = transactionPool; |
||||||
|
return this; |
||||||
|
} |
||||||
|
|
||||||
|
public EthProtocolManagerTestBuilder setEthereumWireProtocolConfiguration( |
||||||
|
final EthProtocolConfiguration ethereumWireProtocolConfiguration) { |
||||||
|
this.ethereumWireProtocolConfiguration = ethereumWireProtocolConfiguration; |
||||||
|
return this; |
||||||
|
} |
||||||
|
|
||||||
|
public EthProtocolManagerTestBuilder setForkIdManager(final ForkIdManager forkIdManager) { |
||||||
|
this.forkIdManager = forkIdManager; |
||||||
|
return this; |
||||||
|
} |
||||||
|
|
||||||
|
public EthProtocolManagerTestBuilder setEthPeers(final EthPeers ethPeers) { |
||||||
|
this.ethPeers = ethPeers; |
||||||
|
return this; |
||||||
|
} |
||||||
|
|
||||||
|
public EthProtocolManagerTestBuilder setEthMessages(final EthMessages ethMessages) { |
||||||
|
this.ethMessages = ethMessages; |
||||||
|
return this; |
||||||
|
} |
||||||
|
|
||||||
|
public EthProtocolManagerTestBuilder setSnapMessages(final EthMessages snapMessages) { |
||||||
|
this.snapMessages = snapMessages; |
||||||
|
return this; |
||||||
|
} |
||||||
|
|
||||||
|
public EthProtocolManagerTestBuilder setEthContext(final EthContext ethContext) { |
||||||
|
this.ethContext = ethContext; |
||||||
|
return this; |
||||||
|
} |
||||||
|
|
||||||
|
public EthProtocolManagerTestBuilder setPeerValidators(final List<PeerValidator> peerValidators) { |
||||||
|
this.peerValidators = peerValidators; |
||||||
|
return this; |
||||||
|
} |
||||||
|
|
||||||
|
public EthProtocolManagerTestBuilder setMergePeerFilter( |
||||||
|
final Optional<MergePeerFilter> mergePeerFilter) { |
||||||
|
this.mergePeerFilter = mergePeerFilter; |
||||||
|
return this; |
||||||
|
} |
||||||
|
|
||||||
|
public EthProtocolManagerTestBuilder setSynchronizerConfiguration( |
||||||
|
final SynchronizerConfiguration synchronizerConfiguration) { |
||||||
|
this.synchronizerConfiguration = synchronizerConfiguration; |
||||||
|
return this; |
||||||
|
} |
||||||
|
|
||||||
|
public EthProtocolManagerTestBuilder setEthScheduler(final EthScheduler ethScheduler) { |
||||||
|
this.ethScheduler = ethScheduler; |
||||||
|
return this; |
||||||
|
} |
||||||
|
|
||||||
|
public EthProtocolManager build() { |
||||||
|
if (protocolSchedule == null) { |
||||||
|
protocolSchedule = DEFAULT_PROTOCOL_SCHEDULE; |
||||||
|
} |
||||||
|
if (genesisConfigFile == null) { |
||||||
|
genesisConfigFile = GenesisConfigFile.mainnet(); |
||||||
|
} |
||||||
|
if (genesisState == null) { |
||||||
|
genesisState = GenesisState.fromConfig(genesisConfigFile, protocolSchedule); |
||||||
|
} |
||||||
|
if (blockchain == null) { |
||||||
|
blockchain = createInMemoryBlockchain(genesisState.getBlock()); |
||||||
|
} |
||||||
|
if (networkId == null) { |
||||||
|
networkId = DEFAULT_NETWORK_ID; |
||||||
|
} |
||||||
|
if (worldStateArchive == null) { |
||||||
|
worldStateArchive = |
||||||
|
BlockchainSetupUtil.forTesting(DataStorageFormat.FOREST).getWorldArchive(); |
||||||
|
} |
||||||
|
if (transactionPool == null) { |
||||||
|
transactionPool = mock(TransactionPool.class); |
||||||
|
} |
||||||
|
if (ethereumWireProtocolConfiguration == null) { |
||||||
|
ethereumWireProtocolConfiguration = EthProtocolConfiguration.defaultConfig(); |
||||||
|
} |
||||||
|
if (forkIdManager == null) { |
||||||
|
forkIdManager = |
||||||
|
new ForkIdManager(blockchain, Collections.emptyList(), Collections.emptyList(), false); |
||||||
|
} |
||||||
|
if (ethPeers == null) { |
||||||
|
ethPeers = |
||||||
|
new EthPeers( |
||||||
|
() -> protocolSchedule.getByBlockHeader(blockchain.getChainHeadHeader()), |
||||||
|
TestClock.fixed(), |
||||||
|
new NoOpMetricsSystem(), |
||||||
|
EthProtocolConfiguration.DEFAULT_MAX_MESSAGE_SIZE, |
||||||
|
Collections.emptyList(), |
||||||
|
Bytes.random(64), |
||||||
|
25, |
||||||
|
25, |
||||||
|
false, |
||||||
|
SyncMode.FAST, |
||||||
|
forkIdManager); |
||||||
|
} |
||||||
|
ethPeers.setChainHeadTracker(EthProtocolManagerTestUtil.getChainHeadTrackerMock()); |
||||||
|
if (ethMessages == null) { |
||||||
|
ethMessages = new EthMessages(); |
||||||
|
} |
||||||
|
if (snapMessages == null) { |
||||||
|
snapMessages = new EthMessages(); |
||||||
|
} |
||||||
|
if (ethScheduler == null) { |
||||||
|
ethScheduler = |
||||||
|
new DeterministicEthScheduler(DeterministicEthScheduler.TimeoutPolicy.NEVER_TIMEOUT); |
||||||
|
} |
||||||
|
if (ethContext == null) { |
||||||
|
ethContext = new EthContext(ethPeers, ethMessages, snapMessages, ethScheduler); |
||||||
|
} |
||||||
|
if (peerValidators == null) { |
||||||
|
peerValidators = Collections.emptyList(); |
||||||
|
} |
||||||
|
if (mergePeerFilter == null) { |
||||||
|
mergePeerFilter = Optional.of(new MergePeerFilter()); |
||||||
|
} |
||||||
|
if (synchronizerConfiguration == null) { |
||||||
|
synchronizerConfiguration = SynchronizerConfiguration.builder().build(); |
||||||
|
} |
||||||
|
return new EthProtocolManager( |
||||||
|
blockchain, |
||||||
|
networkId, |
||||||
|
worldStateArchive, |
||||||
|
transactionPool, |
||||||
|
ethereumWireProtocolConfiguration, |
||||||
|
ethPeers, |
||||||
|
ethMessages, |
||||||
|
ethContext, |
||||||
|
peerValidators, |
||||||
|
mergePeerFilter, |
||||||
|
synchronizerConfiguration, |
||||||
|
ethScheduler, |
||||||
|
forkIdManager); |
||||||
|
} |
||||||
|
} |
||||||
Loading…
Reference in new issue