cleanup default values and add test coverage

Author: ohmayr

google/auth/aio/transport/__init__.py

@@ -71,11 +71,11 @@ def headers(self) -> Mapping[str, str]:
         raise NotImplementedError("headers must be implemented.")
 
     @abc.abstractmethod
-    async def content(self, chunk_size: int = 1024) -> AsyncGenerator[bytes, None]:
+    async def content(self, chunk_size: int) -> AsyncGenerator[bytes, None]:
         """The raw response content.
 
         Args:
-            chunk_size (int): The size of each chunk. Defaults to 1024.
+            chunk_size (int): The size of each chunk.
 
         Yields:
             AsyncGenerator[bytes, None]: An asynchronous generator yielding

google/auth/aio/transport/aiohttp.py

@@ -58,7 +58,7 @@ def headers(self) -> Mapping[str, str]:
     @_helpers.copy_docstring(transport.Response)
     async def content(self, chunk_size: int = 1024) -> AsyncGenerator[bytes, None]:
         try:
-            async for chunk in self._response.content.iter_chunked(chunk_size):
+            async for chunk in self._response.content.iter_chunked(chunk_size): # pragma: no branch
                 yield chunk
         except aiohttp.ClientPayloadError as exc:
             raise exceptions.ResponseError(

tests/transport/aio/test_aiohttp.py

@@ -125,6 +125,19 @@ async def test_request_call_success(self, aiohttp_request):
             assert response.headers == {"Content-Type": "application/json"}
             content = b"".join([chunk async for chunk in response.content()])
             assert content == b"Cavefish have no sight."
+    
+    async def test_request_call_success_with_provided_session(self):
+        mock_session = aiohttp.ClientSession()
+        request = auth_aiohttp.Request(mock_session)
+        with aioresponses() as m:
+            mocked_chunks = [b"Cavefish ", b"have ", b"no ", b"sight."]
+            mocked_response = b"".join(mocked_chunks)
+            m.get("http://example.com", status=200, body=mocked_response)
+            response = await request("http://example.com")
+            assert response.status_code == 200
+            assert response.headers == {"Content-Type": "application/json"}
+            content = b"".join([chunk async for chunk in response.content()])
+            assert content == b"Cavefish have no sight."
 
     async def test_request_call_raises_client_error(self, aiohttp_request):
         with aioresponses() as m: