mirror of
https://github.com/GSA/notifications-admin.git
synced 2026-07-17 13:09:49 -04:00
0d350bdaf8
now that we no longer set it since https://github.com/alphagov/notifications-admin/pull/3841 was merged, we don't need to remove it either. And we can remove checks that expect it when cleaning up the session. And the unit tests that make sure we ignore it if it's in the session. So long, session['invited_user'] and session['invited_org_user']!
138 lines
4.6 KiB
Python
138 lines
4.6 KiB
Python
import pytest
|
|
|
|
from app.models.user import AnonymousUser, InvitedOrgUser, InvitedUser, User
|
|
from tests.conftest import USER_ONE_ID
|
|
|
|
|
|
def test_anonymous_user(app_):
|
|
assert AnonymousUser().is_authenticated is False
|
|
assert AnonymousUser().logged_in_elsewhere() is False
|
|
assert AnonymousUser().default_organisation.name is None
|
|
assert AnonymousUser().default_organisation.crown is None
|
|
assert AnonymousUser().default_organisation.agreement_signed is None
|
|
assert AnonymousUser().default_organisation.domains == []
|
|
assert AnonymousUser().default_organisation.organisation_type is None
|
|
assert AnonymousUser().default_organisation.request_to_go_live_notes is None
|
|
|
|
|
|
def test_user(app_):
|
|
user_data = {'id': 1,
|
|
'name': 'Test User',
|
|
'email_address': 'test@user.gov.uk',
|
|
'mobile_number': '+4412341234',
|
|
'state': 'pending',
|
|
'failed_login_count': 0,
|
|
'platform_admin': False,
|
|
}
|
|
user = User(user_data)
|
|
|
|
assert user.id == 1
|
|
assert user.name == 'Test User'
|
|
assert user.email_address == 'test@user.gov.uk'
|
|
assert user.mobile_number == '+4412341234'
|
|
assert user.state == 'pending'
|
|
|
|
# user has ten failed logins before being locked
|
|
assert user.max_failed_login_count == app_.config['MAX_FAILED_LOGIN_COUNT'] == 10
|
|
assert user.failed_login_count == 0
|
|
assert user.locked is False
|
|
|
|
# set failed logins to threshold
|
|
user.failed_login_count = app_.config['MAX_FAILED_LOGIN_COUNT']
|
|
assert user.locked is True
|
|
|
|
with pytest.raises(TypeError):
|
|
user.has_permissions('to_do_bad_things')
|
|
|
|
|
|
def test_activate_user(app_, api_user_pending, mock_activate_user):
|
|
assert User(api_user_pending).activate() == User(api_user_pending)
|
|
mock_activate_user.assert_called_once_with(api_user_pending['id'])
|
|
|
|
|
|
def test_activate_user_already_active(app_, api_user_active, mock_activate_user):
|
|
assert User(api_user_active).activate() == User(api_user_active)
|
|
assert mock_activate_user.called is False
|
|
|
|
|
|
@pytest.mark.parametrize('is_platform_admin, value_in_session, expected_result', [
|
|
(True, True, False),
|
|
(True, False, True),
|
|
(True, None, True),
|
|
(False, True, False),
|
|
(False, False, False),
|
|
(False, None, False),
|
|
])
|
|
def test_platform_admin_flag_set_in_session(client, mocker, is_platform_admin, value_in_session, expected_result):
|
|
session_dict = {}
|
|
if value_in_session is not None:
|
|
session_dict['disable_platform_admin_view'] = value_in_session
|
|
|
|
mocker.patch.dict('app.models.user.session', values=session_dict, clear=True)
|
|
|
|
assert User({'platform_admin': is_platform_admin}).platform_admin == expected_result
|
|
|
|
|
|
def test_has_live_services(
|
|
client_request,
|
|
mock_get_non_empty_organisations_and_services_for_user,
|
|
fake_uuid,
|
|
):
|
|
user = User({
|
|
'id': fake_uuid,
|
|
'platform_admin': False,
|
|
})
|
|
assert len(user.live_services) == 5
|
|
for service in user.live_services:
|
|
assert service.live
|
|
|
|
|
|
def test_has_live_services_when_there_are_no_services(
|
|
client_request,
|
|
mock_get_organisations_and_services_for_user,
|
|
fake_uuid,
|
|
):
|
|
assert User({
|
|
'id': fake_uuid,
|
|
'platform_admin': False,
|
|
}).live_services == []
|
|
|
|
|
|
def test_has_live_services_when_service_is_not_live(
|
|
client_request,
|
|
mock_get_empty_organisations_and_one_service_for_user,
|
|
fake_uuid,
|
|
):
|
|
assert User({
|
|
'id': fake_uuid,
|
|
'platform_admin': False,
|
|
}).live_services == []
|
|
|
|
|
|
def test_invited_user_from_session_uses_id(client, mocker, mock_get_invited_user_by_id):
|
|
session_dict = {'invited_user_id': USER_ONE_ID}
|
|
mocker.patch.dict('app.models.user.session', values=session_dict, clear=True)
|
|
|
|
assert InvitedUser.from_session().id == USER_ONE_ID
|
|
|
|
mock_get_invited_user_by_id.assert_called_once_with(USER_ONE_ID)
|
|
|
|
|
|
def test_invited_user_from_session_returns_none_if_nothing_present(client, mocker):
|
|
mocker.patch.dict('app.models.user.session', values={}, clear=True)
|
|
assert InvitedUser.from_session() is None
|
|
|
|
|
|
def test_invited_org_user_from_session_uses_id(client, mocker, mock_get_invited_org_user_by_id, sample_org_invite):
|
|
session_dict = {'invited_org_user_id': sample_org_invite['id']}
|
|
mocker.patch.dict('app.models.user.session', values=session_dict, clear=True)
|
|
|
|
assert InvitedOrgUser.from_session().id == sample_org_invite['id']
|
|
|
|
mock_get_invited_org_user_by_id.assert_called_once_with(sample_org_invite['id'])
|
|
|
|
|
|
def test_invited_org_user_from_session_returns_none_if_nothing_present(client, mocker):
|
|
mocker.patch.dict('app.models.user.session', values={}, clear=True)
|
|
assert InvitedOrgUser.from_session() is None
|