populate script now assocatiates a single EngagementEvent with a CAF

2 files changed, 6 insertions, 10 deletions

diff --git a/ctrack/organisations/management/commands/populate_db.py b/ctrack/organisations/management/commands/populate_db.py
index 78efc1d..8a2ad84 100644
--- a/ctrack/organisations/management/commands/populate_db.py
+++ b/ctrack/organisations/management/commands/populate_db.py
@@ -139,6 +139,7 @@ class Command(BaseCommand):
 
         etf1 = EngagementTypeFactory(descriptor="Information Notice")
         etf2 = EngagementTypeFactory(descriptor="Designation Letter")
+        etf3 = EngagementTypeFactory(descriptor="CAF Submission")
 
         ee1 = EngagementEventFactory.create(type=etf1, user=user, participants=[p1, p2])
         ee2 = EngagementEventFactory.create(type=etf2, user=user, participants=[p3])
@@ -175,6 +176,11 @@ class Command(BaseCommand):
             for _ in range(35)
         ]
 
+        # CAF submissions - they create EngagementEvents
+        ee3 = EngagementEventFactory.create(
+            type=etf3, user=user, participants=[inspectors[1], p2], related_caf=cafs[1]
+        )
+
         # TODO - adapt this so that it records more than just Persons created
         self.stdout.write(
             self.style.SUCCESS(
diff --git a/ctrack/register/models.py b/ctrack/register/models.py
index 5f585c3..60733c5 100644
--- a/ctrack/register/models.py
+++ b/ctrack/register/models.py
@@ -37,16 +37,6 @@ class EngagementEvent(models.Model):
     short_description = models.CharField(
         max_length=50, help_text="Short description of the event"
     )
-    # TODO - Do we need an "inspector_engagement" field here?
-    #    We want an EngagementEvent to cover interactions between
-    #    Persons (industry people) and inspectors (Users). We don't want
-    #    to have to create Person() objects for each inspector, do we?
-    #    Actually, if we do, we could allow for the case where a ctrack user
-    #    is able to do delegated work for inspectors. So the task here is to
-    #    adapt the factories so half a dozen inspectors are created as Persons
-    #    and given the role of DfT Compliance Inspector, which have to be created
-    #    also. Then we can start building out the CAF factories to include stuff
-    #    list submission date, submission recipient, etc.
     participants = models.ManyToManyField(Person)
     user = models.ForeignKey(get_user_model(), on_delete=models.SET(get_sentinel_user))
     date = models.DateTimeField()