websocket: handle error (#550)

Co-authored-by: TwiN <twin@linux.com>
2023-08-16 22:48:57 -03:00
parent 67941865db
commit 7d570ce148
2 changed files with 5 additions and 1 deletions
--- a/core/endpoint.go
+++ b/core/endpoint.go
@ -346,6 +346,10 @@ func (endpoint *Endpoint) call(result *Result) {
 	} else if endpointType == EndpointTypeWS {
 		result.Connected, result.Body, err = client.QueryWebSocket(endpoint.URL, endpoint.ClientConfig, endpoint.Body)
 		result.Duration = time.Since(startTime)
+		if err != nil {
+			result.AddError(err.Error())
+			return
+		}
 	} else {
 		response, err = client.GetHTTPClient(endpoint.ClientConfig).Do(request)
 		result.Duration = time.Since(startTime)