Merge branch 'load-tests-in-from-json' into v0.3.5.rc1

intrahospital_api/management/commands/load_tests_from_json.py

@@ -4,7 +4,7 @@
 """
 import json
 from django.core.management.base import BaseCommand
-from django.contrib.auth.models import User
+from django.db import transaction
 from opal import models
 from elcid import models as emodels
 from lab import models as lmodels
@@ -16,12 +16,8 @@ def add_arguments(self, parser):
         parser.add_argument('patient_id', type=int)
         parser.add_argument('file_name', type=str)
 
-    def handle(self, patient_id, file_name, *args, **options):
-        # we assume that there is a user called super
-        # which there won't be on prod, but we shouldn't
-        # be running this on prod
-        user = User.objects.get(username="super")
-        patient = models.Patient.objects.get(id=patient_id)
+    @transaction.atomic
+    def process(self, patient, results):
         lmodels.LabTest.objects.filter(
             patient=patient
         ).filter(
@@ -30,7 +26,14 @@ def handle(self, patient_id, file_name, *args, **options):
                 emodels.UpstreamBloodCulture.get_display_name(),
             ]
         ).delete()
+        update_lab_tests.update_tests(patient, results)
+
+    def handle(self, patient_id, file_name, *args, **options):
+        # we assume that there is a user called super
+        # which there won't be on prod, but we shouldn't
+        # be running this on prod
         with open(file_name) as f:
             results = json.load(f)
-            update_lab_tests.update_tests(patient, results, user)
+        patient = models.Patient.objects.get(id=patient_id)
+        self.process(patient, results)
         print "success"

intrahospital_api/test/test_load_tests_from_json.py

@@ -0,0 +1,84 @@
+import json
+import mock
+import datetime
+from django.test import override_settings
+from django.utils import timezone
+from django.contrib.auth.models import User
+from opal.core.test import OpalTestCase
+from intrahospital_api.management.commands import load_tests_from_json
+from elcid import models as emodels
+
+example_json = """
+[{
+"status": "complete",
+"external_identifier": "123423137",
+"site": "^& ^",
+"test_code": "b12_and_folate_screen",
+"test_name": "B12 AND FOLATE SCREEN",
+"clinical_info": "testing",
+"external_system": "RFH Database",
+"datetime_ordered": "19/07/2018 15:04:22",
+"observations": [
+{
+"units": "ng/L",
+"observation_datetime": "20/07/2018 12:01:23",
+"observation_name": "Vitamin B12",
+"observation_number": "12015325",
+"last_updated": "19/07/2018 14:44:22",
+"observation_value": "89.7",
+"reference_range": "160 - 925"
+},
+{
+"units": "ng/L",
+"observation_datetime": "20/07/2018 12:01:23",
+"observation_name": "Folate",
+"observation_number": "12015326",
+"last_updated": "20/07/2018 11:42:12",
+"observation_value": "-2.7",
+"reference_range": "3.9 - 26.8"
+}]
+}]
+""".strip()
+
+
+@mock.patch(
+    "intrahospital_api.management.commands.load_tests_from_json.json.load"
+)
+@override_settings(API_USER="api_user")
+class LoadTestsFromJson(OpalTestCase):
+    def setUp(self):
+        # initialise the test user
+        User.objects.create(username="api_user")
+
+    def test_loads_in_tests(self, json_load):
+        command = load_tests_from_json.Command()
+        test_results = json.loads(example_json)
+        patient, _ = self.new_patient_and_episode_please()
+        command.process(patient, test_results)
+        self.assertEqual(
+            patient.labtest_set.count(), 1
+        )
+        self.assertEqual(
+            patient.labtest_set.first().extras["test_code"],
+            "b12_and_folate_screen"
+        )
+        expected_ordered = datetime.datetime(2018, 7, 19, 15, 4, 22)
+        self.assertEqual(
+            timezone.make_naive(patient.labtest_set.first().datetime_ordered),
+            expected_ordered
+        )
+
+    def test_deletes_existing_tests(self, json_load):
+        command = load_tests_from_json.Command()
+        patient, _ = self.new_patient_and_episode_please()
+        patient.labtest_set.create(
+            lab_test_type=emodels.UpstreamLabTest.get_display_name()
+        )
+        patient.labtest_set.create(
+            lab_test_type=emodels.GramStain.get_display_name()
+        )
+        command.process(patient, [])
+        self.assertEqual(
+            patient.labtest_set.get().lab_test_type,
+            emodels.GramStain.get_display_name()
+        )