Refactor Application to Application & AppContext

example/dummy_test.go

@@ -1,8 +1,9 @@
 package example
 
 import (
-	// "fmt"
+	"reflect"
 	"testing"
+	"time"
 
 	. "github.com/tendermint/go-common"
 	"github.com/tendermint/go-wire"
@@ -12,6 +13,8 @@ import (
 
 func TestStream(t *testing.T) {
 
+	numAppendTxs := 200000
+
 	// Start the listener
 	_, err := server.StartListener("tcp://127.0.0.1:8080", NewDummyApplication())
 	if err != nil {
@@ -25,7 +28,9 @@ func TestStream(t *testing.T) {
 	}
 
 	// Read response data
+	done := make(chan struct{})
 	go func() {
+		counter := 0
 		for {
 			var n int
 			var err error
@@ -34,19 +39,59 @@ func TestStream(t *testing.T) {
 			if err != nil {
 				Exit(err.Error())
 			}
-			// fmt.Println("Read", n)
+
+			// Process response
+			switch res := res.(type) {
+			case types.ResponseAppendTx:
+				counter += 1
+				if res.RetCode != types.RetCodeOK {
+					t.Error("AppendTx failed with ret_code", res.RetCode)
+				}
+				if counter > numAppendTxs {
+					t.Fatal("Too many AppendTx responses")
+				}
+				t.Log("response", counter)
+				if counter == numAppendTxs {
+					go func() {
+						time.Sleep(time.Second * 2) // Wait for a bit to allow counter overflow
+						close(done)
+					}()
+				}
+			case types.ResponseFlush:
+				// ignore
+			default:
+				t.Error("Unexpected response type", reflect.TypeOf(res))
+			}
 		}
 	}()
 
 	// Write requests
-	for {
+	for counter := 0; counter < numAppendTxs; counter++ {
+		// Send request
 		var n int
 		var err error
 		var req types.Request = types.RequestAppendTx{TxBytes: []byte("test")}
 		wire.WriteBinary(req, conn, &n, &err)
 		if err != nil {
-			Exit(err.Error())
+			t.Fatal(err.Error())
+		}
+
+		// Sometimes send flush messages
+		if counter%123 == 0 {
+			t.Log("flush")
+			wire.WriteBinary(types.RequestFlush{}, conn, &n, &err)
+			if err != nil {
+				t.Fatal(err.Error())
+			}
 		}
-		// fmt.Println("Wrote", n)
 	}
+
+	// Send final flush message
+	var n int
+	wire.WriteBinary(types.RequestFlush{}, conn, &n, &err)
+	if err != nil {
+		t.Fatal(err.Error())
+	}
+
+	<-done
 }