added lead and deputy inspector for org methods to User

1 files changed, 13 insertions, 2 deletions

diff --git a/ctrack/users/tests/test_models.py b/ctrack/users/tests/test_models.py
index 368be34..eff8b42 100644
--- a/ctrack/users/tests/test_models.py
+++ b/ctrack/users/tests/test_models.py
@@ -1,16 +1,27 @@
 import pytest
 
-from ctrack.organisations.models import Stakeholder
+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