[refactor] put all the state in the same place and create a builder for consensus and ledger

Author: Aurélien Richez

src/main/scala/io/iohk/ethereum/jsonrpc/TestService.scala

@@ -10,7 +10,7 @@ import io.iohk.ethereum.{crypto, domain, rlp}
 import io.iohk.ethereum.domain.Block._
 import io.iohk.ethereum.domain.{Account, Address, Block, BlockchainImpl, UInt256}
 import io.iohk.ethereum.ledger._
-import io.iohk.ethereum.testmode.{TestServiceProvider, TestmodeConsensus}
+import io.iohk.ethereum.testmode.{TestModeComponentsProvider, TestmodeConsensus}
 import io.iohk.ethereum.transactions.PendingTransactionsManager
 import io.iohk.ethereum.transactions.PendingTransactionsManager.PendingTransactionsResponse
 import io.iohk.ethereum.utils.{BlockchainConfig, ByteStringUtils, ForkBlockNumbers, Logger}
@@ -108,8 +108,7 @@ class TestService(
     blockchain: BlockchainImpl,
     pendingTransactionsManager: ActorRef,
     consensusConfig: ConsensusConfig,
-    consensus: TestmodeConsensus,
-    testLedgerWrapper: TestServiceProvider,
+    testModeComponentsProvider: TestModeComponentsProvider,
     initialConfig: BlockchainConfig
 )(implicit
     scheduler: Scheduler
@@ -122,6 +121,7 @@ class TestService(
   private var accountAddresses: List[String] = List()
   private var accountRangeOffset = 0
   private var currentConfig: BlockchainConfig = initialConfig
+  private var blockTimestamp: Long = 0
 
   def setChainParams(request: SetChainParamsRequest): ServiceResponse[SetChainParamsResponse] = {
     currentConfig = buildNewConfig(request.chainParams.blockchainParams)
@@ -153,9 +153,6 @@ class TestService(
     storeGenesisAccountCodes(genesisData.alloc)
     storeGenesisAccountStorageData(genesisData.alloc)
 
-    // update test ledger with new config
-    consensus.blockchainConfig = currentConfig
-
     accountAddresses = genesisData.alloc.keys.toList
     accountRangeOffset = 0
 
@@ -217,11 +214,11 @@ class TestService(
   def mineBlocks(request: MineBlocksRequest): ServiceResponse[MineBlocksResponse] = {
     def mineBlock(): Task[Unit] = {
       getBlockForMining(blockchain.getBestBlock().get)
-        .flatMap(blockForMining => testLedgerWrapper.ledger(currentConfig).importBlock(blockForMining.block))
+        .flatMap(blockForMining => testModeComponentsProvider.ledger(currentConfig).importBlock(blockForMining.block))
         .map { res =>
           log.info("Block mining result: " + res)
           pendingTransactionsManager ! PendingTransactionsManager.ClearPendingTransactions
-          consensus.blockTimestamp += 1
+          blockTimestamp += 1
         }
     }
 
@@ -234,7 +231,7 @@ class TestService(
   }
 
   def modifyTimestamp(request: ModifyTimestampRequest): ServiceResponse[ModifyTimestampResponse] = {
-    consensus.blockTimestamp = request.timestamp
+    blockTimestamp = request.timestamp
     ModifyTimestampResponse().rightNow
   }
 
@@ -250,7 +247,8 @@ class TestService(
     Try(decode(request.blockRlp).toBlock) match {
       case Failure(_) => Task.now(Left(JsonRpcError(-1, "block validation failed!", None)))
       case Success(value) =>
-        testLedgerWrapper.ledger(currentConfig)
+        testModeComponentsProvider
+          .ledger(currentConfig)
           .importBlock(value)
           .flatMap(handleResult)
     }
@@ -277,7 +275,9 @@ class TestService(
       .timeout(timeout.duration)
       .onErrorRecover { case _ => PendingTransactionsResponse(Nil) }
       .map { pendingTxs =>
-        consensus.blockGenerator
+        testModeComponentsProvider
+          .consensus(currentConfig, blockTimestamp)
+          .blockGenerator
           .generateBlock(
             parentBlock,
             pendingTxs.pendingTransactions.map(_.stx.tx),

src/main/scala/io/iohk/ethereum/nodebuilder/NodeBuilder.scala

@@ -28,7 +28,7 @@ import io.iohk.ethereum.network.p2p.EthereumMessageDecoder
 import io.iohk.ethereum.network.rlpx.AuthHandshaker
 import io.iohk.ethereum.network.{PeerManagerActor, ServerActor, _}
 import io.iohk.ethereum.ommers.OmmersPool
-import io.iohk.ethereum.testmode.{TestEthBlockServiceWrapper, TestLedgerBuilder, TestmodeConsensusBuilder}
+import io.iohk.ethereum.testmode.{TestEthBlockServiceWrapper, TestModeServiceBuilder, TestmodeConsensusBuilder}
 import io.iohk.ethereum.transactions.{PendingTransactionsManager, TransactionHistoryService}
 import io.iohk.ethereum.utils.Config.SyncConfig
 import io.iohk.ethereum.utils._
@@ -363,14 +363,20 @@ trait TestServiceBuilder {
     with BlockchainConfigBuilder
     with VmBuilder
     with TestmodeConsensusBuilder
-    with TestLedgerBuilder =>
+    with TestModeServiceBuilder =>
 
   lazy val testService =
-    new TestService(blockchain, pendingTransactionsManager, consensusConfig, consensus, testLedgerWrapper, blockchainConfig)(scheduler)
+    new TestService(
+      blockchain,
+      pendingTransactionsManager,
+      consensusConfig,
+      testModeComponentsProvider,
+      blockchainConfig
+    )(scheduler)
 }
 
 trait TestEthBlockServiceBuilder extends EthBlocksServiceBuilder {
-  self: BlockchainBuilder with TestLedgerBuilder with ConsensusBuilder =>
+  self: BlockchainBuilder with TestModeServiceBuilder with ConsensusBuilder =>
   override lazy val ethBlocksService = new TestEthBlockServiceWrapper(blockchain, ledger, consensus)
 }
 

src/main/scala/io/iohk/ethereum/nodebuilder/StdNode.scala

@@ -5,7 +5,7 @@ import io.iohk.ethereum.consensus.StdConsensusBuilder
 import io.iohk.ethereum.metrics.{Metrics, MetricsConfig}
 import io.iohk.ethereum.network.discovery.PeerDiscoveryManager
 import io.iohk.ethereum.network.{PeerManagerActor, ServerActor}
-import io.iohk.ethereum.testmode.{TestLedgerBuilder, TestmodeConsensusBuilder}
+import io.iohk.ethereum.testmode.{TestModeServiceBuilder, TestmodeConsensusBuilder}
 import io.iohk.ethereum.utils.Config
 import scala.concurrent.ExecutionContext.Implicits.global
 import scala.concurrent.Await
@@ -106,7 +106,7 @@ abstract class BaseNode extends Node {
 class StdNode extends BaseNode with StdLedgerBuilder with StdConsensusBuilder
 class TestNode
     extends BaseNode
-    with TestLedgerBuilder
+    with TestModeServiceBuilder
     with TestmodeConsensusBuilder
     with TestServiceBuilder
     with TestEthBlockServiceBuilder

src/main/scala/io/iohk/ethereum/testmode/TestModeComponentsProvider.scala

@@ -0,0 +1,28 @@
+package io.iohk.ethereum.testmode
+
+import io.iohk.ethereum.consensus.difficulty.DifficultyCalculator
+import io.iohk.ethereum.consensus.{Consensus, ConsensusConfig}
+import io.iohk.ethereum.domain.BlockchainImpl
+import io.iohk.ethereum.ledger.Ledger.VMImpl
+import io.iohk.ethereum.ledger.{Ledger, LedgerImpl, StxLedger}
+import io.iohk.ethereum.utils.BlockchainConfig
+import io.iohk.ethereum.utils.Config.SyncConfig
+import monix.execution.Scheduler
+
+/** Provides a ledger instance with modifiable blockchain config (used in test mode). */
+class TestModeComponentsProvider(
+    blockchain: BlockchainImpl,
+    syncConfig: SyncConfig,
+    validationExecutionContext: Scheduler,
+    consensusConfig: ConsensusConfig,
+    difficultyCalculator: DifficultyCalculator,
+    vm: VMImpl
+) {
+
+  def ledger(blockchainConfig: BlockchainConfig): Ledger =
+    new LedgerImpl(blockchain, blockchainConfig, syncConfig, consensus(blockchainConfig), validationExecutionContext)
+  def stxLedger(blockchainConfig: BlockchainConfig): StxLedger =
+    new StxLedger(blockchain, blockchainConfig, consensus(blockchainConfig).blockPreparator)
+  def consensus(blockchainConfig: BlockchainConfig, blockTimestamp: Long = 0) =
+    new TestmodeConsensus(vm, blockchain, blockchainConfig, consensusConfig, difficultyCalculator, blockTimestamp)
+}

src/main/scala/io/iohk/ethereum/testmode/TestLedgerBuilder.scala renamed to src/main/scala/io/iohk/ethereum/testmode/TestModeServiceBuilder.scala

@@ -2,26 +2,36 @@ package io.iohk.ethereum.testmode
 
 import akka.util.ByteString
 import cats.data.NonEmptyList
-import io.iohk.ethereum.consensus.{Consensus, ConsensusBuilder}
+import io.iohk.ethereum.consensus.difficulty.DifficultyCalculator
+import io.iohk.ethereum.consensus.{Consensus, ConsensusBuilder, ConsensusConfigBuilder}
 import io.iohk.ethereum.domain._
 import io.iohk.ethereum.ledger._
 import io.iohk.ethereum.nodebuilder.{ActorSystemBuilder, _}
 import monix.eval.Task
 import monix.execution.Scheduler
 
-trait TestLedgerBuilder extends LedgerBuilder {
+trait TestModeServiceBuilder extends LedgerBuilder {
   self: BlockchainConfigBuilder
     with BlockchainBuilder
     with SyncConfigBuilder
     with ConsensusBuilder
-    with ActorSystemBuilder =>
+    with ActorSystemBuilder
+    with ConsensusConfigBuilder
+    with VmBuilder =>
 
   val scheduler = Scheduler(system.dispatchers.lookup("validation-context"))
 
-  lazy val testLedgerWrapper: TestServiceProvider =
-    new TestServiceProvider(blockchain, syncConfig, consensus, scheduler)
+  lazy val testModeComponentsProvider: TestModeComponentsProvider =
+    new TestModeComponentsProvider(
+      blockchain,
+      syncConfig,
+      scheduler,
+      consensusConfig,
+      DifficultyCalculator(blockchainConfig),
+      vm
+    )
 
-  private def testLedger: Ledger = testLedgerWrapper.ledger(blockchainConfig)
+  private def testLedger: Ledger = testModeComponentsProvider.ledger(blockchainConfig)
 
   class TestLedgerProxy extends Ledger {
     override def consensus: Consensus = testLedger.consensus
@@ -35,5 +45,5 @@ trait TestLedgerBuilder extends LedgerBuilder {
   }
 
   override lazy val ledger: Ledger = new TestLedgerProxy
-  override lazy val stxLedger: StxLedger = testLedgerWrapper.stxLedger(blockchainConfig)
+  override lazy val stxLedger: StxLedger = testModeComponentsProvider.stxLedger(blockchainConfig)
 }

src/main/scala/io/iohk/ethereum/testmode/TestServiceProvider.scala

@@ -20,10 +20,10 @@ import io.iohk.ethereum.consensus.pow.validators.ValidatorsExecutor
 class TestmodeConsensus(
     override val vm: VMImpl,
     blockchain: BlockchainImpl,
-    var blockchainConfig: BlockchainConfig,
+    blockchainConfig: BlockchainConfig,
     consensusConfig: ConsensusConfig,
     override val difficultyCalculator: DifficultyCalculator,
-    var blockTimestamp: Long = 0
+    blockTimestamp: Long = 0
 ) // var, because it can be modified by test_ RPC endpoints
     extends Consensus {