E2e test for num_cases in POST #2714

globaldothealth · Jun 23, 2022 · 5c18a5e · 5c18a5e
1 parent 01d443a
commit 5c18a5e
Show file tree

Hide file tree

Showing 2 changed files with 15 additions and 1 deletion.
diff --git a/data-serving/reusable-data-service/reusable_data_service/main.py b/data-serving/reusable-data-service/reusable_data_service/main.py
@@ -25,7 +25,10 @@ def list_cases():
         filter = request.args.get("q", type=str)
         return case_controller.list_cases(page=page, limit=limit, filter=filter)
     else:
-        return case_controller.create_case(request.get_json())
+        count = request.args.get("num_cases", type=int)
+        if count is None:
+            count = 1
+        return case_controller.create_case(request.get_json(), num_cases=count)
 
 def set_up_controllers():
     global case_controller

diff --git a/data-serving/reusable-data-service/tests/test_case_end_to_end.py b/data-serving/reusable-data-service/tests/test_case_end_to_end.py
@@ -163,3 +163,14 @@ def test_post_case_list_cases_round_trip(client_with_patched_mongo):
     assert get_response.status_code == 200
     assert len(get_response.json["cases"]) == 1
     assert get_response.json["cases"][0]["confirmation_date"] == "2022-01-23"
+
+
+def test_post_multiple_case_list_cases_round_trip(client_with_patched_mongo):
+    post_response = client_with_patched_mongo.post("/api/cases?num_cases=3", json = {
+        "confirmation_date": "2022-01-23T13:45:01.234Z"
+    })
+    assert post_response.status_code == 201
+    get_response = client_with_patched_mongo.get("/api/cases")
+    assert get_response.status_code == 200
+    assert len(get_response.json["cases"]) == 3
+    assert get_response.json["cases"][0]["confirmation_date"] == "2022-01-23"