add some tests

fluency03 · May 4, 2018 · 2615242 · 2615242
1 parent f22c5de
commit 2615242
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/src/test/scala/com/fluency03/blockchain/api/actors/BlockPoolActorTest.scala b/src/test/scala/com/fluency03/blockchain/api/actors/BlockPoolActorTest.scala
@@ -42,7 +42,10 @@ class BlockPoolActorTest extends TestKit(ActorSystem("BlocksActorTest")) with Im
       expectMsg(None)
 
       blockPoolActor ! GetTxOfBlock(Block.genesisBlock.transactions.head.id, "bb")
-      expectMsg(Some(Block.genesisBlock.transactions.head))
+      expectMsg(None)
+
+      blockPoolActor ! GetTxOfBlock(Block.genesisBlock.transactions.head.id, Block.genesisBlock.hash)
+      expectMsg(Some(Block.genesisBlock))
 
       blockPoolActor ! GetBlocks(Set(Block.genesisBlock.hash))
       expectMsg(Seq(Block.genesisBlock))