chore: linting

Author: baywet

kiota_http/middleware/redirect_handler.py

@@ -77,9 +77,7 @@ async def send(
                 request, f"RedirectHandler_send - redirect {len(history)}"
             )
             response = await super().send(request, transport)
-            _redirect_span.set_attribute(
-                HTTP_RESPONSE_STATUS_CODE, response.status_code
-            )
+            _redirect_span.set_attribute(HTTP_RESPONSE_STATUS_CODE, response.status_code)
             redirect_location = self.get_redirect_location(response)
 
             if redirect_location and current_options.should_redirect:

kiota_http/middleware/url_replace_handler.py

@@ -1,8 +1,6 @@
 import httpx
 from kiota_abstractions.request_option import RequestOption
-from opentelemetry.semconv.attributes.url_attributes import (
-    URL_FULL
-)
+from opentelemetry.semconv.attributes.url_attributes import (URL_FULL)
 
 from .middleware import BaseMiddleware
 from .options import UrlReplaceHandlerOption

tests/conftest.py

@@ -13,10 +13,12 @@
 
 from .helpers import MockTransport, MockErrorObject, MockResponseObject, OfficeLocation
 
+
 @pytest.fixture
 def sample_headers():
     return {"Content-Type": "application/json"}
 
+
 @pytest.fixture
 def auth_provider():
     return AnonymousAuthenticationProvider()
@@ -26,6 +28,7 @@ def auth_provider():
 def request_info():
     return RequestInformation()
 
+
 @pytest.fixture
 def mock_async_transport():
     return MockTransport()
@@ -57,27 +60,32 @@ def mock_error_500_map():
         "500": Exception("Internal Server Error"),
     }
 
+
 @pytest.fixture
 def mock_apierror_map(sample_headers):
     return {
         "400": APIError("Resource not found", 400, sample_headers),
         "500": APIError("Custom Internal Server Error", 500, sample_headers)
     }
 
+
 @pytest.fixture
 def mock_apierror_XXX_map(sample_headers):
-    return {"XXX": APIError("OdataError",400, sample_headers)}
-    
+    return {"XXX": APIError("OdataError", 400, sample_headers)}
+
+
 @pytest.fixture
 def mock_request_adapter(sample_headers):
     resp = httpx.Response(json={'error': 'not found'}, status_code=404, headers=sample_headers)
     mock_request_adapter = AsyncMock
     mock_request_adapter.get_http_response_message = AsyncMock(return_value=resp)
 
+
 @pytest.fixture
 def simple_error_response(sample_headers):
     return httpx.Response(json={'error': 'not found'}, status_code=404, headers=sample_headers)
 
+
 @pytest.fixture
 def simple_success_response(sample_headers):
     return httpx.Response(json={'message': 'Success!'}, status_code=200, headers=sample_headers)
@@ -153,9 +161,7 @@ def mock_users_response(mocker):
 
 @pytest.fixture
 def mock_primitive_collection_response(sample_headers):
-    return httpx.Response(
-        200, json=[12.1, 12.2, 12.3, 12.4, 12.5], headers=sample_headers
-    )
+    return httpx.Response(200, json=[12.1, 12.2, 12.3, 12.4, 12.5], headers=sample_headers)
 
 
 @pytest.fixture

tests/helpers/mock_async_transport.py

@@ -1,5 +1,15 @@
 import httpx
 
+
 class MockTransport():
+
     async def handle_async_request(self, request):
-        return httpx.Response(200, request=request, content=b'Hello World', headers={"Content-Type": "application/json", "test": "test_response_header"}) 
+        return httpx.Response(
+            200,
+            request=request,
+            content=b'Hello World',
+            headers={
+                "Content-Type": "application/json",
+                "test": "test_response_header"
+            }
+        )

tests/middleware_tests/test_base_middleware.py

@@ -9,6 +9,7 @@ def test_next_is_none():
     middleware = BaseMiddleware()
     assert middleware.next is None
 
+
 def test_span_created(request_info):
     """Ensures the current span is returned and the parent_span is not set."""
     middleware = BaseMiddleware()

tests/middleware_tests/test_headers_inspection_handler.py

@@ -26,47 +26,42 @@ def test_custom_config():
 
     options = HeadersInspectionHandlerOption(inspect_request_headers=False)
     assert not options.inspect_request_headers
-    
-    
+
+
 def test_headers_inspection_handler_construction():
     """
     Ensures the Header Inspection handler instance is set.
     """
     handler = HeadersInspectionHandler()
     assert handler
-    
+
+
 @pytest.mark.asyncio
 async def test_headers_inspection_handler_gets_headers():
+
     def request_handler(request: httpx.Request):
         return httpx.Response(
-            200,
-            json={"text": "Hello, world!"},
-            headers={'test_response': 'test_response_header'}
+            200, json={"text": "Hello, world!"}, headers={'test_response': 'test_response_header'}
         )
+
     handler = HeadersInspectionHandler()
-    
+
     # First request
     request = httpx.Request(
-        'GET',
-        'https://localhost',
-        headers={'test_request': 'test_request_header'}
+        'GET', 'https://localhost', headers={'test_request': 'test_request_header'}
     )
     mock_transport = httpx.MockTransport(request_handler)
     resp = await handler.send(request, mock_transport)
     assert resp.status_code == 200
     assert handler.options.request_headers.try_get('test_request') == {'test_request_header'}
     assert handler.options.response_headers.try_get('test_response') == {'test_response_header'}
