Add openhim queue model and signal

eventstore/apps.py

@@ -3,3 +3,6 @@
 
 class EventstoreConfig(AppConfig):
     name = "eventstore"
+
+    def ready(self):
+        import eventstore.signals  # noqa

eventstore/migrations/0062_openhimqueue.py

@@ -0,0 +1,54 @@
+# Generated by Django 4.1.7 on 2023-11-23 09:40
+
+import uuid
+
+from django.db import migrations, models
+
+
+class Migration(migrations.Migration):
+    dependencies = [
+        ("eventstore", "0061_alter_optout_reason"),
+    ]
+
+    operations = [
+        migrations.CreateModel(
+            name="OpenHIMQueue",
+            fields=[
+                (
+                    "id",
+                    models.UUIDField(
+                        default=uuid.uuid4,
+                        editable=False,
+                        primary_key=True,
+                        serialize=False,
+                    ),
+                ),
+                ("object_id", models.UUIDField()),
+                (
+                    "object_type",
+                    models.CharField(
+                        choices=[
+                            ("PrebirthRegistration", "PrebirthRegistration"),
+                            ("CHWRegistration", "CHWRegistration"),
+                            ("PublicRegistration", "PublicRegistration"),
+                            ("ChannelSwitch", "ChannelSwitch"),
+                            ("OptOut", "OptOut"),
+                        ],
+                        max_length=30,
+                    ),
+                ),
+                (
+                    "status",
+                    models.PositiveSmallIntegerField(
+                        choices=[
+                            (0, "Pending"),
+                            (1, "Processing"),
+                            (2, "Complete"),
+                            (3, "Error"),
+                        ],
+                        default=0,
+                    ),
+                ),
+            ],
+        ),
+    ]

eventstore/migrations/0063_alter_openhimqueue_object_type.py

@@ -0,0 +1,26 @@
+# Generated by Django 4.1.7 on 2023-11-23 11:40
+
+from django.db import migrations, models
+
+
+class Migration(migrations.Migration):
+    dependencies = [
+        ("eventstore", "0062_openhimqueue"),
+    ]
+
+    operations = [
+        migrations.AlterField(
+            model_name="openhimqueue",
+            name="object_type",
+            field=models.CharField(
+                choices=[
+                    ("prebirth_registration", "PrebirthRegistration"),
+                    ("chw_registration", "CHWRegistration"),
+                    ("public_registration", "PublicRegistration"),
+                    ("channel_switch", "ChannelSwitch"),
+                    ("optout", "OptOut"),
+                ],
+                max_length=30,
+            ),
+        ),
+    ]

eventstore/models.py

@@ -1094,3 +1094,32 @@ def clean(self):
                 date(self.dob_year, self.dob_month, self.dob_day)
             except ValueError as e:
                 raise ValidationError(f"Invalid date of birth date, {str(e)}")
+
+
+class OpenHIMQueue(models.Model):
+    class ObjectType:
+        PREBIRTH_REGISTRATION = "prebirth_registration"
+        CHW_REGISTRATION = "chw_registration"
+        PUBLIC_REGISTRATION = "public_registration"
+        CHANNEL_SWITCH = "channel_switch"
+        OPTOUT = "optout"
+        choices = (
+            (PREBIRTH_REGISTRATION, "PrebirthRegistration"),
+            (CHW_REGISTRATION, "CHWRegistration"),
+            (PUBLIC_REGISTRATION, "PublicRegistration"),
+            (CHANNEL_SWITCH, "ChannelSwitch"),
+            (OPTOUT, "OptOut"),
+        )
+
+    class Status(models.IntegerChoices):
+        PENDING = 0
+        PROCESSING = 1
+        COMPLETE = 2
+        ERROR = 3
+
+    id = models.UUIDField(primary_key=True, default=uuid.uuid4, editable=False)
+    object_id = models.UUIDField()
+    object_type = models.CharField(max_length=30, choices=ObjectType.choices)
+    status = models.PositiveSmallIntegerField(
+        choices=Status.choices, default=Status.PENDING
+    )

eventstore/signals.py

@@ -0,0 +1,13 @@
+from django.db.models.signals import post_save
+from django.dispatch import receiver
+
+from eventstore.models import OpenHIMQueue, PrebirthRegistration
+
+
+@receiver(post_save, sender=PrebirthRegistration)
+def queue_prebirth_registration(sender, instance, created, **kwargs):
+    if created:
+        OpenHIMQueue.objects.create(
+            object_id=instance.id,
+            object_type=OpenHIMQueue.ObjectType.PREBIRTH_REGISTRATION,
+        )

eventstore/test_models.py → eventstore/tests/test_models.py

@@ -8,6 +8,8 @@
     HCSStudyBRandomization,
     HealthCheckUserProfile,
     Message,
+    OpenHIMQueue,
+    PrebirthRegistration,
 )
 
 
@@ -569,3 +571,19 @@ def test_get_random_study_b_arm(self, mock_get_study_totals_per_province):
 
         mock_get_study_totals_per_province.return_value = (2000, 10)
         self.assertIsNone(rand.get_random_study_b_arm())
+
+
+class PrebirthRegistrationTests(TestCase):
+    def test_create_signal(self):
+        prebirthregistration = PrebirthRegistration.objects.create(
+            contact_id="9e12d04c-af25-40b6-aa4f-57c72e8e3f91",
+            device_contact_id="9e12d04c-af25-40b6-aa4f-57c72e8e3f91",
+            edd="2020-12-01",
+        )
+
+        queue_record = OpenHIMQueue.objects.first()
+
+        self.assertEqual(queue_record.object_id, prebirthregistration.id)
+        self.assertEqual(
+            queue_record.object_type, OpenHIMQueue.ObjectType.PREBIRTH_REGISTRATION
+        )