From 0b4a7f7ac96b09d71b088889ecc07714ced6a8f1 Mon Sep 17 00:00:00 2001
From: Mathias Fredriksson <mafredri@gmail.com>
Date: Mon, 29 Jan 2024 13:29:59 +0200
Subject: [PATCH] test(scaletest): fix websocket error during close

Fixes #11735
---
 scaletest/workspacetraffic/run_test.go | 7 +++++++
 1 file changed, 7 insertions(+)

diff --git a/scaletest/workspacetraffic/run_test.go b/scaletest/workspacetraffic/run_test.go
index a177390f9fd96..4e54c45da4bbf 100644
--- 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()