-    
+
     # Second request
     request2 = httpx.Request(
-        'GET',
-        'https://localhost',
-        headers={'test_request_2': 'test_request_header_2'}
+        'GET', 'https://localhost', headers={'test_request_2': 'test_request_header_2'}
     )
     resp = await handler.send(request2, mock_transport)
     assert resp.status_code == 200
     assert not handler.options.request_headers.try_get('test_request') == {'test_request_header'}
     assert handler.options.request_headers.try_get('test_request_2') == {'test_request_header_2'}
     assert handler.options.response_headers.try_get('test_response') == {'test_response_header'}
-    
-    

tests/middleware_tests/test_parameters_name_decoding_handler.py

@@ -5,6 +5,8 @@
 from kiota_http.middleware.options import ParametersNameDecodingHandlerOption
 
 OPTION_KEY = "ParametersNameDecodingHandlerOption"
+
+
 def test_no_config():
     """
     Test that default values are used if no custom confguration is passed
@@ -19,9 +21,7 @@ def test_custom_options():
     """
     Test that default configuration is overrriden if custom configuration is provided
     """
-    options = ParametersNameDecodingHandlerOption(
-        enable=False, characters_to_decode=[".", "-"]
-    )
+    options = ParametersNameDecodingHandlerOption(enable=False, characters_to_decode=[".", "-"])
     handler = ParametersNameDecodingHandler(options)
 
     assert handler.options.enabled is not True
@@ -35,24 +35,40 @@ async def test_decodes_query_parameter_names_only():
     Test that only query parameter names are decoded
     """
     encoded_decoded = [
-    ("http://localhost?%24select=diplayName&api%2Dversion=2", "http://localhost?$select=diplayName&api-version=2"),
-    ("http://localhost?%24select=diplayName&api%7Eversion=2", "http://localhost?$select=diplayName&api~version=2"),
-    ("http://localhost?%24select=diplayName&api%2Eversion=2", "http://localhost?$select=diplayName&api.version=2"),
-    ("http://localhost:888?%24select=diplayName&api%2Dversion=2", "http://localhost:888?$select=diplayName&api-version=2"),
-    ("http://localhost", "http://localhost"),
-    ("https://google.com/?q=1%2b2", "https://google.com/?q=1%2b2"),
-    ("https://google.com/?q=M%26A", "https://google.com/?q=M%26A"),
-    ("https://google.com/?q=1%2B2", "https://google.com/?q=1%2B2"), # Values are not decoded
-    ("https://google.com/?q=M%26A", "https://google.com/?q=M%26A"), # Values are not decoded
-    ("https://google.com/?q%2D1=M%26A", "https://google.com/?q-1=M%26A"), # Values are not decoded but params are
-    ("https://google.com/?q%2D1&q=M%26A=M%26A", "https://google.com/?q-1&q=M%26A=M%26A"), # Values are not decoded but params are
-    ("https://graph.microsoft.com?%24count=true&query=%24top&created%2Din=2022-10-05&q=1%2b2&q2=M%26A&subject%2Ename=%7eWelcome&%24empty",
-     "https://graph.microsoft.com?$count=true&query=%24top&created-in=2022-10-05&q=1%2b2&q2=M%26A&subject.name=%7eWelcome&$empty")
+        (
+            "http://localhost?%24select=diplayName&api%2Dversion=2",
+            "http://localhost?$select=diplayName&api-version=2"
+        ),
+        (
+            "http://localhost?%24select=diplayName&api%7Eversion=2",
+            "http://localhost?$select=diplayName&api~version=2"
+        ),
+        (
+            "http://localhost?%24select=diplayName&api%2Eversion=2",
+            "http://localhost?$select=diplayName&api.version=2"
+        ),
+        (
+            "http://localhost:888?%24select=diplayName&api%2Dversion=2",
+            "http://localhost:888?$select=diplayName&api-version=2"
+        ),
+        ("http://localhost", "http://localhost"),
+        ("https://google.com/?q=1%2b2", "https://google.com/?q=1%2b2"),
+        ("https://google.com/?q=M%26A", "https://google.com/?q=M%26A"),
+        ("https://google.com/?q=1%2B2", "https://google.com/?q=1%2B2"),  # Values are not decoded
+        ("https://google.com/?q=M%26A", "https://google.com/?q=M%26A"),  # Values are not decoded
+        ("https://google.com/?q%2D1=M%26A",
+         "https://google.com/?q-1=M%26A"),  # Values are not decoded but params are
+        ("https://google.com/?q%2D1&q=M%26A=M%26A",
+         "https://google.com/?q-1&q=M%26A=M%26A"),  # Values are not decoded but params are
+        (
+            "https://graph.microsoft.com?%24count=true&query=%24top&created%2Din=2022-10-05&q=1%2b2&q2=M%26A&subject%2Ename=%7eWelcome&%24empty",
+            "https://graph.microsoft.com?$count=true&query=%24top&created-in=2022-10-05&q=1%2b2&q2=M%26A&subject.name=%7eWelcome&$empty"
+        )
     ]
-    
+
     def request_handler(request: httpx.Request):
         return httpx.Response(200, json={"text": "Hello, world!"})
-    
+
     handler = ParametersNameDecodingHandler()
     for encoded, decoded in encoded_decoded:
         request = httpx.Request('GET', encoded)