mirror of
https://github.com/tendermint/tendermint.git
synced 2026-02-07 12:30:45 +00:00
e8b4226b17
Closes https://github.com/tendermint/tendermint/issues/3857 Moves `lib/` folder to `jsonrpc/`. Renames: **packages** `rpc` package -> `jsonrpc` package `rpcclient` package -> `client` package `rpcserver` package -> `server` package **structs and interfaces** ``` JSONRPCClient to Client JSONRPCRequestBatch to RequestBatch JSONRPCCaller to Caller ``` **functions** ``` StartHTTPServer to Serve StartHTTPAndTLSServer to ServeTLS rpc/jsonrpc/client: rename NewURIClient to NewURI NewJSONRPCClient to New NewJSONRPCClientWithHTTPClient to NewWithHTTPClient NewWSClient to NewWS ``` **misc** - unexpose `ResponseWriterWrapper` - remove unused http_params.go
60 lines
1.4 KiB
Go
60 lines
1.4 KiB
Go
package server
|
|
|
|
import (
|
|
"net/http"
|
|
"net/http/httptest"
|
|
"testing"
|
|
|
|
"github.com/gorilla/websocket"
|
|
"github.com/stretchr/testify/require"
|
|
|
|
amino "github.com/tendermint/go-amino"
|
|
|
|
"github.com/tendermint/tendermint/libs/log"
|
|
types "github.com/tendermint/tendermint/rpc/jsonrpc/types"
|
|
)
|
|
|
|
func TestWebsocketManagerHandler(t *testing.T) {
|
|
s := newWSServer()
|
|
defer s.Close()
|
|
|
|
// check upgrader works
|
|
d := websocket.Dialer{}
|
|
c, dialResp, err := d.Dial("ws://"+s.Listener.Addr().String()+"/websocket", nil)
|
|
require.NoError(t, err)
|
|
|
|
if got, want := dialResp.StatusCode, http.StatusSwitchingProtocols; got != want {
|
|
t.Errorf("dialResp.StatusCode = %q, want %q", got, want)
|
|
}
|
|
|
|
// check basic functionality works
|
|
req, err := types.MapToRequest(
|
|
amino.NewCodec(),
|
|
types.JSONRPCStringID("TestWebsocketManager"),
|
|
"c",
|
|
map[string]interface{}{"s": "a", "i": 10},
|
|
)
|
|
require.NoError(t, err)
|
|
err = c.WriteJSON(req)
|
|
require.NoError(t, err)
|
|
|
|
var resp types.RPCResponse
|
|
err = c.ReadJSON(&resp)
|
|
require.NoError(t, err)
|
|
require.Nil(t, resp.Error)
|
|
dialResp.Body.Close()
|
|
}
|
|
|
|
func newWSServer() *httptest.Server {
|
|
funcMap := map[string]*RPCFunc{
|
|
"c": NewWSRPCFunc(func(ctx *types.Context, s string, i int) (string, error) { return "foo", nil }, "s,i"),
|
|
}
|
|
wm := NewWebsocketManager(funcMap, amino.NewCodec())
|
|
wm.SetLogger(log.TestingLogger())
|
|
|
|
mux := http.NewServeMux()
|
|
mux.HandleFunc("/websocket", wm.WebsocketHandler)
|
|
|
|
return httptest.NewServer(mux)
|
|
}
|