From 8be95a2efb31d598936f2a4195f375555ed5237a Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Zolt=C3=A1n=20Papp?= <zoltan.pmail@gmail.com>
Date: Mon, 2 Jun 2025 17:11:25 +0200
Subject: [PATCH] Fix test

---
 client/internal/peer/status_test.go | 7 +++++--
 1 file changed, 5 insertions(+), 2 deletions(-)

diff --git a/client/internal/peer/status_test.go b/client/internal/peer/status_test.go
index 4d7a4ca78..33ea4ee0b 100644
--- a/client/internal/peer/status_test.go
+++ b/client/internal/peer/status_test.go
@@ -1,6 +1,7 @@
 package peer
 
 import (
+	"context"
 	"errors"
 	"sync"
 	"testing"
@@ -99,10 +100,12 @@ func TestGetPeerStateChangeNotifierLogic(t *testing.T) {
 	err := status.UpdatePeerRelayedStateToDisconnected(peerState)
 	assert.NoError(t, err, "shouldn't return error")
 
+	timeoutCtx, cancel := context.WithTimeout(context.Background(), 1*time.Second)
+	defer cancel()
 	select {
 	case <-sub.eventsChan:
-	default:
-		t.Errorf("channel wasn't closed after update")
+	case <-timeoutCtx.Done():
+		t.Errorf("timed out waiting for event")
 	}
 }