revert change to aiohttp

iscai-msft · Mar 4, 2021 · 9159595 · 9159595
1 parent bf96ec9
commit 9159595
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/sdk/core/azure-core/azure/core/pipeline/transport/_aiohttp.py b/sdk/core/azure-core/azure/core/pipeline/transport/_aiohttp.py
@@ -267,7 +267,7 @@ def __init__(self, request: HttpRequest, aiohttp_response: aiohttp.ClientRespons
         self.status_code = aiohttp_response.status
         self.headers = CIMultiDict(aiohttp_response.headers)
         self.reason = aiohttp_response.reason
-        self.content_type = self.headers.get('content-type')
+        self.content_type = aiohttp_response.headers.get('content-type')
         self._body = None
 
     def body(self) -> bytes:

diff --git a/sdk/core/azure-core/tests/async_tests/test_universal_http_async.py b/sdk/core/azure-core/tests/async_tests/test_universal_http_async.py
@@ -126,6 +126,8 @@ async def test_aiohttp_response_text():
 def test_repr():
     res = _create_aiohttp_response(
         b'\xef\xbb\xbf56',
-        {'Content-Type': 'text/plain'}
+        {}
     )
+    res.content_type = "text/plain"
+
     assert repr(res) == "<AioHttpTransportResponse: 200 OK, Content-Type: text/plain>"