add test for allTransAreValid

fluency03 · Apr 27, 2018 · eceee02 · eceee02
1 parent b2fc78b
commit eceee02
Show file tree

Hide file tree

Showing 3 changed files with 36 additions and 5 deletions.
diff --git a/src/main/scala/com/fluency03/blockchain/core/Block.scala b/src/main/scala/com/fluency03/blockchain/core/Block.scala
@@ -41,10 +41,6 @@ case class Block(header: BlockHeader, transactions: Seq[Transaction], hash: Stri
   def hasValidHeaderHash: Boolean = hash == header.hash
 
   def allTransAreValid(uTxOs: mutable.Map[Outpoint, TxOut]): Boolean = allTransValidOf(transactions, index, uTxOs)
-//    transactions match {
-//    case Nil => false
-//    case init :+ last => validateCoinbaseTx(last, index) && init.forall(tx => tx.isValid(uTxOs))
-//  }
 
   def noDuplicateTxIn(): Boolean = noDuplicateTxInOf(transactions)
 

diff --git a/src/test/scala/com/fluency03/blockchain/CryptoTest.scala b/src/test/scala/com/fluency03/blockchain/CryptoTest.scala
@@ -18,6 +18,8 @@ class CryptoTest extends FlatSpec with Matchers {
     generatePublicKey(pair.getPrivate) shouldEqual pair.getPublic
     recoverPublicKey(publicKeyToHex(pair.getPublic)) shouldEqual pair.getPublic
     recoverPrivateKey(privateKeyToHex(pair.getPrivate)) shouldEqual pair.getPrivate
+    recoverPublicKey(pair.getPublic.toHex) shouldEqual pair.getPublic
+    recoverPrivateKey(pair.getPrivate.toHex) shouldEqual pair.getPrivate
   }
 
 

diff --git a/src/test/scala/com/fluency03/blockchain/core/BlockTest.scala b/src/test/scala/com/fluency03/blockchain/core/BlockTest.scala
@@ -1,14 +1,17 @@
 package com.fluency03.blockchain
 package core
 
+import java.security.KeyPair
+
 import com.fluency03.blockchain.core.BlockHeader.hashOfBlockHeader
-import com.fluency03.blockchain.core.Transaction.createCoinbaseTx
+import com.fluency03.blockchain.core.Transaction.{COINBASE_AMOUNT, createCoinbaseTx, signTxIn, updateUTxOs}
 import org.json4s.JValue
 import org.json4s.JsonAST.{JArray, JInt, JObject, JString}
 import org.json4s.JsonDSL._
 import org.json4s.native.JsonMethods.parse
 import org.scalatest.{FlatSpec, Matchers}
 
+import scala.collection.mutable
 import scala.io.Source
 
 class BlockTest extends FlatSpec with Matchers {
@@ -145,5 +148,35 @@ class BlockTest extends FlatSpec with Matchers {
     parse(newBlock.toString) shouldEqual json
   }
 
+  "allTransAreValid" should "check whether all transactions of a Block are valid." in {
+    val uTxOs: mutable.Map[Outpoint, TxOut] = mutable.Map.empty[Outpoint, TxOut]
+    genesis.allTransAreValid(uTxOs) shouldEqual true
+
+    val pair1 = Crypto.generateKeyPair()
+    val address1 = pair1.getPublic.toHex
+
+    val ts = getCurrentTimestamp
+    val tx: Transaction = Transaction(
+      Seq(TxIn(Outpoint(genesis.transactions.head.id, 0), "")),
+      Seq(TxOut(address1, COINBASE_AMOUNT)),
+      ts)
+
+    val nextBlock = Block.mineNextBlock(genesis, "This is next Block!", ts, genesis.difficulty, Seq(tx))
+    nextBlock.allTransAreValid(uTxOs) shouldEqual false
+
+    val uTxOs2 = updateUTxOs(genesis.transactions, uTxOs.toMap)
+    uTxOs ++= uTxOs2
+    nextBlock.allTransAreValid(uTxOs) shouldEqual false
+
+    val genesisPrivate: String = Source.fromResource("private-key").getLines.mkString
+    val keyPair = new KeyPair(Crypto.recoverPublicKey(genesisMiner), Crypto.recoverPrivateKey(genesisPrivate))
+    val signedTxIns = tx.txIns.map(txIn => signTxIn(tx.id.hex2Bytes, txIn, keyPair, uTxOs)).filter(_.isDefined).map(_.get)
+    signedTxIns.length shouldEqual tx.txIns.length
+    val signedTx = Transaction(signedTxIns, Seq(TxOut(address1, COINBASE_AMOUNT)), ts)
+
+    val validNewBlock = Block.mineNextBlock(genesis, "This is next Block!", ts, genesis.difficulty,
+      Seq(signedTx, createCoinbaseTx(1, address1, ts)))
+    validNewBlock.allTransAreValid(uTxOs) shouldEqual true
+  }
 
 }