Merge pull request #82 from soramitsu/feature/user-flow

create user flow scn

src/main/kotlin/jp/co/soramitsu/load/UserFlow.kt

@@ -0,0 +1,158 @@
+package jp.co.soramitsu.load
+
+import io.gatling.javaapi.core.CoreDsl.csv
+import io.gatling.javaapi.core.CoreDsl.scenario
+import io.gatling.javaapi.core.ScenarioBuilder
+import io.ktor.util.Identity.encode
+import jp.co.soramitsu.iroha2.*
+import jp.co.soramitsu.iroha2.client.Iroha2Client
+import jp.co.soramitsu.iroha2.generated.*
+import jp.co.soramitsu.iroha2.query.QueryBuilder
+import jp.co.soramitsu.load.infrastructure.config.SimulationConfig
+import jp.co.soramitsu.load.toolbox.Wrench13
+import kotlinx.coroutines.runBlocking
+import java.security.KeyPair
+import java.util.*
+
+class UserFlow : Wrench13() {
+    lateinit var domainIdSender: DomainId
+    lateinit var anotherDevAccountIdSender: AccountId
+    lateinit var anotherDevAssetIdSender: AssetId
+    lateinit var targetDevAccountIdReceiver: AccountId
+    lateinit var anotherDevKeyPairSender: KeyPair
+    lateinit var signature: PublicKey
+    lateinit var iroha2Client: Iroha2Client
+    lateinit var txs: List<TransactionQueryOutput>
+    lateinit var hash: ByteArray
+
+    companion object {
+        @JvmStatic
+        fun apply() = runBlocking {
+            UserFlow().applyScn()
+        }
+    }
+
+    fun applyScn(): ScenarioBuilder {
+        return userFlowScn
+    }
+
+    val userFlowScn = scenario("userFlowScn")
+        .feed(csv("preconditionList.csv").circular())
+        .exec { Session ->
+            iroha2Client = buildClient(SimulationConfig.simulation.configuration())
+            anotherDevKeyPairSender = adminKeyPair
+            domainIdSender = Session.get<String>("domainIdSender")!!.asDomainId()
+            anotherDevAccountIdSender = Session.get<String>("anotherDevAccountIdSender")!!.asAccountId()
+            targetDevAccountIdReceiver = Session.get<String>("anotherDevAccountIdReceiver")!!.asAccountId()
+            anotherDevAssetIdSender = Session.get<String>("anotherDevAssetIdSender")!!.asAssetId()
+            Session
+        }.exec { Session ->
+            runBlocking {
+                QueryBuilder.findAssetsByAccountId(anotherDevAccountIdSender)
+                    .account(anotherDevAccountIdSender)
+                    .buildSigned(anotherDevKeyPairSender)
+                    .let { query ->
+                        /*println("QUERY BODY")
+                        println(query)*/
+                        iroha2Client.sendQuery(query)
+                    }
+            }
+            Session
+        }.exec { Session ->
+            runBlocking {
+                QueryBuilder.findAccountsByDomainId(domainIdSender)
+                    .account(anotherDevAccountIdSender)
+                    .buildSigned(anotherDevKeyPairSender)
+                    .let { query ->
+                        /*println("QUERY BODY")
+                        println(query)*/
+                        iroha2Client.sendQuery(query)
+                    }
+            }
+            Session
+        }.exec { Session ->
+            runBlocking {
+                QueryBuilder.findAllAssets()
+                    .account(anotherDevAccountIdSender)
+                    .buildSigned(anotherDevKeyPairSender)
+                    .let { query ->
+                        /*println("QUERY BODY")
+                        println(query)*/
+                        iroha2Client.sendQuery(query)
+                    }
+            }
+            Session
+        }.exec { Session ->
+            runBlocking {
+                QueryBuilder.findAllAssets()
+                    .account(anotherDevAccountIdSender)
+                    .buildSigned(anotherDevKeyPairSender)
+                    .let { query ->
+                        /*println("QUERY BODY")
+                        println(query)*/
+                        iroha2Client.sendQuery(query)
+                    }
+            }
+            Session
+        }.exec { Session ->
+            runBlocking {
+                val assetDefinition = "xor_" + UUID.randomUUID() + "_" + UUID.randomUUID()
+                val assetDefinitionId = AssetDefinitionId(
+                    domainIdSender,
+                    assetDefinition.asName()
+                )
+                iroha2Client.sendTransaction {
+                    account(anotherDevAccountIdSender)
+                    registerAssetDefinition(assetDefinitionId, AssetValueType.numeric())
+                    buildSigned(anotherDevKeyPairSender)
+                }
+            }
+            Session
+        }.exec { Session ->
+            runBlocking {
+                txs = QueryBuilder.findTransactionsByAccountId(anotherDevAccountIdSender)
+                    .account(anotherDevAccountIdSender)
+                    .buildSigned(anotherDevKeyPairSender)
+                    .let { query ->
+                        iroha2Client.sendQuery(query)
+                    }
+
+                hash = SignedTransaction.encode(txs[2].transaction.value).hash()
+            }
+            Session
+        }.exec { Session ->
+            runBlocking {
+                val assetDefinition = "xor_" + UUID.randomUUID() + "_" + UUID.randomUUID()
+                val assetDefinitionId = AssetDefinitionId(
+                    domainIdSender,
+                    assetDefinition.asName()
+                )
+                iroha2Client.sendTransaction {
+                    account(anotherDevAccountIdSender)
+                    registerAssetDefinition(assetDefinitionId, AssetValueType.numeric())
+                    buildSigned(anotherDevKeyPairSender)
+                }
+            }
+            Session
+        }.exec { Session ->
+            runBlocking {
+                QueryBuilder.findAllTransactions()
+                    .account(anotherDevAccountIdSender)
+                    .buildSigned(anotherDevKeyPairSender)
+                    .let { query ->
+                        iroha2Client.sendQuery(query)
+                    }
+            }
+            Session
+        }.exec { Session ->
+            runBlocking {
+                QueryBuilder.findTransactionByHash(hash)
+                    .account(anotherDevAccountIdSender)
+                    .buildSigned(anotherDevKeyPairSender)
+                    .let { query ->
+                        iroha2Client.sendQuery(query)
+                    }
+            }
+            Session
+        }
+}

src/test/java/jp/co/soramitsu/load/simulation/transaction/PerformanceSimulation.java

@@ -7,9 +7,7 @@
 public class PerformanceSimulation extends Simulation {
     {
         setUp(
-                //jp.co.soramitsu.load.SetUp.Companion.apply().injectOpen(OpenInjectionStep.atOnceUsers(1))
-                        TransferAssets.Companion.apply().injectOpen(LoadProfiles.getMaxPerformanceOpenProfile())
-                        //.andThen(TransferAssets.Companion.apply().injectOpen(LoadProfiles.getStabilityOpenProfile()))
+                UserFlow.Companion.apply().injectOpen(LoadProfiles.getMaxPerformanceOpenProfile())
                         .andThen(CleanUp.Companion.apply().injectOpen(OpenInjectionStep.atOnceUsers(1)))
         ).protocols(Protocols.httpProtocol);
     }