feat: batch create schedule (#1714)

* chore: pytestify schedule

* chore: remove unused imports

* feat: batch update schedule state

---------

Co-authored-by: Jordan Woods <13803242+jorwoods@users.noreply.github.com>

Author: jorwoods

test/test_schedule.py

@@ -26,7 +26,7 @@
 ADD_DATASOURCE_TO_SCHEDULE = TEST_ASSET_DIR / "schedule_add_datasource.xml"
 ADD_FLOW_TO_SCHEDULE = TEST_ASSET_DIR / "schedule_add_flow.xml"
 GET_EXTRACT_TASKS_XML = TEST_ASSET_DIR / "schedule_get_extract_refresh_tasks.xml"
-BATCH_UPDATE_STATE = TEST_ASSET_DIR / "schedules_batch_update_state.xml"
+BATCH_UPDATE_STATE = TEST_ASSET_DIR / "schedule_batch_update_state.xml"
 
 WORKBOOK_GET_BY_ID_XML = TEST_ASSET_DIR / "workbook_get_by_id.xml"
 DATASOURCE_GET_BY_ID_XML = TEST_ASSET_DIR / "datasource_get_by_id.xml"
@@ -421,3 +421,58 @@ def test_get_extract_refresh_tasks(server: TSC.Server) -> None:
         assert isinstance(extracts[0], list)
         assert 2 == len(extracts[0])
         assert "task1" == extracts[0][0].id
+
+
+def test_batch_update_state_items(server: TSC.Server) -> None:
+    server.version = "3.27"
+    hourly_interval = TSC.HourlyInterval(start_time=time(2, 30), end_time=time(23, 0), interval_value=2)
+    args = ("hourly", 50, TSC.ScheduleItem.Type.Extract, TSC.ScheduleItem.ExecutionOrder.Parallel, hourly_interval)
+    new_schedules = [TSC.ScheduleItem(*args), TSC.ScheduleItem(*args), TSC.ScheduleItem(*args)]
+    new_schedules[0]._id = "593d2ebf-0d18-4deb-9d21-b113a4902583"
+    new_schedules[1]._id = "cecbb71e-def0-4030-8068-5ae50f51db1c"
+    new_schedules[2]._id = "f39a6e7d-405e-4c07-8c18-95845f9da80e"
+
+    state = "active"
+    with requests_mock.mock() as m:
+        m.put(f"{server.schedules.baseurl}?state={state}", text=BATCH_UPDATE_STATE.read_text())
+        resp = server.schedules.batch_update_state(new_schedules, state)
+
+    assert len(resp) == 3
+    for sch, r in zip(new_schedules, resp):
+        assert sch.id == r
+
+
+def test_batch_update_state_str(server: TSC.Server) -> None:
+    server.version = "3.27"
+    new_schedules = [
+        "593d2ebf-0d18-4deb-9d21-b113a4902583",
+        "cecbb71e-def0-4030-8068-5ae50f51db1c",
+        "f39a6e7d-405e-4c07-8c18-95845f9da80e",
+    ]
+
+    state = "suspended"
+    with requests_mock.mock() as m:
+        m.put(f"{server.schedules.baseurl}?state={state}", text=BATCH_UPDATE_STATE.read_text())
+        resp = server.schedules.batch_update_state(new_schedules, state)
+
+    assert len(resp) == 3
+    for sch, r in zip(new_schedules, resp):
+        assert sch == r
+
+
+def test_batch_update_state_all(server: TSC.Server) -> None:
+    server.version = "3.27"
+    new_schedules = [
+        "593d2ebf-0d18-4deb-9d21-b113a4902583",
+        "cecbb71e-def0-4030-8068-5ae50f51db1c",
+        "f39a6e7d-405e-4c07-8c18-95845f9da80e",
+    ]
+
+    state = "suspended"
+    with requests_mock.mock() as m:
+        m.put(f"{server.schedules.baseurl}?state={state}&updateAll=true", text=BATCH_UPDATE_STATE.read_text())
+        _ = server.schedules.batch_update_state(new_schedules, state, True)
+
+        history = m.request_history[0]
+
+    assert history.text == "<tsRequest />"