diff --git a/stress/robot/robot.go b/stress/robot/robot.go index f5e618b52..70f7fef96 100644 --- a/stress/robot/robot.go +++ b/stress/robot/robot.go @@ -204,3 +204,11 @@ func (r *Robot) CloseHandler() { r.ws.Close() } + +func (r *Robot) GetMessagedata() []byte { + _, data, err := r.ws.ReadMessage() + if err != nil { + fmt.Printf("readMessage err:%v", err) + } + return data +} diff --git a/stress/server/golink/websocket_link.go b/stress/server/golink/websocket_link.go index 874e2ec88..d9e13e95e 100644 --- a/stress/server/golink/websocket_link.go +++ b/stress/server/golink/websocket_link.go @@ -83,6 +83,7 @@ func webSocketRequest(chanID uint64, ch chan<- *model.RequestResults, i uint64, for { if r.MessageRsp("user", "login") { isSucceed = true + errCode, isSucceed = request.GetVerifyWebSocket()(request, "webSocket.json", r.GetMessagedata()) break } } diff --git a/stress/stress.go b/stress/stress.go index 54e6017f4..515c1004a 100644 --- a/stress/stress.go +++ b/stress/stress.go @@ -81,7 +81,7 @@ func main() { flag.Usage() return } - debug := strings.ToLower(debugStr) == "true" + debug := strings.ToLower(debugStr) == "false" request, err := model.NewRequest(requestURL, verify, code, 0, debug, path, headers, body, maxCon, http2, keepalive) if err != nil { fmt.Printf("参数不合法 %v \n", err)