test(scaletest): fix websocket error during close (#11879)

mafredri · web-flow · commit 3e89ba23e546 · 2024-01-29T13:42:30.000+02:00
Fixes #11735
diff --git a/scaletest/workspacetraffic/run_test.go b/scaletest/workspacetraffic/run_test.go
@@ -296,6 +296,13 @@ func TestRun(t *testing.T) {
 			if err == nil || errors.Is(err, io.EOF) {
 				return
 			}
+			// Ignore policy violations, we expect these, e.g.:
+			//
+			// 	failed to get reader: received close frame: status = StatusPolicyViolation and reason = "timed out"
+			if websocket.CloseStatus(err) == websocket.StatusPolicyViolation {
+				return
+			}
+
 			t.Error(err)
 		}))
 		defer srv.Close()
