TrueCloudLab/neo-go
8
0
Fork 0
mirror of https://github.com/nspcc-dev/neo-go.git synced 2025-05-03 23:02:27 +00:00
Code Issues Projects Releases Packages Wiki Activity

rpc: adjust cases when WS connection close error is returned

Browse source 
Do not return error in case of (*WSClient).Close() method was the
initiator of connection closing.
This commit is contained in:
Anna Shaleva 2022-06-10 17:20:45 +03:00
parent 4afdb9fd89
commit 8f53c7d78a
2 changed files with 27 additions and 15 deletions
Download patch file Download diff file Expand all files Collapse all files

16
pkg/rpc/client/wsclient_test.go
View file

@ -472,20 +472,24 @@ func TestWS_RequestAfterClose(t *testing.T) {
}
func TestWSClient_ConnClosedError(t *testing.T) {
srv := initTestServer(t, "")
t.Run("standard closing", func(t *testing.T) {
srv := initTestServer(t, `{"jsonrpc": "2.0", "id": 1, "result": 123}`)
c, err := NewWS(context.TODO(), httpURLtoWS(srv.URL), Options{})
require.NoError(t, err)
c.Close()
// Check client is working.
_, err = c.GetBlockCount()
require.NoError(t, err)
err = c.GetError()
require.Error(t, err)
require.True(t, strings.Contains(err.Error(), "use of closed network connection"))
require.NoError(t, err)
c.Close()
err = c.GetError()
require.NoError(t, err)
})
t.Run("malformed request", func(t *testing.T) {
srv := initTestServer(t, "")
c, err := NewWS(context.TODO(), httpURLtoWS(srv.URL), Options{})
require.NoError(t, err)