You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
53 lines
1.2 KiB
53 lines
1.2 KiB
package main
|
|
|
|
import (
|
|
"strconv"
|
|
"strings"
|
|
"testing"
|
|
|
|
. "github.com/tendermint/go-common"
|
|
"github.com/tendermint/tmsp/client"
|
|
"github.com/tendermint/tmsp/server"
|
|
"github.com/tendermint/tmsp/types"
|
|
)
|
|
|
|
func TestChainAware(t *testing.T) {
|
|
|
|
app := NewChainAwareApplication()
|
|
|
|
// Start the listener
|
|
_, err := server.NewServer("unix://test.sock", "socket", app)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
// Connect to the socket
|
|
client, err := tmspcli.NewSocketClient("unix://test.sock", false)
|
|
if err != nil {
|
|
Exit(Fmt("Error starting socket client: %v", err.Error()))
|
|
}
|
|
client.Start()
|
|
defer client.Stop()
|
|
|
|
n := uint64(5)
|
|
hash := []byte("fake block hash")
|
|
header := &types.Header{}
|
|
for i := uint64(0); i < n; i++ {
|
|
client.BeginBlockSync(hash, header)
|
|
client.EndBlockSync(i)
|
|
client.CommitSync()
|
|
}
|
|
|
|
r := app.Query(nil)
|
|
spl := strings.Split(string(r.Data), ",")
|
|
if len(spl) != 2 {
|
|
t.Fatal("expected %d,%d ; got %s", n, n, string(r.Data))
|
|
}
|
|
beginCount, _ := strconv.Atoi(spl[0])
|
|
endCount, _ := strconv.Atoi(spl[1])
|
|
if uint64(beginCount) != n {
|
|
t.Fatalf("expected beginCount of %d, got %d", n, beginCount)
|
|
} else if uint64(endCount) != n {
|
|
t.Fatalf("expected endCount of %d, got %d", n, endCount)
|
|
}
|
|
}
|