fix after CR

plgd-dev · Nov 28, 2023 · a9e8964 · a9e8964
1 parent 55a61d6
commit a9e8964
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 2 deletions.
diff --git a/tcp/client_test.go b/tcp/client_test.go
@@ -752,7 +752,11 @@ func TestConnRequestMonitorCloseConnection(t *testing.T) {
 	deleteReq.SetMessageID(2)
 	_, err = cc.Do(deleteReq)
 	require.Error(t, err)
-	<-reqMonitorErr
+	select {
+	case <-reqMonitorErr:
+	case <-ctx.Done():
+		require.Fail(t, "request monitor not called")
+	}
 }
 
 func TestConnRequestMonitorDropRequest(t *testing.T) {

diff --git a/udp/client/conn_test.go b/udp/client/conn_test.go
@@ -919,7 +919,11 @@ func TestConnRequestMonitorCloseConnection(t *testing.T) {
 	deleteReq.SetMessageID(2)
 	_, err = cc.Do(deleteReq)
 	require.Error(t, err)
-	<-reqMonitorErr
+	select {
+	case <-reqMonitorErr:
+	case <-ctx.Done():
+		require.Fail(t, "request monitor not called")
+	}
 }
 
 func TestConnRequestMonitorDropRequest(t *testing.T) {