[ETCM-283] Fix tests

input-output-hk · Nov 19, 2020 · e48c144 · e48c144
1 parent 397d0bd
commit e48c144
Showing 1 changed file with 4 additions and 6 deletions.
diff --git a/src/test/scala/io/iohk/ethereum/blockchain/sync/regular/BlockFetcherStateSpec.scala b/src/test/scala/io/iohk/ethereum/blockchain/sync/regular/BlockFetcherStateSpec.scala
@@ -3,10 +3,7 @@ package io.iohk.ethereum.blockchain.sync.regular
 import akka.actor.ActorSystem
 import akka.testkit.{TestKit, TestProbe}
 import io.iohk.ethereum.Mocks.MockValidatorsAlwaysSucceed
-import io.iohk.ethereum.domain.Block
 import io.iohk.ethereum.BlockHelpers
-import io.iohk.ethereum.Fixtures.Blocks.ValidBlock
-import io.iohk.ethereum.domain.Block
 import io.iohk.ethereum.network.PeerId
 import org.scalatest.matchers.should.Matchers
 import org.scalatest.wordspec.AnyWordSpecLike
@@ -30,11 +27,12 @@ class BlockFetcherStateSpec extends TestKit(ActorSystem()) with AnyWordSpecLike
 
     "handling requested blocks" should {
       "clear headers queue if got empty list of blocks" in {
+        val importer = TestProbe().ref
         val headers = BlockHelpers.generateChain(5, BlockHelpers.genesis).map(_.header)
         val peer = PeerId("foo")
 
         val result = BlockFetcherState
-          .initial(TestProbe().ref, 0)
+          .initial(importer, validators.blockValidator, 0)
           .appendHeaders(headers)
           .map(_.handleRequestedBlocks(peer, List()))
           .map(_.waitingHeaders)
@@ -44,12 +42,12 @@ class BlockFetcherStateSpec extends TestKit(ActorSystem()) with AnyWordSpecLike
       }
 
       "enqueue requested blocks" in {
-
+        val importer = TestProbe().ref
         val blocks = BlockHelpers.generateChain(5, BlockHelpers.genesis)
         val peer = PeerId("foo")
 
         val result = BlockFetcherState
-          .initial(TestProbe().ref, 0)
+          .initial(importer, validators.blockValidator, 0)
           .appendHeaders(blocks.map(_.header))
           .map(_.handleRequestedBlocks(peer, blocks.map(_.body)))