From 8ac41f610c97a0c78b5cdba7f78a879e6200bd63 Mon Sep 17 00:00:00 2001
From: meixiongfeng <766881921@qq.com>
Date: Mon, 28 Nov 2022 17:35:28 +0800
Subject: [PATCH] =?UTF-8?q?websocket=E8=BF=94=E5=9B=9E=E9=AA=8C=E8=AF=81?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 stress/robot/robot.go                  | 8 ++++++++
 stress/server/golink/websocket_link.go | 1 +
 stress/stress.go                       | 2 +-
 3 files changed, 10 insertions(+), 1 deletion(-)

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)