Add a test!

Just one

training/tests/conftest.py

@@ -0,0 +1,11 @@
+import pytest
+from training import models
+from RIGS.models import Profile
+
+@pytest.fixture
+def trainee(db):
+    trainee = Profile.objects.create(username="trainee", first_name="Train", last_name="EE",
+                                                initials="TRN",
+                                                email="trainee@example.com", is_active=True, is_approved=True)
+    yield trainee
+    trainee.delete()

training/tests/test_unit.py

@@ -1,5 +1,19 @@
+import datetime
 import pytest
 
+from django.utils import timezone
+from django.urls import reverse
+
 from pytest_django.asserts import assertFormError, assertRedirects, assertContains, assertNotContains
 
-pytestmark = pytest.mark.django_db
+from training import models
+
+
+def test_add_qualification(admin_client, trainee, admin_user):
+    url = reverse('add_qualification', kwargs={'pk': trainee.pk})
+    date = (timezone.now() + datetime.timedelta(days=3)).strftime("%Y-%m-%d")
+    response = admin_client.post(url, {'date': date, 'trainee': trainee.pk, 'supervisor': trainee.pk })
+    assertFormError(response, 'form', 'date', 'Qualification date may not be in the future')
+    assertFormError(response, 'form', 'supervisor', 'One may not supervise oneself...')
+    response = admin_client.post(url, {'date': date, 'trainee': trainee.pk, 'supervisor': admin_user.pk })
+    assertFormError(response, 'form', 'supervisor', 'Selected supervisor must actually *be* a supervisor...')

training/urls.py

@@ -13,7 +13,7 @@ urlpatterns = [
          has_perm_or_supervisor('RIGS.view_profile')(views.TraineeDetail.as_view()),
          name='trainee_detail'),
     path('trainee/<int:pk>/history', has_perm_or_supervisor('RIGS.view_profile')(VersionHistory.as_view()), name='trainee_history', kwargs={'model': models.Trainee, 'app': 'training'}),  # Not picked up automatically because proxy model (I think)
-    path('trainee/<int:pk>/add_qualification/', has_perm_or_supervisor('training.add_trainingitemqualificaiton')(views.AddQualification.as_view()),
+    path('trainee/<int:pk>/add_qualification/', has_perm_or_supervisor('training.add_trainingitemqualification')(views.AddQualification.as_view()),
          name='add_qualification'),
     path('trainee/<int:pk>/edit_qualification/', has_perm_or_supervisor('training.change_trainingitemqualification')(views.EditQualification.as_view()),
          name='edit_qualification'),