From 3919b129df7cd01af267e64bedebc54736202f2a Mon Sep 17 00:00:00 2001 From: wuwei1972 Date: Mon, 29 Apr 2019 21:07:48 +0800 Subject: [PATCH] =?UTF-8?q?=E6=B7=BB=E5=8A=A0=E5=90=88=E7=BA=A6=E6=B5=81?= =?UTF-8?q?=E7=A8=8B=E6=B5=8B=E8=AF=95=E4=BB=A3=E7=A0=81?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- src/test/scala/rep/sc/BlockStubActor.scala | 2 + src/test/scala/rep/sc/ContractTest.scala | 162 +++++++++++++++++++++ 2 files changed, 164 insertions(+) create mode 100644 src/test/scala/rep/sc/ContractTest.scala diff --git a/src/test/scala/rep/sc/BlockStubActor.scala b/src/test/scala/rep/sc/BlockStubActor.scala index ce344c94..d9c1796a 100644 --- a/src/test/scala/rep/sc/BlockStubActor.scala +++ b/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) diff --git a/src/test/scala/rep/sc/ContractTest.scala b/src/test/scala/rep/sc/ContractTest.scala new file mode 100644 index 00000000..a9318832 --- /dev/null +++ b/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) + + + + } +} +