mirror of https://github.com/hyperledger/besu
Ibft pantheon controller (#461)
Jason Frame
6 years ago
committed by
GitHub
parent
2fc38e7373
commit
7f0058066c
@ -1,40 +0,0 @@ |
||||
/* |
||||
* Copyright 2018 ConsenSys AG. |
||||
* |
||||
* 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.pantheon.consensus.ibft; |
||||
|
||||
import tech.pegasys.pantheon.consensus.ibft.blockcreation.IbftBlockCreatorFactory; |
||||
import tech.pegasys.pantheon.consensus.ibft.ibftevent.IbftEvent; |
||||
|
||||
/** Stateful evaluator for ibft events */ |
||||
public class IbftStateMachine { |
||||
|
||||
private final IbftBlockCreatorFactory blockCreatorFactory; |
||||
|
||||
public IbftStateMachine(final IbftBlockCreatorFactory blockCreatorFactory) { |
||||
this.blockCreatorFactory = blockCreatorFactory; |
||||
} |
||||
|
||||
/** |
||||
* Attempt to consume the event and update the maintained state |
||||
* |
||||
* @param event the external action that has occurred |
||||
* @param roundTimer timer that will fire expiry events that are expected to be received back into |
||||
* this machine |
||||
* @return whether this event was consumed or requires reprocessing later once the state machine |
||||
* catches up |
||||
*/ |
||||
public boolean processEvent(final IbftEvent event, final RoundTimer roundTimer) { |
||||
// TODO: don't just discard the event, do some logic
|
||||
return true; |
||||
} |
||||
} |
||||
@ -0,0 +1,108 @@ |
||||
/* |
||||
* Copyright 2018 ConsenSys AG. |
||||
* |
||||
* 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.pantheon.consensus.ibft.blockcreation; |
||||
|
||||
import static org.assertj.core.api.Assertions.assertThat; |
||||
import static org.mockito.Mockito.verify; |
||||
import static org.mockito.Mockito.when; |
||||
|
||||
import tech.pegasys.pantheon.consensus.ibft.IbftEventQueue; |
||||
import tech.pegasys.pantheon.consensus.ibft.IbftProcessor; |
||||
import tech.pegasys.pantheon.consensus.ibft.ibftevent.NewChainHead; |
||||
import tech.pegasys.pantheon.ethereum.chain.BlockAddedEvent; |
||||
import tech.pegasys.pantheon.ethereum.chain.Blockchain; |
||||
import tech.pegasys.pantheon.ethereum.core.Block; |
||||
import tech.pegasys.pantheon.ethereum.core.BlockBody; |
||||
import tech.pegasys.pantheon.ethereum.core.BlockHeader; |
||||
import tech.pegasys.pantheon.ethereum.core.Wei; |
||||
import tech.pegasys.pantheon.util.bytes.BytesValue; |
||||
|
||||
import java.util.concurrent.TimeUnit; |
||||
|
||||
import org.assertj.core.util.Lists; |
||||
import org.junit.Before; |
||||
import org.junit.Test; |
||||
import org.junit.runner.RunWith; |
||||
import org.mockito.Mock; |
||||
import org.mockito.junit.MockitoJUnitRunner; |
||||
|
||||
@RunWith(MockitoJUnitRunner.class) |
||||
public class IbftMiningCoordinatorTest { |
||||
@Mock private IbftProcessor ibftProcessor; |
||||
@Mock private IbftBlockCreatorFactory ibftBlockCreatorFactory; |
||||
@Mock private Blockchain blockChain; |
||||
@Mock private Block block; |
||||
@Mock private BlockBody blockBody; |
||||
@Mock private BlockHeader blockHeader; |
||||
private final IbftEventQueue eventQueue = new IbftEventQueue(); |
||||
private IbftMiningCoordinator ibftMiningCoordinator; |
||||
|
||||
@Before |
||||
public void setup() { |
||||
ibftMiningCoordinator = |
||||
new IbftMiningCoordinator(ibftProcessor, ibftBlockCreatorFactory, blockChain, eventQueue); |
||||
when(block.getBody()).thenReturn(blockBody); |
||||
when(block.getHeader()).thenReturn(blockHeader); |
||||
when(blockBody.getTransactions()).thenReturn(Lists.emptyList()); |
||||
} |
||||
|
||||
@Test |
||||
public void enablesMining() { |
||||
ibftMiningCoordinator.enable(); |
||||
verify(ibftProcessor).start(); |
||||
} |
||||
|
||||
@Test |
||||
public void disablesMining() { |
||||
ibftMiningCoordinator.disable(); |
||||
verify(ibftProcessor).stop(); |
||||
} |
||||
|
||||
@Test |
||||
public void setsMinTransactionGasPrice() { |
||||
final Wei minGasPrice = Wei.of(10); |
||||
ibftMiningCoordinator.setMinTransactionGasPrice(minGasPrice); |
||||
verify(ibftBlockCreatorFactory).setMinTransactionGasPrice(minGasPrice); |
||||
} |
||||
|
||||
@Test |
||||
public void getsMinTransactionGasPrice() { |
||||
final Wei minGasPrice = Wei.of(10); |
||||
when(ibftBlockCreatorFactory.getMinTransactionGasPrice()).thenReturn(minGasPrice); |
||||
assertThat(ibftMiningCoordinator.getMinTransactionGasPrice()).isEqualTo(minGasPrice); |
||||
} |
||||
|
||||
@Test |
||||
public void setsTheExtraData() { |
||||
final BytesValue extraData = BytesValue.fromHexStringLenient("0x1234"); |
||||
ibftMiningCoordinator.setExtraData(extraData); |
||||
verify(ibftBlockCreatorFactory).setExtraData(extraData); |
||||
} |
||||
|
||||
@Test |
||||
public void addsNewChainHeadEventWhenNewCanonicalHeadBlockEventReceived() throws Exception { |
||||
BlockAddedEvent headAdvancement = BlockAddedEvent.createForHeadAdvancement(block); |
||||
ibftMiningCoordinator.onBlockAdded(headAdvancement, blockChain); |
||||
|
||||
assertThat(eventQueue.size()).isEqualTo(1); |
||||
final NewChainHead ibftEvent = (NewChainHead) eventQueue.poll(1, TimeUnit.SECONDS); |
||||
assertThat(ibftEvent.getNewChainHeadHeader()).isEqualTo(blockHeader); |
||||
} |
||||
|
||||
@Test |
||||
public void doesntAddNewChainHeadEventWhenNotACanonicalHeadBlockEvent() { |
||||
final BlockAddedEvent fork = BlockAddedEvent.createForFork(block); |
||||
ibftMiningCoordinator.onBlockAdded(fork, blockChain); |
||||
assertThat(eventQueue.isEmpty()).isTrue(); |
||||
} |
||||
} |
||||
Loading…
Reference in new issue