"""Tests for POST /report-runs/dataloader endpoint.""" import json from oto import status from tests.testutils import db, mock_auth @db.test_schema_default_seed def test_post_report_runs_dataloader(auth_client, mocker): """Expect to bulk fetch report runs.""" mock_auth(mocker, [24601, 120594]) request_data = { "report_runs": [ # Report run exists {"report_run_id": 1}, # Report run doesn't exist {"report_run_id": 10}, ] } result = auth_client.post( "/report-runs/dataloader", data=json.dumps(request_data), content_type="application/json", ) result_data = json.loads(result.data.decode()) assert result.status_code == status.OK assert result_data == [ { "data": { "report_run_uuid": "abcd-1234", "report_run_name": "Test Run", "period_ids": "1,2,3", "period_name": "Sometime", "requested_datetime": "2019-09-08T07:06:05", "trigger_type": "AUTO", } }, {"data": None}, ]