RanchiMall/blockbook
3
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Add option to create new connection for each request to socket.io test

Browse Source
This commit is contained in:
Martin Boehm 2018-05-10 10:02:25 +02:00
parent 882bcf8512
commit 1cb2338587
1 changed files with 21 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

29
server/socketio_test.go
View File

@ -19,8 +19,7 @@ import (
var ( var (
// verifier functionality // verifier functionality
verifylog = flag.String("verifylog", "", "path to logfile containing socket.io requests/responses") newSocket = flag.Bool("newsocket", false, "Create new socket.io connection for each request")
wsurl = flag.String("wsurl", "", "URL of socket.io interface to verify")
) )
type verifyStats struct { type verifyStats struct {
@ -374,11 +373,7 @@ func verifyMessage(t *testing.T, ws *gosocketio.Client, id int, lrs *logRequestR
} }
} }
func Test_VerifyLog(t *testing.T) { func connectSocketIO(t *testing.T) *gosocketio.Client {
if *verifylog == "" || *wsurl == "" {
t.Skip("skipping test, flags verifylog or wsurl not specified")
}
t.Log("Verifying log", *verifylog, "against service", *wsurl)
tr := transport.GetDefaultWebsocketTransport() tr := transport.GetDefaultWebsocketTransport()
tr.WebsocketDialer = websocket.Dialer{ tr.WebsocketDialer = websocket.Dialer{
TLSClientConfig: &tls.Config{InsecureSkipVerify: true}, TLSClientConfig: &tls.Config{InsecureSkipVerify: true},
@ -386,9 +381,21 @@ func Test_VerifyLog(t *testing.T) {
ws, err := gosocketio.Dial(*wsurl, tr) ws, err := gosocketio.Dial(*wsurl, tr)
if err != nil { if err != nil {
t.Fatal("Dial error ", err) t.Fatal("Dial error ", err)
return return nil
} }
return ws
}
func Test_VerifyLog(t *testing.T) {
if *verifylog == "" || *wsurl == "" {
t.Skip("skipping test, flags verifylog or wsurl not specified")
}
t.Log("Verifying log", *verifylog, "against service", *wsurl)
var ws *gosocketio.Client
if !*newSocket {
ws = connectSocketIO(t)
defer ws.Close() defer ws.Close()
}
file, err := os.Open(*verifylog) file, err := os.Open(*verifylog)
if err != nil { if err != nil {
t.Fatal("File read error", err) t.Fatal("File read error", err)
@ -430,7 +437,13 @@ func Test_VerifyLog(t *testing.T) {
lrs.LogElapsedTime = msg.Et lrs.LogElapsedTime = msg.Et
} }
if lrs.Request != nil && lrs.Response != nil { if lrs.Request != nil && lrs.Response != nil {
if *newSocket {
ws = connectSocketIO(t)
}
verifyMessage(t, ws, msg.ID, lrs, stats) verifyMessage(t, ws, msg.ID, lrs, stats)
if *newSocket {
ws.Close()
}
delete(pairs, msg.ID) delete(pairs, msg.ID)
} }
} }