添加合约流程测试代码

src/test/scala/rep/sc/BlockStubActor.scala

@@ -84,6 +84,8 @@ class BlockStubActor(moduleName: String) extends ModuleBase(moduleName) {
     RepLogger.info(RepLogger.Consensus_Logger, this.getLogMsgPrefix("BlockStubActor Start"))
   }
 
+  
+  
   override def receive = {
     case wb: WriteBlockStub =>
      val newblock =  CreateBlock(wb.trans)

src/test/scala/rep/sc/ContractTest.scala

@@ -0,0 +1,162 @@
+package rep.sc
+
+import akka.actor.{ActorSystem,ActorRef}
+import akka.testkit.{TestKit, TestProbe}
+import org.json4s.{DefaultFormats, jackson}
+import org.json4s.native.Serialization.{write, writePretty}
+import org.scalatest.{BeforeAndAfterAll, FlatSpecLike, Matchers}
+import rep.app.system.ClusterSystem
+import rep.app.system.ClusterSystem.InitType
+import rep.network.PeerHelper
+import rep.network.module.ModuleManager
+import rep.protos.peer.{Certificate, ChaincodeId, Signer}
+import rep.sc.TransferSpec.{ACTION, SetMap}
+import rep.sc.tpl.{CertInfo, Transfer}
+import rep.storage.ImpDataAccess
+import rep.utils.SerializeUtils.toJson
+import rep.app.conf.SystemProfile
+
+import scala.concurrent.duration._
+import scala.collection.mutable.Map
+import rep.sc.SandboxDispatcher.DoTransaction
+import rep.protos.peer.Transaction
+
+object ContractTest {
+
+  type SetMap = scala.collection.mutable.Map[String,Int]
+
+  object ACTION {
+    val transfer = "transfer"
+    val set = "set"
+    val SignUpSigner = "SignUpSigner"
+    val SignUpCert = "SignUpCert"
+    val UpdateCertStatus = "UpdateCertStatus"
+    val UpdateSigner = "UpdateSigner"
+  }
+}
+
+class ContractTest(_system: ActorSystem)
+  extends TestKit(_system) with Matchers with FlatSpecLike with BeforeAndAfterAll {
+
+  def this() = this(ActorSystem("TransferSpec", new ClusterSystem("121000005l35120456.node1", InitType.MULTI_INIT, false).getConf))
+
+  override def afterAll: Unit = { shutdown(system) }
+
+  implicit val serialization = jackson.Serialization
+  // or native.Serialization
+  implicit val formats = DefaultFormats
+
+  private def get_ContractAssetsTPL_Deploy_Trans(sysName:String,version:Int) : Transaction= {
+    // 部署资产管理
+    val s1 = scala.io.Source.fromFile("src/main/scala/rep/sc/tpl/ContractAssetsTPL.scala")
+    val l1 = try s1.mkString finally s1.close()
+    
+    val cid1 = ChaincodeId("ContractAssetsTPL",version)
+     //生成deploy交易
+    val t1 = PeerHelper.createTransaction4Deploy(sysName,cid1 ,
+      l1, "",5000, rep.protos.peer.ChaincodeDeploy.CodeType.CODE_SCALA)
+      t1
+  }
+  
+  private def get_ContractCert_Deploy_Trans(sysName:String,version:Int) : Transaction= {
+    val s2 = scala.io.Source.fromFile("src/main/scala/rep/sc/tpl/ContractCert.scala")
+    val l2 = try s2.mkString finally  s2.close()
+    val cid2 =  ChaincodeId(SystemProfile.getAccountChaincodeName,version)
+    val t2 = PeerHelper.createTransaction4Deploy(sysName,cid2,
+      l2, "",5000, rep.protos.peer.ChaincodeDeploy.CodeType.CODE_SCALA)
+      t2
+  }
+  
+  private def createCertTransInvoke(sysName:String,version:Int,action:String,param:String):Transaction={
+     val cid2 =  ChaincodeId(SystemProfile.getAccountChaincodeName,version)
+    PeerHelper.createTransaction4Invoke(sysName,cid2, action, Seq(param))
+  }
+  
+  private def createAssertTransInvoke(sysName:String,version:Int,action:String,param:String):Transaction={
+    val cid2 =  ChaincodeId("ContractAssetsTPL",version)
+    PeerHelper.createTransaction4Invoke(sysName,cid2, action, Seq(param))
+  }
+  
+  private def get_parallelPutProofTPL_Deploy_Trans(sysName:String,version:Int) : Transaction = {
+    val s3 = scala.io.Source.fromFile("src/main/scala/rep/sc/tpl/parallelPutProofTPL.scala")
+    val l3 = try s3.mkString finally  s3.close()
+    val cid3 =  ChaincodeId("parallelPutProofTPL",version)
+    val t3 = PeerHelper.createTransaction4Deploy(sysName,cid3,
+      l3, "",5000, rep.protos.peer.ChaincodeDeploy.CodeType.CODE_SCALA_PARALLEL)
+      t3
+  }
+  
+  private def createParallelTransInvoke(sysName:String,version:Int,action:String,param:String):Transaction={
+    val cid2 =  ChaincodeId("parallelPutProofTPL",version)
+    PeerHelper.createTransaction4Invoke(sysName,cid2, action, Seq(param))
+  }
+  
+  private def ExecuteTrans(probe:TestProbe,sandbox:ActorRef,t:Transaction,snapshotName:String,sendertype:TypeOfSender.Value)={
+    val msg_send1 = DoTransaction(t,snapshotName,sendertype)
+    probe.send(sandbox, msg_send1)
+    val msg_recv1 = probe.expectMsgType[Sandbox.DoTransactionResult](1000.seconds)
+    if(msg_recv1.err.isEmpty){
+      msg_recv1.err.isEmpty should be (true)
+    }else{
+      println(msg_recv1.err.toString())
+      !msg_recv1.err.isEmpty should be (false)
+    }
+  }
+  
+  
+  "ContractAssetsTPL" should "can set assets and transfer from a to b" in {
+    val sysName = "121000005l35120456.node1"
+    val dbTag = "121000005l35120456.node1"
+    //建立PeerManager实例是为了调用transactionCreator(需要用到密钥签名)，无他
+    val pm = system.actorOf(ModuleManager.props("modulemanager", sysName, false, false,false), "modulemanager")
+    
+    val sm: SetMap = Map("121000005l35120456" -> 50, "12110107bi45jh675g" -> 50, "122000002n00123567" -> 50)
+    val sms = write(sm)
+    
+    val tcs = Array(
+          Transfer("121000005l35120456", "12110107bi45jh675g", 5),
+          Transfer("121000005l35120456", "12110107bi45jh675g0", 5),
+           Transfer("121000005l35120456", "12110107bi45jh675g", 500))
+    val rcs = Array(None,  "目标账户不存在", "余额不足")
+    
+    val signer = Signer("node2", "12110107bi45jh675g", "13856789234", Seq("node2"))
+    val cert = scala.io.Source.fromFile("jks/certs/12110107bi45jh675g.node2.cer")
+    val certStr = try cert.mkString finally  cert.close()
+    val certinfo = CertInfo("12110107bi45jh675g", "node2", Certificate(certStr, "SHA1withECDSA", true, None, None) )
+    
+    //准备探针以验证调用返回结果
+    val probe = TestProbe()
+    val sandbox = system.actorOf(TransactionDispatcher.props("transactiondispatcher"),"transactiondispatcher")
+
+    //内存中没有合约，建立合约，此时合约在快照中
+    val t1 = this.get_ContractCert_Deploy_Trans(sysName, 1)
+    ExecuteTrans(probe,sandbox:ActorRef,t1,"dbnumber1",TypeOfSender.FromAPI)
+
+   val t2 =  this.createCertTransInvoke(sysName, 1, ACTION.SignUpSigner, write(signer))
+   ExecuteTrans(probe,sandbox:ActorRef,t2,"dbnumber1",TypeOfSender.FromAPI)
+    
+   //这个应该错误，应该是找不到合约
+   val t3 =  this.createCertTransInvoke(sysName, 1, ACTION.SignUpSigner, write(signer))
+   ExecuteTrans(probe,sandbox:ActorRef,t3,"dbnumber2",TypeOfSender.FromAPI)
+   
+   var t4 = PeerHelper.createTransaction4State(sysName, ChaincodeId(SystemProfile.getAccountChaincodeName,1),false)
+   ExecuteTrans(probe,sandbox:ActorRef,t3,"dbnumber1",TypeOfSender.FromAPI)
+   
+    val t5 =  this.createCertTransInvoke(sysName,1, ACTION.SignUpCert, writePretty(certinfo))
+   ExecuteTrans(probe,sandbox:ActorRef,t5,"dbnumber1",TypeOfSender.FromAPI)
+    
+    //会失败
+    val t6 = this.get_ContractCert_Deploy_Trans(sysName, 1)
+    ExecuteTrans(probe,sandbox:ActorRef,t6,"dbnumber1",TypeOfSender.FromAPI)
+    
+    val t7 = this.get_ContractCert_Deploy_Trans(sysName, 1)
+    ExecuteTrans(probe,sandbox:ActorRef,t7,"dbnumber2",TypeOfSender.FromAPI)
+
+    val t8 = this.get_ContractCert_Deploy_Trans(sysName, 2)
+    ExecuteTrans(probe,sandbox:ActorRef,t6,"dbnumber1",TypeOfSender.FromAPI)
+
+
+     
+  }
+}
+