async def test_content(self): mock_response = mock.AsyncMock() mock_response.content.read.return_value = mock.sentinel.read combined_response = aiohttp_requests._CombinedResponse( response=mock_response) content = await combined_response.content() assert content == mock.sentinel.read
async def test_content_compressed(self, urllib3_mock): rm = core.RequestMatch("url", headers={"Content-Encoding": "gzip"}, payload="compressed") response = await rm.build_response(core.URL("url")) combined_response = aiohttp_requests._CombinedResponse( response=response) content = await combined_response.content() urllib3_mock.assert_called_once() assert content == "decompressed"
async def _get_body(response): """Access the response body from an HTTP response. Args: response (~requests.Response): The HTTP response object. Returns: bytes: The body of the ``response``. """ wrapped_response = aiohttp_requests._CombinedResponse(response) content = await wrapped_response.data.read() return content
async def test_raw_content(self): mock_response = mock.AsyncMock() mock_response.content.read.return_value = mock.sentinel.read combined_response = aiohttp_requests._CombinedResponse( response=mock_response) raw_content = await combined_response.raw_content() assert raw_content == mock.sentinel.read # Second call to validate the preconfigured path. combined_response._raw_content = mock.sentinel.stored_raw raw_content = await combined_response.raw_content() assert raw_content == mock.sentinel.stored_raw
async def test_download_full(self, add_files, authorized_transport, checksum): for info in ALL_FILES: actual_contents = self._get_contents(info) blob_name = get_blob_name(info) # Create the actual download object. media_url = utils.DOWNLOAD_URL_TEMPLATE.format(blob_name=blob_name) download = self._make_one(media_url, checksum=checksum) # Consume the resource. response = await download.consume(authorized_transport) response = tr_requests._CombinedResponse(response) assert response.status == http_client.OK content = await self._read_response_content(response) assert content == actual_contents await check_tombstoned(download, authorized_transport)
def test__is_compressed_not(self): response = core.CallbackResult(headers={"Content-Encoding": "not"}) combined_response = aiohttp_requests._CombinedResponse(response) compressed = combined_response._is_compressed() assert not compressed
async def _get_body(response): # TODO(asyncio): This code differs from sync. Leaving this comment in case # this difference causes downstream issues. wrapped_response = aiohttp_requests._CombinedResponse(response) content = await wrapped_response.raw_content() return content