Switch unit test to use decoratior instead of with waterfall

CURA-5744

1 files changed, 14 insertions, 14 deletions

diff --git a/tests/TestOAuth2.py b/tests/TestOAuth2.py
index 7deb712aea..312d71fd5f 100644
--- a/tests/TestOAuth2.py
+++ b/tests/TestOAuth2.py
@@ -55,22 +55,22 @@ def test_failedLogin() -> None:
     assert authorization_service.getAccessToken() is None
 
 
-def test_localAuthServer() -> None:
+@patch.object(LocalAuthorizationServer, "stop")
+@patch.object(LocalAuthorizationServer, "start")
+@patch.object(webbrowser, "open_new")
+def test_localAuthServer(webbrowser_open, start_auth_server, stop_auth_server) -> None:
     preferences = Preferences()
     authorization_service = AuthorizationService(preferences, OAUTH_SETTINGS)
-    with patch.object(webbrowser, "open_new") as webrowser_open:
-        with patch.object(LocalAuthorizationServer, "start") as start_auth_server:
-            with patch.object(LocalAuthorizationServer, "stop") as stop_auth_server:
-                authorization_service.startAuthorizationFlow()
-                assert webrowser_open.call_count == 1
-
-                # Ensure that the Authorization service tried to start the server.
-                assert start_auth_server.call_count == 1
-                assert stop_auth_server.call_count == 0
-                authorization_service._onAuthStateChanged(FAILED_AUTH_RESPONSE)
-
-                # Ensure that it stopped the server.
-                assert stop_auth_server.call_count == 1
+    authorization_service.startAuthorizationFlow()
+    assert webbrowser_open.call_count == 1
+
+    # Ensure that the Authorization service tried to start the server.
+    assert start_auth_server.call_count == 1
+    assert stop_auth_server.call_count == 0
+    authorization_service._onAuthStateChanged(FAILED_AUTH_RESPONSE)
+
+    # Ensure that it stopped the server.
+    assert stop_auth_server.call_count == 1
 
 
 def test_loginAndLogout() -> None: