Add difficulty_breathing field to covid19 triage

eventstore/migrations/0025_add_difficulty_breathing.py

@@ -0,0 +1,34 @@
+# Generated by Django 2.2.8 on 2020-04-07 14:55
+
+from django.db import migrations, models
+
+
+class Migration(migrations.Migration):
+
+    dependencies = [("eventstore", "0024_auto_20200407_1242")]
+
+    operations = [
+        migrations.AddField(
+            model_name="covid19triage",
+            name="difficulty_breathing",
+            field=models.BooleanField(blank=True, default=None, null=True),
+        ),
+        migrations.AlterField(
+            model_name="covid19triage",
+            name="province",
+            field=models.CharField(
+                choices=[
+                    ("ZA-EC", "Eastern Cape"),
+                    ("ZA-FS", "Free State"),
+                    ("ZA-GT", "Gauteng"),
+                    ("ZA-LP", "Limpopo"),
+                    ("ZA-MP", "Mpumalanga"),
+                    ("ZA-NC", "Northern Cape"),
+                    ("ZA-NL", "Kwazulu-Natal"),
+                    ("ZA-NW", "North-West (South Africa)"),
+                    ("ZA-WC", "Western Cape"),
+                ],
+                max_length=6,
+            ),
+        ),
+    ]

eventstore/models.py

@@ -388,7 +388,7 @@ class Covid19Triage(models.Model):
         (AGE_O65, AGE_O65),
     )
 
-    PROVINCE_CHOICES = tuple(
+    PROVINCE_CHOICES = sorted(
         (s.code, s.name) for s in pycountry.subdivisions.get(country_code="ZA")
     )
 
@@ -421,6 +421,7 @@ class Covid19Triage(models.Model):
     fever = models.BooleanField()
     cough = models.BooleanField()
     sore_throat = models.BooleanField()
+    difficulty_breathing = models.BooleanField(null=True, blank=True, default=None)
     exposure = models.CharField(max_length=9, choices=EXPOSURE_CHOICES)
     tracing = models.BooleanField(help_text="Whether the NDoH can contact the user")
     risk = models.CharField(max_length=8, choices=RISK_CHOICES)

eventstore/test_views.py

@@ -1517,6 +1517,7 @@ def test_successful_request(self):
                 "tracing": True,
                 "risk": Covid19Triage.RISK_LOW,
             },
+            format="json",
         )
         self.assertEqual(response.status_code, status.HTTP_201_CREATED)
         [covid19triage] = Covid19Triage.objects.all()
@@ -1528,6 +1529,7 @@ def test_successful_request(self):
         self.assertEqual(covid19triage.fever, False)
         self.assertEqual(covid19triage.cough, False)
         self.assertEqual(covid19triage.sore_throat, False)
+        self.assertEqual(covid19triage.difficulty_breathing, None)
         self.assertEqual(covid19triage.exposure, Covid19Triage.EXPOSURE_NO)
         self.assertEqual(covid19triage.tracing, True)
         self.assertEqual(covid19triage.risk, Covid19Triage.RISK_LOW)