UT: Fix AuthenticationService.needs_registration() tests

guardicore · Apr 3, 2023 · 360ad96 · 360ad96
1 parent 05bec12
commit 360ad96
Showing 1 changed file with 2 additions and 1 deletion.
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
@@ -73,14 +73,15 @@ def authentication_facade(
 
 
 def test_needs_registration__true(authentication_facade: AuthenticationFacade):
+    authentication_facade._datastore.find_user.return_value = False
     assert authentication_facade.needs_registration()
 
 
 def test_needs_registration__false(
     monkeypatch,
     authentication_facade: AuthenticationFacade,
 ):
-    User(username=USERNAME, password=PASSWORD).save()
+    authentication_facade._datastore.find_user.return_value = True
     assert not authentication_facade.needs_registration()