|
|
@ -1,152 +0,0 @@ |
|
|
|
package rpctest |
|
|
|
|
|
|
|
import ( |
|
|
|
"testing" |
|
|
|
"time" |
|
|
|
|
|
|
|
"github.com/stretchr/testify/require" |
|
|
|
. "github.com/tendermint/go-common" |
|
|
|
"github.com/tendermint/go-wire" |
|
|
|
|
|
|
|
client "github.com/tendermint/go-rpc/client" |
|
|
|
ctypes "github.com/tendermint/tendermint/rpc/core/types" |
|
|
|
) |
|
|
|
|
|
|
|
/* |
|
|
|
// global variables for use across all tests
|
|
|
|
var ( |
|
|
|
config cfg.Config |
|
|
|
node *nm.Node |
|
|
|
chainID string |
|
|
|
rpcAddr string |
|
|
|
requestAddr string |
|
|
|
websocketAddr string |
|
|
|
websocketEndpoint string |
|
|
|
grpcAddr string |
|
|
|
clientURI *client.ClientURI |
|
|
|
clientJSON *client.ClientJSONRPC |
|
|
|
clientGRPC core_grpc.BroadcastAPIClient |
|
|
|
) |
|
|
|
|
|
|
|
// initialize config and create new node
|
|
|
|
func init() { |
|
|
|
config = tendermint_test.ResetConfig("rpc_test_client_test") |
|
|
|
chainID = config.GetString("chain_id") |
|
|
|
rpcAddr = config.GetString("rpc_laddr") |
|
|
|
grpcAddr = config.GetString("grpc_laddr") |
|
|
|
requestAddr = rpcAddr |
|
|
|
websocketAddr = rpcAddr |
|
|
|
websocketEndpoint = "/websocket" |
|
|
|
|
|
|
|
clientURI = client.NewClientURI(requestAddr) |
|
|
|
clientJSON = client.NewClientJSONRPC(requestAddr) |
|
|
|
clientGRPC = core_grpc.StartGRPCClient(grpcAddr) |
|
|
|
|
|
|
|
// TODO: change consensus/state.go timeouts to be shorter
|
|
|
|
|
|
|
|
// start a node
|
|
|
|
ready := make(chan struct{}) |
|
|
|
go newNode(ready) |
|
|
|
<-ready |
|
|
|
} |
|
|
|
|
|
|
|
// create a new node and sleep forever
|
|
|
|
func newNode(ready chan struct{}) { |
|
|
|
// Create & start node
|
|
|
|
node = nm.NewNodeDefault(config) |
|
|
|
node.Start() |
|
|
|
|
|
|
|
time.Sleep(time.Second) |
|
|
|
|
|
|
|
ready <- struct{}{} |
|
|
|
|
|
|
|
// Sleep forever
|
|
|
|
ch := make(chan struct{}) |
|
|
|
<-ch |
|
|
|
} |
|
|
|
|
|
|
|
//--------------------------------------------------------------------------------
|
|
|
|
// Utilities for testing the websocket service
|
|
|
|
|
|
|
|
// create a new connection
|
|
|
|
func newWSClient(t *testing.T) *client.WSClient { |
|
|
|
wsc := client.NewWSClient(websocketAddr, websocketEndpoint) |
|
|
|
if _, err := wsc.Start(); err != nil { |
|
|
|
panic(err) |
|
|
|
} |
|
|
|
return wsc |
|
|
|
} |
|
|
|
*/ |
|
|
|
|
|
|
|
// subscribe to an event
|
|
|
|
func subscribe(t *testing.T, wsc *client.WSClient, eventid string) { |
|
|
|
require.Nil(t, wsc.Subscribe(eventid)) |
|
|
|
} |
|
|
|
|
|
|
|
// unsubscribe from an event
|
|
|
|
func unsubscribe(t *testing.T, wsc *client.WSClient, eventid string) { |
|
|
|
require.Nil(t, wsc.Unsubscribe(eventid)) |
|
|
|
} |
|
|
|
|
|
|
|
// wait for an event; do things that might trigger events, and check them when they are received
|
|
|
|
// the check function takes an event id and the byte slice read off the ws
|
|
|
|
func waitForEvent(t *testing.T, wsc *client.WSClient, eventid string, dieOnTimeout bool, f func(), check func(string, interface{}) error) { |
|
|
|
// go routine to wait for webscoket msg
|
|
|
|
goodCh := make(chan interface{}) |
|
|
|
errCh := make(chan error) |
|
|
|
|
|
|
|
// Read message
|
|
|
|
go func() { |
|
|
|
var err error |
|
|
|
LOOP: |
|
|
|
for { |
|
|
|
select { |
|
|
|
case r := <-wsc.ResultsCh: |
|
|
|
result := new(ctypes.TMResult) |
|
|
|
wire.ReadJSONPtr(result, r, &err) |
|
|
|
if err != nil { |
|
|
|
errCh <- err |
|
|
|
break LOOP |
|
|
|
} |
|
|
|
event, ok := (*result).(*ctypes.ResultEvent) |
|
|
|
if ok && event.Name == eventid { |
|
|
|
goodCh <- event.Data |
|
|
|
break LOOP |
|
|
|
} |
|
|
|
case err := <-wsc.ErrorsCh: |
|
|
|
errCh <- err |
|
|
|
break LOOP |
|
|
|
case <-wsc.Quit: |
|
|
|
break LOOP |
|
|
|
} |
|
|
|
} |
|
|
|
}() |
|
|
|
|
|
|
|
// do stuff (transactions)
|
|
|
|
f() |
|
|
|
|
|
|
|
// wait for an event or timeout
|
|
|
|
timeout := time.NewTimer(10 * time.Second) |
|
|
|
select { |
|
|
|
case <-timeout.C: |
|
|
|
if dieOnTimeout { |
|
|
|
wsc.Stop() |
|
|
|
panic(Fmt("%s event was not received in time", eventid)) |
|
|
|
} |
|
|
|
// else that's great, we didn't hear the event
|
|
|
|
// and we shouldn't have
|
|
|
|
case eventData := <-goodCh: |
|
|
|
if dieOnTimeout { |
|
|
|
// message was received and expected
|
|
|
|
// run the check
|
|
|
|
require.Nil(t, check(eventid, eventData)) |
|
|
|
} else { |
|
|
|
wsc.Stop() |
|
|
|
panic(Fmt("%s event was not expected", eventid)) |
|
|
|
} |
|
|
|
case err := <-errCh: |
|
|
|
panic(err) // Show the stack trace.
|
|
|
|
} |
|
|
|
} |
|
|
|
|
|
|
|
//--------------------------------------------------------------------------------
|