Prevent endless loop in case of context timeout / cancellation.

This commit is contained in:
Joachim Bauch 2023-01-18 16:45:37 +01:00
parent 05b9f4d6c9
commit 86ee075a3b
No known key found for this signature in database
GPG key ID: 77C1D22D53E15F02

View file

@ -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...)