From 38b9f5fe7fb47f8b13298e0aaa617bf20b77ab23 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=E5=BE=90=E8=81=AA?= Date: Tue, 26 Jul 2022 10:51:12 +0800 Subject: [PATCH] fix: unittest --- hrp/step_websocket.go | 13 ++++++------- 1 file changed, 6 insertions(+), 7 deletions(-) diff --git a/hrp/step_websocket.go b/hrp/step_websocket.go index 0e3131b9..90feec2d 100644 --- a/hrp/step_websocket.go +++ b/hrp/step_websocket.go @@ -232,18 +232,17 @@ type WebSocketAction struct { } func initWebSocket(testcase *TestCase) { - tCase := testcase.ToTCase() - for _, step := range tCase.TestSteps { - if step.WebSocket == nil { + for _, step := range testcase.TestSteps { + if step.Struct().WebSocket == nil { continue } // init websocket action parameters - if step.WebSocket.Timeout <= 0 { - step.WebSocket.Timeout = defaultTimeout + if step.Struct().WebSocket.Timeout <= 0 { + step.Struct().WebSocket.Timeout = defaultTimeout } // close status code range: [1000, 4999]. ref: https://datatracker.ietf.org/doc/html/rfc6455#section-11.7 - if step.WebSocket.CloseStatusCode < 1000 || step.WebSocket.CloseStatusCode > 4999 { - step.WebSocket.CloseStatusCode = defaultCloseStatus + if step.Struct().WebSocket.CloseStatusCode < 1000 || step.Struct().WebSocket.CloseStatusCode > 4999 { + step.Struct().WebSocket.CloseStatusCode = defaultCloseStatus } } }