UT: Strengthen the assertion in __fails_if_token_invalid()

guardicore · Apr 3, 2023 · a2ce7d3 · a2ce7d3
1 parent 88091d5
commit a2ce7d3
Showing 1 changed file with 7 additions and 3 deletions.
diff --git a/...nit_tests/monkey_island/cc/services/authentication_service/test_authentication_service.py b/...nit_tests/monkey_island/cc/services/authentication_service/test_authentication_service.py
@@ -11,7 +11,11 @@
     AuthenticationFacade,
 )
 from monkey_island.cc.services.authentication_service.setup import setup_authentication
-from monkey_island.cc.services.authentication_service.token import TokenGenerator, TokenParser
+from monkey_island.cc.services.authentication_service.token import (
+    TokenGenerator,
+    TokenParser,
+    TokenValidationError,
+)
 from monkey_island.cc.services.authentication_service.user import User
 
 USERNAME = "user1"
@@ -157,9 +161,9 @@ def test_generate_new_token_pair__fails_if_token_invalid(
     user = User(username=USERNAME, password=PASSWORD, fs_uniquifier="a")
     user.save()
     refresh_token = authentication_facade.generate_refresh_token(user)
-    mock_token_parser.parse.side_effect = Exception()
+    mock_token_parser.parse.side_effect = TokenValidationError()
 
-    with pytest.raises(Exception):
+    with pytest.raises(TokenValidationError):
         authentication_facade.generate_new_token_pair(refresh_token)