From cd3da2bf293936128a65489d55ac8525de5236e4 Mon Sep 17 00:00:00 2001
From: Jae Kwon <jae@tendermint.com>
Date: Mon, 28 Mar 2016 09:35:19 -0700
Subject: [PATCH] Add TestSequence in tests/tmsp

---
 tests/tendermint/main.go |  13 +++--
 tests/tmsp/main.go       | 102 ++++++++++++++++++++++++++++++++++++++-
 2 files changed, 108 insertions(+), 7 deletions(-)

diff --git a/tests/tendermint/main.go b/tests/tendermint/main.go
index 6f669cdbc..9c037e192 100644
--- a/tests/tendermint/main.go
+++ b/tests/tendermint/main.go
@@ -15,7 +15,10 @@ import (
 )
 
 func main() {
-	ws := rpcclient.NewWSClient("127.0.0.1:46657", "/websocket")
+	// ws := rpcclient.NewWSClient("127.0.0.1:46657", "/websocket")
+	ws := rpcclient.NewWSClient("192.168.99.100:46657", "/websocket")
+	chainID := "test_chain_id"
+
 	_, err := ws.Start()
 	if err != nil {
 		Exit(err.Error())
@@ -34,7 +37,7 @@ func main() {
 	}()
 
 	// Get the root account
-	root := tests.PrivAccountFromSecret("root")
+	root := tests.PrivAccountFromSecret("test")
 	sequence := int(0)
 	// Make a bunch of PrivAccounts
 	privAccounts := tests.RandAccounts(1000, 1000000, 0)
@@ -62,7 +65,7 @@ func main() {
 		sequence += 1
 
 		// Sign request
-		signBytes := wire.BinaryBytes(tx)
+		signBytes := tx.SignBytes(chainID)
 		sig := root.PrivKey.Sign(signBytes)
 		tx.Inputs[0].Signature = sig
 		//fmt.Println("tx:", tx)
@@ -100,7 +103,7 @@ func main() {
 					Address:  privAccountA.Account.PubKey.Address(),
 					PubKey:   privAccountA.Account.PubKey,
 					Amount:   3,
-					Sequence: privAccountASequence,
+					Sequence: privAccountASequence + 1,
 				},
 			},
 			Outputs: []types.TxOutput{
@@ -112,7 +115,7 @@ func main() {
 		}
 
 		// Sign request
-		signBytes := wire.BinaryBytes(tx)
+		signBytes := tx.SignBytes(chainID)
 		sig := privAccountA.PrivKey.Sign(signBytes)
 		tx.Inputs[0].Signature = sig
 		//fmt.Println("tx:", tx)
diff --git a/tests/tmsp/main.go b/tests/tmsp/main.go
index 7f0eb583e..bd2971ea4 100644
--- a/tests/tmsp/main.go
+++ b/tests/tmsp/main.go
@@ -13,8 +13,9 @@ import (
 )
 
 func main() {
-	testSendTx()
-	testGov()
+	//testSendTx()
+	//testGov()
+	testSequence()
 }
 
 func testSendTx() {
@@ -85,3 +86,100 @@ func testGov() {
 	}
 	// TODO test proposals or something.
 }
+
+func testSequence() {
+	eyesCli := eyescli.NewLocalClient()
+	bcApp := app.NewBasecoin(eyesCli)
+	chainID := "test_chain_id"
+
+	// Get the root account
+	root := tests.PrivAccountFromSecret("test")
+	rootAcc := root.Account
+	rootAcc.Balance = 1 << 53
+	fmt.Println(bcApp.SetOption("base/chainID", "test_chain_id"))
+	fmt.Println(bcApp.SetOption("base/account", string(wire.JSONBytes(rootAcc))))
+
+	sequence := int(1)
+	// Make a bunch of PrivAccounts
+	privAccounts := tests.RandAccounts(1000, 1000000, 0)
+	privAccountSequences := make(map[string]int)
+
+	// Send coins to each account
+	for i := 0; i < len(privAccounts); i++ {
+		privAccount := privAccounts[i]
+		tx := &types.SendTx{
+			Inputs: []types.TxInput{
+				types.TxInput{
+					Address:  root.Account.PubKey.Address(),
+					PubKey:   root.Account.PubKey, // TODO is this needed?
+					Amount:   1000002,
+					Sequence: sequence,
+				},
+			},
+			Outputs: []types.TxOutput{
+				types.TxOutput{
+					Address: privAccount.Account.PubKey.Address(),
+					Amount:  1000000,
+				},
+			},
+		}
+		sequence += 1
+
+		// Sign request
+		signBytes := tx.SignBytes(chainID)
+		sig := root.PrivKey.Sign(signBytes)
+		tx.Inputs[0].Signature = sig
+		//fmt.Println("tx:", tx)
+
+		// Write request
+		txBytes := wire.BinaryBytes(tx)
+		res := bcApp.CheckTx(txBytes)
+		if res.IsErr() {
+			Exit("AppendTx error: " + res.Error())
+		}
+	}
+
+	// Now send coins between these accounts
+	for {
+		randA := RandInt() % len(privAccounts)
+		randB := RandInt() % len(privAccounts)
+		if randA == randB {
+			continue
+		}
+
+		privAccountA := privAccounts[randA]
+		privAccountASequence := privAccountSequences[privAccountA.Account.PubKey.KeyString()]
+		privAccountSequences[privAccountA.Account.PubKey.KeyString()] = privAccountASequence + 1
+		privAccountB := privAccounts[randB]
+
+		tx := &types.SendTx{
+			Inputs: []types.TxInput{
+				types.TxInput{
+					Address:  privAccountA.Account.PubKey.Address(),
+					PubKey:   privAccountA.Account.PubKey,
+					Amount:   3,
+					Sequence: privAccountASequence + 1,
+				},
+			},
+			Outputs: []types.TxOutput{
+				types.TxOutput{
+					Address: privAccountB.Account.PubKey.Address(),
+					Amount:  1,
+				},
+			},
+		}
+
+		// Sign request
+		signBytes := tx.SignBytes(chainID)
+		sig := privAccountA.PrivKey.Sign(signBytes)
+		tx.Inputs[0].Signature = sig
+		//fmt.Println("tx:", tx)
+
+		// Write request
+		txBytes := wire.BinaryBytes(tx)
+		res := bcApp.AppendTx(txBytes)
+		if res.IsErr() {
+			Exit("AppendTx error: " + res.Error())
+		}
+	}
+}