ctrack/users/tests/test_models.py


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40

import pytest

from ctrack.organisations.models import Stakeholder, Submode

pytestmark = pytest.mark.django_db


def test_user_can_be_lead_inspector(inspector1, org):
    org.lead_inspector = inspector1
    org.save()
    assert inspector1.get_lead_inspector_organisations().first() == org


def test_user_can_be_deputy_lead_inspector(inspector1, org):
    org.deputy_lead_inspector = inspector1
    org.save()
    assert inspector1.get_deputy_lead_inspector_organisations().first() == org


def test_user_get_absolute_url(user):
    assert user.get_absolute_url() == f"/users/{user.username}/"


def test_user_is_person_object(user):
    """User comes from ctrack.conftest.
    """
    assert user


def test_stakeholder_model(person, user):
    """
    A stakeholder is someone who is part of the regime but also has user access to the
    the system.
    """
    stakeholder = Stakeholder(person=person)
    org = person.organisation.name
    user.stakeholder = stakeholder
    assert user.stakeholder.person.first_name == "Toss"
    assert user.is_stakeholder is True
    assert user.get_organisation_name() == org