azure-core: ignore Connection header in tests (Azure#24450)

Flask includes that since pallets/werkzeug#2399, so test fail on newest Flask versions.
sarkar-rajarshi · Jun 9, 2022 · 13a6fce · 13a6fce
1 parent 966af81
commit 13a6fce
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/sdk/core/azure-core/tests/async_tests/test_rest_headers_async.py b/sdk/core/azure-core/tests/async_tests/test_rest_headers_async.py
@@ -12,6 +12,7 @@
 
 # flask returns these response headers, which we don't really need for these following tests
 RESPONSE_HEADERS_TO_IGNORE = [
+    "Connection",
     "Content-Type",
     "Content-Length",
     "Server",
@@ -24,7 +25,7 @@ async def _get_response_headers(request):
         response = await client.send_request(request)
         response.raise_for_status
         for header in RESPONSE_HEADERS_TO_IGNORE:
-            response.headers.pop(header)
+            response.headers.pop(header, None)
         return response.headers
     return _get_response_headers
 

diff --git a/sdk/core/azure-core/tests/test_rest_headers.py b/sdk/core/azure-core/tests/test_rest_headers.py
@@ -21,6 +21,7 @@ def _get_request_headers(header_value):
 
 # flask returns these response headers, which we don't really need for these following tests
 RESPONSE_HEADERS_TO_IGNORE = [
+    "Connection",
     "Content-Type",
     "Content-Length",
     "Server",
@@ -33,7 +34,7 @@ def _get_response_headers(request):
         response = client.send_request(request)
         response.raise_for_status
         for header in RESPONSE_HEADERS_TO_IGNORE:
-            response.headers.pop(header)
+            response.headers.pop(header, None)
         return response.headers
     return _get_response_headers