Slight cleanup

training/templates/level_list.html

@@ -1,7 +1,6 @@
 {% extends 'base_training.html' %}
 
 {% load markdown_tags %}
-{% load get_supervisor from tags %}
 
 {% block content %}
 {% if request.user.is_staff %}

training/templatetags/tags.py

@@ -33,11 +33,6 @@ def colour_from_depth(depth):
     return models.TrainingItemQualification.get_colour_from_depth(depth)
 
 
-@register.filter
-def get_supervisor(tech):
-    return models.TrainingLevel.objects.get(department=tech.department, level=models.TrainingLevel.SUPERVISOR)
-
-
 @register.filter
 def get_levels_of_depth(trainee, level):
     return trainee.level_qualifications.all().exclude(confirmed_on=None).exclude(level__department=models.TrainingLevel.HAULAGE).select_related('level').filter(level__level=level)

training/tests/conftest.py

@@ -2,6 +2,7 @@ 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",
@@ -9,3 +10,10 @@ def trainee(db):
                                      email="trainee@example.com", is_active=True, is_approved=True)
     yield trainee
     trainee.delete()
+
+
+@pytest.fixture
+def level(db):
+    level = models.TrainingLevel.objects.create(description="There is no description.", level=models.TrainingLevel.TECHNICIAN)
+    yield level
+    level.delete()

training/tests/test_unit.py

@@ -17,3 +17,9 @@ def test_add_qualification(admin_client, trainee, admin_user):
     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...')
+
+
+def test_add_requirement(admin_client, level):
+    url = reverse('add_requirement', kwargs={'pk': level.pk})
+    response = admin_client.post(url)
+    assertContains(response, level.pk)