GithubBackup/httprunner
1
0
Fork 0
mirror of https://github.com/httprunner/httprunner.git synced 2026-09-05 07:26:55 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix: add ws url log info

Browse Source
This commit is contained in:
buyuxiang
2022-07-19 11:21:36 +08:00
parent aa5de4ec83
commit 7b47a920b4
1 changed files with 6 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files
hrp/step_websocket.go
+6 -6
View File
@@ -323,7 +323,7 @@ func runStepWebSocket(r *SessionRunner, step *TStep) (stepResult *StepResult, er
} }
switch step.WebSocket.Type { switch step.WebSocket.Type {
case wsOpen: case wsOpen:
log.Info().Int64("timeout(ms)", step.WebSocket.Timeout).Msg("open websocket connection") log.Info().Int64("timeout(ms)", step.WebSocket.Timeout).Str("url", parsedURL).Msg("open websocket connection")
// use the current websocket connection if existed // use the current websocket connection if existed
if r.wsConnMap[parsedURL] != nil { if r.wsConnMap[parsedURL] != nil {
break break
@@ -333,7 +333,7 @@ func runStepWebSocket(r *SessionRunner, step *TStep) (stepResult *StepResult, er
return stepResult, errors.Wrap(err, "open connection failed") return stepResult, errors.Wrap(err, "open connection failed")
} }
case wsPing: case wsPing:
log.Info().Int64("timeout(ms)", step.WebSocket.Timeout).Msg("send ping and expect pong") log.Info().Int64("timeout(ms)", step.WebSocket.Timeout).Str("url", parsedURL).Msg("send ping and expect pong")
err = writeWebSocket(parsedURL, r, step, stepVariables) err = writeWebSocket(parsedURL, r, step, stepVariables)
if err != nil { if err != nil {
return stepResult, errors.Wrap(err, "send ping message failed") return stepResult, errors.Wrap(err, "send ping message failed")
@@ -351,7 +351,7 @@ func runStepWebSocket(r *SessionRunner, step *TStep) (stepResult *StepResult, er
} }
}() }()
case wsWriteAndRead: case wsWriteAndRead:
log.Info().Int64("timeout(ms)", step.WebSocket.Timeout).Msg("write a message and read response") log.Info().Int64("timeout(ms)", step.WebSocket.Timeout).Str("url", parsedURL).Msg("write a message and read response")
err = writeWebSocket(parsedURL, r, step, stepVariables) err = writeWebSocket(parsedURL, r, step, stepVariables)
if err != nil { if err != nil {
return stepResult, errors.Wrap(err, "write message failed") return stepResult, errors.Wrap(err, "write message failed")
@@ -361,19 +361,19 @@ func runStepWebSocket(r *SessionRunner, step *TStep) (stepResult *StepResult, er
return stepResult, errors.Wrap(err, "read message failed") return stepResult, errors.Wrap(err, "read message failed")
} }
case wsRead: case wsRead:
log.Info().Int64("timeout(ms)", step.WebSocket.Timeout).Msg("read only") log.Info().Int64("timeout(ms)", step.WebSocket.Timeout).Str("url", parsedURL).Msg("read only")
resp, err = readMessageWithTimeout(parsedURL, r, step) resp, err = readMessageWithTimeout(parsedURL, r, step)
if err != nil { if err != nil {
return stepResult, errors.Wrap(err, "read message failed") return stepResult, errors.Wrap(err, "read message failed")
} }
case wsWrite: case wsWrite:
log.Info().Msg("write only") log.Info().Str("url", parsedURL).Msg("write only")
err = writeWebSocket(parsedURL, r, step, stepVariables) err = writeWebSocket(parsedURL, r, step, stepVariables)
if err != nil { if err != nil {
return stepResult, errors.Wrap(err, "write message failed") return stepResult, errors.Wrap(err, "write message failed")
} }
case wsClose: case wsClose:
log.Info().Int64("timeout(ms)", step.WebSocket.Timeout).Msg("close webSocket connection") log.Info().Int64("timeout(ms)", step.WebSocket.Timeout).Str("url", parsedURL).Msg("close webSocket connection")
resp, err = closeWithTimeout(parsedURL, r, step, stepVariables) resp, err = closeWithTimeout(parsedURL, r, step, stepVariables)
if err != nil { if err != nil {
return stepResult, errors.Wrap(err, "close connection failed") return stepResult, errors.Wrap(err, "close connection failed")