p41275903
/
monkey
forked from p15670423/monkey
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Tests: Clarify needs_registration() tests for AuthenticationService

This commit is contained in:
Mike Salvatore 2021-11-18 12:26:42 -05:00
parent 3994730f78
commit f08ac42ac9
1 changed files with 12 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

22
monkey/tests/unit_tests/monkey_island/cc/services/authentication/test_authentication_service.py
View File

@ -73,16 +73,8 @@ def patch_datastore_utils(
def test_needs_registration__true(tmp_path): def test_needs_registration__true(tmp_path):
mock_user_datastore = MockUserDatastore(lambda: True, None, None) has_registered_users = False
mock_user_datastore = MockUserDatastore(lambda: has_registered_users, None, None)
a_s = AuthenticationService()
a_s.initialize(tmp_path, mock_user_datastore)
assert not a_s.needs_registration()
def test_needs_registration__false(tmp_path):
mock_user_datastore = MockUserDatastore(lambda: False, None, None)
a_s = AuthenticationService() a_s = AuthenticationService()
a_s.initialize(tmp_path, mock_user_datastore) a_s.initialize(tmp_path, mock_user_datastore)
@ -90,6 +82,16 @@ def test_needs_registration__false(tmp_path):
assert a_s.needs_registration() assert a_s.needs_registration()
def test_needs_registration__false(tmp_path):
has_registered_users = True
mock_user_datastore = MockUserDatastore(lambda: has_registered_users, None, None)
a_s = AuthenticationService()
a_s.initialize(tmp_path, mock_user_datastore)
assert not a_s.needs_registration()
@pytest.mark.parametrize("error", [InvalidRegistrationCredentialsError, AlreadyRegisteredError]) @pytest.mark.parametrize("error", [InvalidRegistrationCredentialsError, AlreadyRegisteredError])
def test_register_new_user__fails( def test_register_new_user__fails(
tmp_path, mock_reset_datastore_encryptor, mock_reset_database, error tmp_path, mock_reset_datastore_encryptor, mock_reset_database, error