Potentially fix tests

Pipfile

@@ -33,7 +33,6 @@ envparse = "~=0.2.0"
 gunicorn = "~=20.0.4"
 icalendar = "~=4.0.7"
 idna = "~=2.10"
-importlib-metadata = "~=3.4.0"
 lxml = "~=4.6.3"
 Markdown = "~=3.3.3"
 msgpack = "~=1.0.2"
@@ -78,6 +77,7 @@ sentry-sdk = "*"
 diff-match-patch = "*"
 python-barcode = "*"
 django-hCaptcha = "*"
+importlib-metadata = "*"
 
 [dev-packages]
 selenium = "~=3.141.0"
@@ -89,8 +89,15 @@ pytest-django = "*"
 pluggy = "*"
 pytest-splinter = "*"
 pytest = "*"
-pytest-xdist = {extras = [ "psutil",], version = "*"}
-PyPOM = {extras = [ "splinter",], version = "*"}
+pytest-reverse = "*"
 
 [requires]
 python_version = "3.9"
+
+[dev-packages.pytest-xdist]
+extras = [ "psutil",]
+version = "*"
+
+[dev-packages.PyPOM]
+extras = [ "splinter",]
+version = "*"

PyRIGS/tests/test_unit.py

@@ -19,7 +19,7 @@ from django.urls.exceptions import NoReverseMatch
 from RIGS.models import Event
 from assets.models import Asset
 from django.db import connection
-from django.test import TestCase
+from django.test import TestCase, TransactionTestCase
 from django.test.utils import override_settings
 
 
@@ -67,11 +67,18 @@ class TestSampleDataGenerator(TestCase):
         assert Event.objects.all().count() == 0
 
 
-class TestSampleDataGenerator(TestCase):
-    @override_settings(DEBUG=True)
-    def setUp(self):
+@override_settings(DEBUG=True)
+class TestsWithSampleDataGenerator(TransactionTestCase):
+    @classmethod
+    def setUpClass(cls):
+        super().setUpClass()
         call_command('generateSampleData')
 
+    @classmethod
+    def tearDownClass(cls):
+        call_command('deleteSampleData')
+        super().tearDownClass()
+
     def test_unauthenticated(self):  # Nothing should be available to the unauthenticated
         for url in find_urls_recursive(urls.urlpatterns):
             request_url = get_request_url(url)

RIGS/management/commands/deleteSampleData.py

@@ -35,6 +35,8 @@ class Command(BaseCommand):
         self.delete_objects(tmodels.TrainingCategory)
         self.delete_objects(tmodels.TrainingItem)
         self.delete_objects(tmodels.TrainingLevel)
+        self.delete_objects(tmodels.TrainingItemQualification)
+        self.delete_objects(tmodels.TrainingLevelRequirement)
 
     def delete_objects(self, model):
         for obj in model.objects.all():

training/migrations/0013_auto_20220104_1940.py

@@ -0,0 +1,59 @@
+# Generated by Django 3.2.11 on 2022-01-04 19:40
+
+from django.db import migrations, models
+import django.db.models.deletion
+
+
+class Migration(migrations.Migration):
+
+    dependencies = [
+        ('training', '0012_auto_20220102_2051'),
+    ]
+
+    operations = [
+        migrations.AlterField(
+            model_name='trainingitem',
+            name='category',
+            field=models.ForeignKey(on_delete=django.db.models.deletion.CASCADE, related_name='items', to='training.trainingcategory'),
+        ),
+        migrations.AlterField(
+            model_name='trainingitemqualification',
+            name='item',
+            field=models.ForeignKey(on_delete=django.db.models.deletion.CASCADE, to='training.trainingitem'),
+        ),
+        migrations.AlterField(
+            model_name='trainingitemqualification',
+            name='supervisor',
+            field=models.ForeignKey(on_delete=django.db.models.deletion.CASCADE, related_name='qualifications_granted', to='training.trainee'),
+        ),
+        migrations.AlterField(
+            model_name='trainingitemqualification',
+            name='trainee',
+            field=models.ForeignKey(on_delete=django.db.models.deletion.CASCADE, related_name='qualifications_obtained', to='training.trainee'),
+        ),
+        migrations.AlterField(
+            model_name='traininglevelqualification',
+            name='confirmed_by',
+            field=models.ForeignKey(null=True, on_delete=django.db.models.deletion.CASCADE, related_name='confirmer', to='training.trainee'),
+        ),
+        migrations.AlterField(
+            model_name='traininglevelqualification',
+            name='level',
+            field=models.ForeignKey(on_delete=django.db.models.deletion.CASCADE, to='training.traininglevel'),
+        ),
+        migrations.AlterField(
+            model_name='traininglevelqualification',
+            name='trainee',
+            field=models.ForeignKey(on_delete=django.db.models.deletion.CASCADE, related_name='level_qualifications', to='training.trainee'),
+        ),
+        migrations.AlterField(
+            model_name='traininglevelrequirement',
+            name='item',
+            field=models.ForeignKey(on_delete=django.db.models.deletion.CASCADE, to='training.trainingitem'),
+        ),
+        migrations.AlterField(
+            model_name='traininglevelrequirement',
+            name='level',
+            field=models.ForeignKey(on_delete=django.db.models.deletion.CASCADE, related_name='requirements', to='training.traininglevel'),
+        ),
+    ]

training/models.py

@@ -57,7 +57,7 @@ class TrainingCategory(models.Model):
 
 class TrainingItem(models.Model):
     reference_number = models.IntegerField()
-    category = models.ForeignKey('TrainingCategory', related_name='items', on_delete=models.RESTRICT)
+    category = models.ForeignKey('TrainingCategory', related_name='items', on_delete=models.CASCADE)
     name = models.CharField(max_length=50)
     active = models.BooleanField(default=True)
 
