Added symptoms to SuspectedCase aggregate.

costello/domainmodel/suspected_case.py

@@ -9,6 +9,7 @@ class SuspectedCase(BaseAggregateRoot):
 
     def __init__(self, *, person_id, **kwargs: Any):
         super().__init__(**kwargs)
+        self.symptoms = []
         self.person_id = person_id
         self.test_results = []
         self.is_closed = False
@@ -50,3 +51,21 @@ def mutate(self, obj: "SuspectedCase") -> None:
     @property
     def has_been_tested(self):
         return bool(len(self.test_results))
+
+    def record_symptoms(self, symptoms: tuple, date: date):
+        self.__trigger_event__(SuspectedCase.SymptomsRecorded, symptoms=symptoms, date=date)
+
+    class SymptomsRecorded(BaseAggregateRoot.Event):
+        @property
+        def symptoms(self):
+            return self.__dict__['symptoms']
+
+        @property
+        def date(self):
+            return self.__dict__['date']
+
+        def mutate(self, obj: "SuspectedCase"):
+            obj.symptoms.append({
+                "symptoms": self.symptoms,
+                "date": self.date
+            })

test_suspected_case.py

@@ -33,3 +33,11 @@ def test_record_test_result(self):
         self.assertEqual(case.test_results[1]["type"], "PCR")
         self.assertEqual(case.test_results[1]["date"], date(2020, 4, 8))
         self.assertEqual(case.test_results[1]["result"], "negative")
+
+    def test_record_symptoms(self):
+        case = SuspectedCase.open(person_id=uuid4())
+
+        # Has cough.
+        case.record_symptoms(symptoms=["cough", "fever"], date=date(2020, 4, 1))
+        self.assertEqual(len(case.symptoms), 1)
+