From 3e721456f5c01556bdc35495e0c537f653d81ca1 Mon Sep 17 00:00:00 2001
From: Ethan Buchman <ethan@coinculture.info>
Date: Tue, 1 Dec 2015 01:49:54 -0500
Subject: [PATCH] tmsp batch and some tests

---
 cmd/counter/main.go |  22 ++++++++++
 cmd/tmsp/cli.go     |  29 ++++++++++++-
 example/counter.go  |   2 +-
 test.sh             | 100 ++++++++++++++++++++++++++++++++++++++++++++
 4 files changed, 150 insertions(+), 3 deletions(-)
 create mode 100644 cmd/counter/main.go
 create mode 100644 test.sh

diff --git a/cmd/counter/main.go b/cmd/counter/main.go
new file mode 100644
index 000000000..36485996e
--- /dev/null
+++ b/cmd/counter/main.go
@@ -0,0 +1,22 @@
+package main
+
+import (
+	. "github.com/tendermint/go-common"
+	"github.com/tendermint/tmsp/example"
+	"github.com/tendermint/tmsp/server"
+)
+
+func main() {
+
+	// Start the listener
+	_, err := server.StartListener("tcp://127.0.0.1:8080", example.NewCounterApplication())
+	if err != nil {
+		Exit(err.Error())
+	}
+
+	// Wait forever
+	TrapSignal(func() {
+		// Cleanup
+	})
+
+}
diff --git a/cmd/tmsp/cli.go b/cmd/tmsp/cli.go
index aa5a5a8b8..242ffa60f 100644
--- a/cmd/tmsp/cli.go
+++ b/cmd/tmsp/cli.go
@@ -4,6 +4,7 @@ import (
 	"bufio"
 	"encoding/hex"
 	"fmt"
+	"io"
 	"net"
 	"os"
 	"strings"
@@ -30,6 +31,13 @@ func main() {
 		},
 	}
 	app.Commands = []cli.Command{
+		{
+			Name:  "batch",
+			Usage: "Run a batch of tmsp commands against an application",
+			Action: func(c *cli.Context) {
+				cmdBatch(app, c)
+			},
+		},
 		{
 			Name:  "console",
 			Usage: "Start an interactive tmsp console for multiple commands",
@@ -98,6 +106,23 @@ func before(c *cli.Context) error {
 
 //--------------------------------------------------------------------------------
 
+func cmdBatch(app *cli.App, c *cli.Context) {
+	bufReader := bufio.NewReader(os.Stdin)
+	for {
+		line, more, err := bufReader.ReadLine()
+		if more {
+			Exit("input line is too long")
+		} else if err == io.EOF {
+			break
+		} else if err != nil {
+			Exit(err.Error())
+		}
+		args := []string{"tmsp"}
+		args = append(args, strings.Split(string(line), " ")...)
+		app.Run(args)
+	}
+}
+
 func cmdConsole(app *cli.App, c *cli.Context) {
 	for {
 		fmt.Printf("> ")
@@ -134,7 +159,7 @@ func cmdSetOption(c *cli.Context) {
 	if err != nil {
 		Exit(err.Error())
 	}
-	fmt.Printf("Set option %s = %s\n", args[0], args[1])
+	fmt.Printf("%s=%s\n", args[0], args[1])
 }
 
 // Append a new tx to application
@@ -166,7 +191,7 @@ func cmdGetHash(c *cli.Context) {
 	if err != nil {
 		Exit(err.Error())
 	}
-	fmt.Println("Got hash:", Fmt("%X", res.(types.ResponseGetHash).Hash))
+	fmt.Printf("%X\n", res.(types.ResponseGetHash).Hash)
 }
 
 // Commit the application state
diff --git a/example/counter.go b/example/counter.go
index 870d5a808..d4b0e4776 100644
--- a/example/counter.go
+++ b/example/counter.go
@@ -69,7 +69,7 @@ func (appC *CounterAppContext) AppendTx(tx []byte) ([]types.Event, types.RetCode
 
 func (appC *CounterAppContext) GetHash() ([]byte, types.RetCode) {
 	hash := make([]byte, 32)
-	binary.PutVarint(hash, int64(appC.hashCount))
+	binary.PutVarint(hash, int64(appC.txCount))
 	appC.hashCount += 1
 	return hash, 0
 }
diff --git a/test.sh b/test.sh
new file mode 100644
index 000000000..00c1f2037
--- /dev/null
+++ b/test.sh
@@ -0,0 +1,100 @@
+#! /bin/bash
+
+# Make sure the tmsp cli can connect to the dummy
+echo "Dummy test ..."
+dummy &> /dev/null &
+PID=`echo $!`
+sleep 1
+RESULT_HASH=`tmsp get_hash`
+if [[ "$RESULT_HASH" != "" ]]; then
+	echo "Expected nothing but got: $RESULT_HASH"
+	exit 1
+fi
+echo "... Pass!"
+echo ""
+
+# Add a tx, get hash, commit, get hash
+# hashes should be non-empty and identical
+echo "Dummy batch test ..."
+OUTPUT=`(tmsp batch) <<STDIN 
+append_tx abc
+get_hash
+commit
+get_hash
+STDIN`
+
+HASH1=`echo "$OUTPUT" | tail -n 3 | head -n 1`
+HASH2=`echo "$OUTPUT" | tail -n 1`
+
+if [[ "$HASH1" == "" ]]; then
+	echo "Expected non empty hash!"
+	exit 1
+fi
+
+if [[ "$HASH1" != "$HASH2" ]]; then
+	echo "Expected hashes before and after commit to match: $HASH1, $HASH2"
+	exit 1
+fi
+echo "... Pass!"
+echo ""
+
+# Start a new connection and ensure the hash is the same
+echo "New connection test ..."
+RESULT_HASH=`tmsp get_hash`
+if [[ "$HASH1" != "$RESULT_HASH" ]]; then
+	echo "Expected hash to persist as $HASH1 for new connection. Got $RESULT_HASH"
+	exit 1
+fi
+echo "... Pass!"
+echo ""
+
+
+kill $PID
+sleep 1
+
+# test the counter app
+echo "Counter test ..."
+counter &> /dev/null &
+PID=`echo $!`
+sleep 1
+OUTPUT=`(tmsp batch) <<STDIN 
+set_option serial on
+get_hash
+append_tx abc
+STDIN`
+
+# why can't we pick up the non-zero exit code here?
+# echo $?
+
+HASH1=`echo "$OUTPUT" | tail -n +2 | head -n 1`
+if [[ "${HASH1:0:2}" != "00" ]]; then
+	echo "Expected opening hash to lead with 00. Got $HASH1"	
+	exit 1
+fi
+
+OUTPUT=`(tmsp batch) <<STDIN 
+set_option serial on
+append_tx 0x00
+get_hash
+append_tx 0x02
+get_hash
+STDIN`
+
+HASH1=`echo "$OUTPUT" | tail -n +3 | head -n 1`
+HASH2=`echo "$OUTPUT" | tail -n +5 | head -n 1`
+
+if [[ "${HASH1:0:2}" != "02" ]]; then
+	echo "Expected hash to lead with 02. Got $HASH1"	
+	exit 1
+fi
+
+if [[ "${HASH2:0:2}" != "04" ]]; then
+	echo "Expected hash to lead with 04. Got $HASH2"	
+	exit 1
+fi
+
+echo "... Pass!"
+echo ""
+
+kill $PID
+