TrueCloudLab/neoneo-go
7
0
Fork 2
Code Issues Pull requests Projects Releases Packages Wiki Activity

rpc/client: add minimalistic websocket client

Browse source
This commit is contained in:
Roman Khimov 2020-04-29 22:51:43 +03:00
parent a458a17748
commit 3de48d7d90
5 changed files with 230 additions and 11 deletions
Download patch file Download diff file Expand all files Collapse all files

53
pkg/rpc/client/rpc_test.go
View file

@ -3,11 +3,13 @@ package client
import (
"context"
"encoding/hex"
"encoding/json"
"net/http"
"net/http/httptest"
"strings"
"testing"
"time"
"github.com/gorilla/websocket"
"github.com/nspcc-dev/neo-go/pkg/core"
"github.com/nspcc-dev/neo-go/pkg/core/block"
"github.com/nspcc-dev/neo-go/pkg/core/transaction"
@ -1433,7 +1435,22 @@ var rpcClientErrorCases = map[string][]rpcClientErrorCase{
},
}
func TestRPCClient(t *testing.T) {
func TestRPCClients(t *testing.T) {
t.Run("Client", func(t *testing.T) {
testRPCClient(t, func(ctx context.Context, endpoint string, opts Options) (*Client, error) {
return New(ctx, endpoint, opts)
})
})
t.Run("WSClient", func(t *testing.T) {
testRPCClient(t, func(ctx context.Context, endpoint string, opts Options) (*Client, error) {
wsc, err := NewWS(ctx, httpURLtoWS(endpoint), opts)
require.NoError(t, err)
return &wsc.Client, nil
})
})
}
func testRPCClient(t *testing.T, newClient func(context.Context, string, Options) (*Client, error)) {
for method, testBatch := range rpcClientTestCases {
t.Run(method, func(t *testing.T) {
for _, testCase := range testBatch {
@ -1443,7 +1460,7 @@ func TestRPCClient(t *testing.T) {
endpoint := srv.URL
opts := Options{}
c, err := New(context.TODO(), endpoint, opts)
c, err := newClient(context.TODO(), endpoint, opts)
if err != nil {
t.Fatal(err)
}
@ -1467,7 +1484,7 @@ func TestRPCClient(t *testing.T) {
endpoint := srv.URL
opts := Options{}
c, err := New(context.TODO(), endpoint, opts)
c, err := newClient(context.TODO(), endpoint, opts)
if err != nil {
t.Fatal(err)
}
@ -1481,8 +1498,31 @@ func TestRPCClient(t *testing.T) {
}
}
func httpURLtoWS(url string) string {
return "ws" + strings.TrimPrefix(url, "http") + "/ws"
}
func initTestServer(t *testing.T, resp string) *httptest.Server {
srv := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, req *http.Request) {
if req.URL.Path == "/ws" && req.Method == "GET" {
var upgrader = websocket.Upgrader{}
ws, err := upgrader.Upgrade(w, req, nil)
require.NoError(t, err)
for {
ws.SetReadDeadline(time.Now().Add(2 * time.Second))
_, _, err = ws.ReadMessage()
if err != nil {
break
}
ws.SetWriteDeadline(time.Now().Add(2 * time.Second))
err = ws.WriteMessage(1, []byte(resp))
if err != nil {
break
}
}
ws.Close()
return
}
requestHandler(t, w, resp)
}))
@ -1491,11 +1531,10 @@ func initTestServer(t *testing.T, resp string) *httptest.Server {
func requestHandler(t *testing.T, w http.ResponseWriter, resp string) {
w.Header().Set("Content-Type", "application/json; charset=utf-8")
encoder := json.NewEncoder(w)
err := encoder.Encode(json.RawMessage(resp))
_, err := w.Write([]byte(resp))
if err != nil {
t.Fatalf("Error encountered while encoding response: %s", err.Error())
t.Fatalf("Error writing response: %s", err.Error())
}
}