Browse Source

Switch unit test to use decoratior instead of with waterfall

CURA-5744
Jaime van Kessel 6 years ago
parent
commit
d5dbf91a4f
1 changed files with 14 additions and 14 deletions
  1. 14 14
      tests/TestOAuth2.py

+ 14 - 14
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: