From 86ee075a3bdc62a36b7c7cc21aba6558c671c1c1 Mon Sep 17 00:00:00 2001
From: Joachim Bauch <bauch@struktur.de>
Date: Wed, 18 Jan 2023 16:45:37 +0100
Subject: [PATCH] Prevent endless loop in case of context timeout /
 cancellation.

---
 virtualsession_test.go | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/virtualsession_test.go b/virtualsession_test.go
index d8a0ffb..d4651e4 100644
--- a/virtualsession_test.go
+++ b/virtualsession_test.go
@@ -24,6 +24,7 @@ package signaling
 import (
 	"context"
 	"encoding/json"
+	"errors"
 	"testing"
 )
 
@@ -219,6 +220,9 @@ func TestVirtualSession(t *testing.T) {
 		messages, err := client2.GetPendingMessages(ctx)
 		if err != nil {
 			t.Error(err)
+			if errors.Is(err, context.DeadlineExceeded) || errors.Is(err, context.Canceled) {
+				break
+			}
 		}
 
 		receivedMessages = append(receivedMessages, messages...)