11import pytest
22from pytest_mock import MockerFixture
33
4- from common .core import middleware as middleware_module , utils
4+ from common .core import middleware as middleware_module
5+ from common .core import utils
56
67
78@pytest .fixture (autouse = True )
@@ -19,7 +20,8 @@ def test_APIResponseVersionHeaderMiddleware__valid_version_info___adds_version_h
1920 get_response = mocker .Mock (return_value = response )
2021 middleware = middleware_module .APIResponseVersionHeaderMiddleware (get_response )
2122 get_versions_from_manifest = mocker .patch .object (
22- middleware_module , "get_versions_from_manifest" ,
23+ middleware_module ,
24+ "get_versions_from_manifest" ,
2325 return_value = {"." : "v1.2.3" },
2426 )
2527
@@ -33,10 +35,13 @@ def test_APIResponseVersionHeaderMiddleware__valid_version_info___adds_version_h
3335 assert get_versions_from_manifest .call_args_list == [mocker .call ()]
3436
3537
36- @pytest .mark .parametrize ("version_info" , [
37- {"foo" : "bar" },
38- {},
39- ])
38+ @pytest .mark .parametrize (
39+ "version_info" ,
40+ [
41+ {"foo" : "bar" },
42+ {},
43+ ],
44+ )
4045def test_APIResponseVersionHeaderMiddleware__invalid_version_info___adds_unknown_header (
4146 mocker : MockerFixture ,
4247 version_info : dict [str , str ] | None ,
@@ -47,7 +52,8 @@ def test_APIResponseVersionHeaderMiddleware__invalid_version_info___adds_unknown
4752 get_response = mocker .Mock (return_value = response )
4853 middleware = middleware_module .APIResponseVersionHeaderMiddleware (get_response )
4954 get_versions_from_manifest = mocker .patch .object (
50- middleware_module , "get_versions_from_manifest" ,
55+ middleware_module ,
56+ "get_versions_from_manifest" ,
5157 return_value = version_info ,
5258 )
5359
@@ -58,4 +64,4 @@ def test_APIResponseVersionHeaderMiddleware__invalid_version_info___adds_unknown
5864 assert result == response
5965 assert response .headers ["Flagsmith-Version" ] == "unknown"
6066 assert get_response .call_args_list == [mocker .call (request )]
61- assert get_versions_from_manifest .call_args_list == [mocker .call ()]
67+ assert get_versions_from_manifest .call_args_list == [mocker .call ()]
0 commit comments