@@ -90,12 +90,12 @@ class TrainingItemQualification(models.Model):
         (COMPLETE, 'Training Complete'),
         (PASSED_OUT, 'Passed Out'),
     )
-    item = models.ForeignKey('TrainingItem', on_delete=models.RESTRICT)
+    item = models.ForeignKey('TrainingItem', on_delete=models.CASCADE)
     depth = models.IntegerField(choices=CHOICES)
-    trainee = models.ForeignKey('Trainee', related_name='qualifications_obtained', on_delete=models.RESTRICT)
+    trainee = models.ForeignKey('Trainee', related_name='qualifications_obtained', on_delete=models.CASCADE)
     date = models.DateField()
     # TODO Remember that some training is external. Support for making an organisation the trainer?
-    supervisor = models.ForeignKey('Trainee', related_name='qualifications_granted', on_delete=models.RESTRICT)
+    supervisor = models.ForeignKey('Trainee', related_name='qualifications_granted', on_delete=models.CASCADE)
     notes = models.TextField(blank=True)
     # TODO Maximum depth - some things stop at Complete and you can't be passed out in them
 
@@ -232,8 +232,8 @@ class TrainingLevel(models.Model, RevisionMixin):
 
 @reversion.register
 class TrainingLevelRequirement(models.Model, RevisionMixin):
-    level = models.ForeignKey('TrainingLevel', related_name='requirements', on_delete=models.RESTRICT)
-    item = models.ForeignKey('TrainingItem', on_delete=models.RESTRICT)
+    level = models.ForeignKey('TrainingLevel', related_name='requirements', on_delete=models.CASCADE)
+    item = models.ForeignKey('TrainingItem', on_delete=models.CASCADE)
     depth = models.IntegerField(choices=TrainingItemQualification.CHOICES)
 
     reversion_hide = True
@@ -247,10 +247,10 @@ class TrainingLevelRequirement(models.Model, RevisionMixin):
 
 @reversion.register(follow=['trainee'])
 class TrainingLevelQualification(models.Model, RevisionMixin):
-    trainee = models.ForeignKey('Trainee', related_name='level_qualifications', on_delete=models.RESTRICT)
-    level = models.ForeignKey('TrainingLevel', on_delete=models.RESTRICT)
+    trainee = models.ForeignKey('Trainee', related_name='level_qualifications', on_delete=models.CASCADE)
+    level = models.ForeignKey('TrainingLevel', on_delete=models.CASCADE)
     confirmed_on = models.DateTimeField(null=True)
-    confirmed_by = models.ForeignKey('Trainee', related_name='confirmer', on_delete=models.RESTRICT, null=True)
+    confirmed_by = models.ForeignKey('Trainee', related_name='confirmer', on_delete=models.CASCADE, null=True)
 
     reversion_hide = True
 

users/management/commands/generateSampleUserData.py

@@ -85,39 +85,32 @@ class Command(BaseCommand):
             self.profiles.append(new_profile)
 
     def setup_useful_profiles(self):
-        super_user = models.Profile.objects.create(username="superuser", first_name="Super", last_name="User",
-                                                   initials="SU",
-                                                   email="superuser@example.com", is_superuser=True, is_active=True,
-                                                   is_staff=True)
-        super_user.set_password('superuser')
+        super_user = models.Profile.objects.create_superuser(username="superuser",
+                                                   email="superuser@example.com", password="superuser", first_name="Super", last_name="User",
+                                                   initials="SU", is_active=True)
         super_user.save()
 
-        finance_user = models.Profile.objects.create(username="finance", first_name="Finance", last_name="User",
-                                                     initials="FU",
-                                                     email="financeuser@example.com", is_active=True, is_approved=True)
+        finance_user = models.Profile.objects.create_user(username="finance",
+                                                     email="financeuser@example.com", password="finance", first_name="Finance", last_name="User",
+                                                     initials="FU", is_active=True, is_approved=True)
         finance_user.groups.add(self.finance_group)
         finance_user.groups.add(self.keyholder_group)
-        finance_user.set_password('finance')
         finance_user.save()
 
-        hs_user = models.Profile.objects.create(username="hs", first_name="HS", last_name="User",
-                                                initials="HSU",
-                                                email="hsuser@example.com", is_active=True, is_approved=True)
+        hs_user = models.Profile.objects.create_user(username="hs",
+                                                email="hsuser@example.com", password="hs", first_name="HS", last_name="User",
+                                                initials="HSU", is_active=True, is_approved=True)
         hs_user.groups.add(self.hs_group)
         hs_user.groups.add(self.keyholder_group)
-        hs_user.set_password('hs')
         hs_user.save()
 
-        keyholder_user = models.Profile.objects.create(username="keyholder", first_name="Keyholder", last_name="User",
-                                                       initials="KU",
-                                                       email="keyholderuser@example.com", is_active=True,
+        keyholder_user = models.Profile.objects.create_user(username="keyholder",
+                                                       email="keyholderuser@example.com", password="keyholder", first_name="Keyholder", last_name="User",
+                                                       initials="KU", is_active=True,
                                                        is_approved=True)
         keyholder_user.groups.add(self.keyholder_group)
-        keyholder_user.set_password('keyholder')
         keyholder_user.save()
 
-        basic_user = models.Profile.objects.create(username="basic", first_name="Basic", last_name="User",
-                                                   initials="BU",
-                                                   email="basicuser@example.com", is_active=True, is_approved=True)
-        basic_user.set_password('basic')
-        basic_user.save()
+        basic_user = models.Profile.objects.create_user(username="basic",
+                                                   email="basicuser@example.com", password="basic", first_name="Basic", last_name="User",
+                                                   initials="BU", is_active=True, is_approved=True)