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 ( import (
"context" "context"
"encoding/json" "encoding/json"
"errors"
"testing" "testing"
) )
@ -219,6 +220,9 @@ func TestVirtualSession(t *testing.T) {
messages, err := client2.GetPendingMessages(ctx) messages, err := client2.GetPendingMessages(ctx)
if err != nil { if err != nil {
t.Error(err) t.Error(err)
if errors.Is(err, context.DeadlineExceeded) || errors.Is(err, context.Canceled) {
break
}
} }
receivedMessages = append(receivedMessages, messages...) receivedMessages = append(receivedMessages, messages...)