Put initial_request to None.

Author: pujawadare

google/api_core/bidi.py

@@ -286,6 +286,7 @@ def close(self):
 
         # Put None in request queue to signal termination.
         self._request_queue.put(None)
+        self._initial_request = None
         self._request_generator = None
         self._callbacks = []
         # Don't set self.call to None. Keep it around so that send/recv can

tests/asyncio/test_bidi_async.py

@@ -252,6 +252,7 @@ async def test_close(self):
         assert bidi_rpc.pending_requests == 1
         assert await bidi_rpc._request_queue.get() is None
         # ensure request and callbacks are cleaned up
+        assert bidi_rpc._initial_request is None
         assert not bidi_rpc._callbacks
 
     @pytest.mark.asyncio
@@ -266,6 +267,7 @@ async def test_close_with_no_rpc(self):
         assert bidi_rpc.pending_requests == 1
         assert await bidi_rpc._request_queue.get() is None
         # ensure request and callbacks are cleaned up
+        assert bidi_rpc._initial_request is None
         assert not bidi_rpc._callbacks
 
     @pytest.mark.asyncio

tests/unit/test_bidi.py

@@ -298,6 +298,7 @@ def test_close(self):
         assert bidi_rpc.pending_requests == 1
         assert bidi_rpc._request_queue.get() is None
         # ensure request and callbacks are cleaned up
+        assert bidi_rpc._initial_request is None
         assert not bidi_rpc._callbacks
 
     def test_close_with_no_rpc(self):