Update derived location fields when source location imported, refs #650

vaccinate/api/test-data/importSourceLocations/002-new-location.json

@@ -1,6 +1,6 @@
 {
-  "source_uid": "vaccinespotter:7382088",
-  "source_name": "vaccinespotter",
+  "source_uid": "vaccinespotter_org:7382088",
+  "source_name": "vaccinespotter_org",
   "content_hash": "6cafddd7de8861754ee23c5e4a1f64e1",
   "name": "Rite Aid",
   "latitude": 33.4962,
@@ -9,7 +9,7 @@
     "action": "new"
   },
   "import_json": {
-    "id": "vaccinespotter:7382088",
+    "id": "vaccinespotter_org:7382088",
     "name": "Rite Aid",
     "address": {
       "street1": "32121 Camino Capistrano",
@@ -26,8 +26,13 @@
       "website": "https://www.riteaid.com/pharmacy/covid-qualifier"
     }],
     "availability": {
-      "appointments": false
+      "appointments": false,
+      "drop_in": true
     },
+    "inventory": [
+      {"vaccine": "pfizer_biontech"},
+      {"vaccine": "johnson_johnson_janssen", "supply_level": "in_stock"}
+    ],
     "parent_organization": {
       "id": "rite_aid",
       "name": "Rite Aid"
@@ -43,7 +48,7 @@
       }
     ],
     "source": {
-      "source": "vaccinespotter",
+      "source": "vaccinespotter_org",
       "id": "7382088",
       "fetched_from_uri": "https://www.vaccinespotter.org/api/v0/states/CA.json",
       "fetched_at": "2021-04-06T20:17:55.148+00:00",

vaccinate/api/test_import_source_locations.py

@@ -237,3 +237,25 @@ def test_import_source_location_does_not_overwrite_existing_match(
     concordances = set(second_location.concordances.all())
     source_concordanes = set(source_location.concordances.all())
     assert not source_concordanes.issubset(concordances)
+
+
+def test_import_source_location_updates_location_fields(client, api_key):
+    with (tests_dir / "002-new-location.json").open() as fixture_file:
+        fixture = orjson.loads(fixture_file.read())
+    import_run_id = client.post(
+        "/api/startImportRun", HTTP_AUTHORIZATION="Bearer {}".format(api_key)
+    ).json()["import_run_id"]
+    # Make the API request
+    client.post(
+        "/api/importSourceLocations?import_run_id={}".format(import_run_id),
+        fixture,
+        content_type="application/json",
+        HTTP_AUTHORIZATION="Bearer {}".format(api_key),
+    )
+    source_location = SourceLocation.objects.first()
+    location = source_location.matched_location
+    assert set(location.vaccines_offered) == {"Pfizer", "Johnson & Johnson"}
+    assert location.accepts_walkins
+    assert not location.accepts_appointments
+    assert location.vaccines_offered_provenance_source_location == source_location
+    assert location.appointments_walkins_provenance_source_location == source_location

vaccinate/api/views.py

@@ -332,6 +332,12 @@ def import_source_locations(request, on_request_logged):
             created.append(source_location.pk)
         else:
             updated.append(source_location.pk)
+
+        if source_location.matched_location:
+            source_location.matched_location.derive_availability_and_inventory(
+                save=True
+            )
+
     return JsonResponse({"created": created, "updated": updated})