Fixup testcase

connectrpc · Jun 3, 2024 · b252beb · b252beb
1 parent 077f6a0
commit b252beb
Showing 1 changed file with 5 additions and 7 deletions.
diff --git a/interceptor_test.go b/interceptor_test.go
@@ -1629,14 +1629,12 @@ func TestStreamingClientContextCancellation(t *testing.T) {
 	require.NoError(t, stream.Send(nil))
 	require.NoError(t, stream.CloseRequest())
 	resp, err := stream.Receive()
-	assert.Equal(t, msg.GetData(), resp.GetData())
-	go cancel() // Close the stream before the response is received.
-	go func() {
-		_, err := stream.Receive() // Receive the end of the stream.
-		require.Error(t, err)
-	}()
-	require.NoError(t, stream.CloseResponse())
 	require.NoError(t, err)
+	assert.Equal(t, msg.GetData(), resp.GetData())
+	go cancel()               // Cancel the stream before the response is received.
+	_, err = stream.Receive() // Receive the end of the stream.
+	assert.Error(t, err)
+	assert.Nil(t, stream.CloseResponse())
 }
 
 func TestStreamingHandlerTracing(t *testing.T) {