forked from cerc-io/plugeth
rpc, internal/guide: speed up tests a bit (#26193)
This removes an RPC test which takes > 90s to execute, and updates the internal/guide tests to use lighter scrypt parameters. Co-authored-by: Felix Lange <fjl@twurst.com>
This commit is contained in:
parent
722bb210bf
commit
97c563e055
@ -38,8 +38,8 @@ func TestAccountManagement(t *testing.T) {
|
|||||||
// Create a temporary folder to work with
|
// Create a temporary folder to work with
|
||||||
workdir := t.TempDir()
|
workdir := t.TempDir()
|
||||||
|
|
||||||
// Create an encrypted keystore with standard crypto parameters
|
// Create an encrypted keystore (using light scrypt parameters)
|
||||||
ks := keystore.NewKeyStore(filepath.Join(workdir, "keystore"), keystore.StandardScryptN, keystore.StandardScryptP)
|
ks := keystore.NewKeyStore(filepath.Join(workdir, "keystore"), keystore.LightScryptN, keystore.LightScryptP)
|
||||||
|
|
||||||
// Create a new account with the specified encryption passphrase
|
// Create a new account with the specified encryption passphrase
|
||||||
newAcc, err := ks.NewAccount("Creation password")
|
newAcc, err := ks.NewAccount("Creation password")
|
||||||
|
@ -19,14 +19,10 @@ package rpc
|
|||||||
import (
|
import (
|
||||||
"context"
|
"context"
|
||||||
"errors"
|
"errors"
|
||||||
"io"
|
|
||||||
"net"
|
"net"
|
||||||
"net/http"
|
"net/http"
|
||||||
"net/http/httptest"
|
"net/http/httptest"
|
||||||
"net/http/httputil"
|
|
||||||
"net/url"
|
|
||||||
"strings"
|
"strings"
|
||||||
"sync/atomic"
|
|
||||||
"testing"
|
"testing"
|
||||||
"time"
|
"time"
|
||||||
|
|
||||||
@ -227,63 +223,6 @@ func TestClientWebsocketLargeMessage(t *testing.T) {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
func TestClientWebsocketSevered(t *testing.T) {
|
|
||||||
t.Parallel()
|
|
||||||
|
|
||||||
var (
|
|
||||||
server = wsPingTestServer(t, nil)
|
|
||||||
ctx = context.Background()
|
|
||||||
)
|
|
||||||
defer server.Shutdown(ctx)
|
|
||||||
|
|
||||||
u, err := url.Parse("http://" + server.Addr)
|
|
||||||
if err != nil {
|
|
||||||
t.Fatal(err)
|
|
||||||
}
|
|
||||||
rproxy := httputil.NewSingleHostReverseProxy(u)
|
|
||||||
var severable *severableReadWriteCloser
|
|
||||||
rproxy.ModifyResponse = func(response *http.Response) error {
|
|
||||||
severable = &severableReadWriteCloser{ReadWriteCloser: response.Body.(io.ReadWriteCloser)}
|
|
||||||
response.Body = severable
|
|
||||||
return nil
|
|
||||||
}
|
|
||||||
frontendProxy := httptest.NewServer(rproxy)
|
|
||||||
defer frontendProxy.Close()
|
|
||||||
|
|
||||||
wsURL := "ws:" + strings.TrimPrefix(frontendProxy.URL, "http:")
|
|
||||||
client, err := DialWebsocket(ctx, wsURL, "")
|
|
||||||
if err != nil {
|
|
||||||
t.Fatalf("client dial error: %v", err)
|
|
||||||
}
|
|
||||||
defer client.Close()
|
|
||||||
|
|
||||||
resultChan := make(chan int)
|
|
||||||
sub, err := client.EthSubscribe(ctx, resultChan, "foo")
|
|
||||||
if err != nil {
|
|
||||||
t.Fatalf("client subscribe error: %v", err)
|
|
||||||
}
|
|
||||||
|
|
||||||
// sever the connection
|
|
||||||
severable.Sever()
|
|
||||||
|
|
||||||
// Wait for subscription error.
|
|
||||||
timeout := time.NewTimer(3 * wsPingInterval)
|
|
||||||
defer timeout.Stop()
|
|
||||||
for {
|
|
||||||
select {
|
|
||||||
case err := <-sub.Err():
|
|
||||||
t.Log("client subscription error:", err)
|
|
||||||
return
|
|
||||||
case result := <-resultChan:
|
|
||||||
t.Error("unexpected result:", result)
|
|
||||||
return
|
|
||||||
case <-timeout.C:
|
|
||||||
t.Error("didn't get any error within the test timeout")
|
|
||||||
return
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
// wsPingTestServer runs a WebSocket server which accepts a single subscription request.
|
// wsPingTestServer runs a WebSocket server which accepts a single subscription request.
|
||||||
// When a value arrives on sendPing, the server sends a ping frame, waits for a matching
|
// When a value arrives on sendPing, the server sends a ping frame, waits for a matching
|
||||||
// pong and finally delivers a single subscription result.
|
// pong and finally delivers a single subscription result.
|
||||||
@ -386,31 +325,3 @@ func wsPingTestHandler(t *testing.T, conn *websocket.Conn, shutdown, sendPing <-
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
// severableReadWriteCloser wraps an io.ReadWriteCloser and provides a Sever() method to drop writes and read empty.
|
|
||||||
type severableReadWriteCloser struct {
|
|
||||||
io.ReadWriteCloser
|
|
||||||
severed int32 // atomic
|
|
||||||
}
|
|
||||||
|
|
||||||
func (s *severableReadWriteCloser) Sever() {
|
|
||||||
atomic.StoreInt32(&s.severed, 1)
|
|
||||||
}
|
|
||||||
|
|
||||||
func (s *severableReadWriteCloser) Read(p []byte) (n int, err error) {
|
|
||||||
if atomic.LoadInt32(&s.severed) > 0 {
|
|
||||||
return 0, nil
|
|
||||||
}
|
|
||||||
return s.ReadWriteCloser.Read(p)
|
|
||||||
}
|
|
||||||
|
|
||||||
func (s *severableReadWriteCloser) Write(p []byte) (n int, err error) {
|
|
||||||
if atomic.LoadInt32(&s.severed) > 0 {
|
|
||||||
return len(p), nil
|
|
||||||
}
|
|
||||||
return s.ReadWriteCloser.Write(p)
|
|
||||||
}
|
|
||||||
|
|
||||||
func (s *severableReadWriteCloser) Close() error {
|
|
||||||
return s.ReadWriteCloser.Close()
|
|
||||||
}
|
|
||||||
|
Loading…
Reference in New Issue
Block a user