From 9d886a6093d57973bd92463412badbb4087faa36 Mon Sep 17 00:00:00 2001 From: Adam Dangoor Date: Fri, 30 Aug 2024 00:19:16 +0100 Subject: [PATCH] De-dent a bunch of MockVWS __enter__ Do not .start() the mock twice --- .../_requests_mock_server/decorators.py | 46 +++++++++---------- 1 file changed, 23 insertions(+), 23 deletions(-) diff --git a/src/mock_vws/_requests_mock_server/decorators.py b/src/mock_vws/_requests_mock_server/decorators.py index d4b581a97..85a17582c 100644 --- a/src/mock_vws/_requests_mock_server/decorators.py +++ b/src/mock_vws/_requests_mock_server/decorators.py @@ -116,33 +116,33 @@ def __enter__(self) -> Self: Returns: ``self``. """ - with Mocker(real_http=self._real_http) as mock: - for vws_route in self._mock_vws_api.routes: - url_pattern = urljoin( - base=self._base_vws_url, - url=f"{vws_route.path_pattern}$", + mock = Mocker(real_http=self._real_http) + for vws_route in self._mock_vws_api.routes: + url_pattern = urljoin( + base=self._base_vws_url, + url=f"{vws_route.path_pattern}$", + ) + + for vws_http_method in vws_route.http_methods: + mock.register_uri( + method=vws_http_method, + url=re.compile(url_pattern), + text=getattr(self._mock_vws_api, vws_route.route_name), ) - for vws_http_method in vws_route.http_methods: - mock.register_uri( - method=vws_http_method, - url=re.compile(url_pattern), - text=getattr(self._mock_vws_api, vws_route.route_name), - ) - - for vwq_route in self._mock_vwq_api.routes: - url_pattern = urljoin( - base=self._base_vwq_url, - url=f"{vwq_route.path_pattern}$", + for vwq_route in self._mock_vwq_api.routes: + url_pattern = urljoin( + base=self._base_vwq_url, + url=f"{vwq_route.path_pattern}$", + ) + + for vwq_http_method in vwq_route.http_methods: + mock.register_uri( + method=vwq_http_method, + url=re.compile(url_pattern), + text=getattr(self._mock_vwq_api, vwq_route.route_name), ) - for vwq_http_method in vwq_route.http_methods: - mock.register_uri( - method=vwq_http_method, - url=re.compile(url_pattern), - text=getattr(self._mock_vwq_api, vwq_route.route_name), - ) - self._mock = mock self._mock.start()