the more explicitly concrete the unit tests the better

openhealthcare · Jan 4, 2018 · 829c6e4 · 829c6e4
1 parent b672d7c
commit 829c6e4
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/opal/tests/test_models.py b/opal/tests/test_models.py
@@ -232,9 +232,10 @@ def test_bulk_update_create_new_episode_for_preexisting_patient_if_not_passed_ex
         original_patient.create_episode()
         self.assertEqual(1, original_patient.episode_set.count())
         original_patient.bulk_update(d, self.user)
-
-        patient = Patient.objects.get()
         self.assertEqual(2, original_patient.episode_set.count())
+        self.assertEqual(models.Episode.objects.count(), 2)
+        location = original_patient.episode_set.last().location_set.first()
+        self.assertEqual(location.ward, "a ward")
 
 
 class SubrecordTestCase(OpalTestCase):