|
15 | 15 | GET_DAILY_ID_XML = TEST_ASSET_DIR / "schedule_get_daily_id.xml" |
16 | 16 | GET_MONTHLY_ID_XML = TEST_ASSET_DIR / "schedule_get_monthly_id.xml" |
17 | 17 | GET_MONTHLY_ID_2_XML = TEST_ASSET_DIR / "schedule_get_monthly_id_2.xml" |
18 | | -GET_CUSTOMIZED_MONTHLY_ID_XML = TEST_ASSET_DIR / "schedule_get_customized_monthly_id.xml" |
19 | 18 | GET_EMPTY_XML = TEST_ASSET_DIR / "schedule_get_empty.xml" |
20 | 19 | CREATE_HOURLY_XML = TEST_ASSET_DIR / "schedule_create_hourly.xml" |
21 | 20 | CREATE_DAILY_XML = TEST_ASSET_DIR / "schedule_create_daily.xml" |
|
27 | 26 | ADD_DATASOURCE_TO_SCHEDULE = TEST_ASSET_DIR / "schedule_add_datasource.xml" |
28 | 27 | ADD_FLOW_TO_SCHEDULE = TEST_ASSET_DIR / "schedule_add_flow.xml" |
29 | 28 | GET_EXTRACT_TASKS_XML = TEST_ASSET_DIR / "schedule_get_extract_refresh_tasks.xml" |
30 | | -BATCH_UPDATE_STATE = TEST_ASSET_DIR / "schedule_batch_update_state.xml" |
| 29 | +BATCH_UPDATE_STATE = TEST_ASSET_DIR / "schedules_batch_update_state.xml" |
31 | 30 |
|
32 | 31 | WORKBOOK_GET_BY_ID_XML = TEST_ASSET_DIR / "workbook_get_by_id.xml" |
33 | 32 | DATASOURCE_GET_BY_ID_XML = TEST_ASSET_DIR / "datasource_get_by_id.xml" |
@@ -179,21 +178,6 @@ def test_get_monthly_by_id_2(server: TSC.Server) -> None: |
179 | 178 | assert ("Monday", "First") == schedule.interval_item.interval |
180 | 179 |
|
181 | 180 |
|
182 | | -def test_get_customized_monthly_by_id(server: TSC.Server) -> None: |
183 | | - server.version = "3.15" |
184 | | - response_xml = GET_CUSTOMIZED_MONTHLY_ID_XML.read_text() |
185 | | - with requests_mock.mock() as m: |
186 | | - schedule_id = "f048d794-90dc-40b0-bfad-2ca78e437369" |
187 | | - baseurl = f"{server.baseurl}/schedules/{schedule_id}" |
188 | | - m.get(baseurl, text=response_xml) |
189 | | - schedule = server.schedules.get_by_id(schedule_id) |
190 | | - assert schedule is not None |
191 | | - assert schedule_id == schedule.id |
192 | | - assert "Monthly customized" == schedule.name |
193 | | - assert "Active" == schedule.state |
194 | | - assert ("Customized Monthly",) == schedule.interval_item.interval |
195 | | - |
196 | | - |
197 | 181 | def test_delete(server: TSC.Server) -> None: |
198 | 182 | with requests_mock.mock() as m: |
199 | 183 | m.delete(server.schedules.baseurl + "/c9cff7f9-309c-4361-99ff-d4ba8c9f5467", status_code=204) |
@@ -437,58 +421,3 @@ def test_get_extract_refresh_tasks(server: TSC.Server) -> None: |
437 | 421 | assert isinstance(extracts[0], list) |
438 | 422 | assert 2 == len(extracts[0]) |
439 | 423 | assert "task1" == extracts[0][0].id |
440 | | - |
441 | | - |
442 | | -def test_batch_update_state_items(server: TSC.Server) -> None: |
443 | | - server.version = "3.27" |
444 | | - hourly_interval = TSC.HourlyInterval(start_time=time(2, 30), end_time=time(23, 0), interval_value=2) |
445 | | - args = ("hourly", 50, TSC.ScheduleItem.Type.Extract, TSC.ScheduleItem.ExecutionOrder.Parallel, hourly_interval) |
446 | | - new_schedules = [TSC.ScheduleItem(*args), TSC.ScheduleItem(*args), TSC.ScheduleItem(*args)] |
447 | | - new_schedules[0]._id = "593d2ebf-0d18-4deb-9d21-b113a4902583" |
448 | | - new_schedules[1]._id = "cecbb71e-def0-4030-8068-5ae50f51db1c" |
449 | | - new_schedules[2]._id = "f39a6e7d-405e-4c07-8c18-95845f9da80e" |
450 | | - |
451 | | - state = "active" |
452 | | - with requests_mock.mock() as m: |
453 | | - m.put(f"{server.schedules.baseurl}?state={state}", text=BATCH_UPDATE_STATE.read_text()) |
454 | | - resp = server.schedules.batch_update_state(new_schedules, state) |
455 | | - |
456 | | - assert len(resp) == 3 |
457 | | - for sch, r in zip(new_schedules, resp): |
458 | | - assert sch.id == r |
459 | | - |
460 | | - |
461 | | -def test_batch_update_state_str(server: TSC.Server) -> None: |
462 | | - server.version = "3.27" |
463 | | - new_schedules = [ |
464 | | - "593d2ebf-0d18-4deb-9d21-b113a4902583", |
465 | | - "cecbb71e-def0-4030-8068-5ae50f51db1c", |
466 | | - "f39a6e7d-405e-4c07-8c18-95845f9da80e", |
467 | | - ] |
468 | | - |
469 | | - state = "suspended" |
470 | | - with requests_mock.mock() as m: |
471 | | - m.put(f"{server.schedules.baseurl}?state={state}", text=BATCH_UPDATE_STATE.read_text()) |
472 | | - resp = server.schedules.batch_update_state(new_schedules, state) |
473 | | - |
474 | | - assert len(resp) == 3 |
475 | | - for sch, r in zip(new_schedules, resp): |
476 | | - assert sch == r |
477 | | - |
478 | | - |
479 | | -def test_batch_update_state_all(server: TSC.Server) -> None: |
480 | | - server.version = "3.27" |
481 | | - new_schedules = [ |
482 | | - "593d2ebf-0d18-4deb-9d21-b113a4902583", |
483 | | - "cecbb71e-def0-4030-8068-5ae50f51db1c", |
484 | | - "f39a6e7d-405e-4c07-8c18-95845f9da80e", |
485 | | - ] |
486 | | - |
487 | | - state = "suspended" |
488 | | - with requests_mock.mock() as m: |
489 | | - m.put(f"{server.schedules.baseurl}?state={state}&updateAll=true", text=BATCH_UPDATE_STATE.read_text()) |
490 | | - _ = server.schedules.batch_update_state(new_schedules, state, True) |
491 | | - |
492 | | - history = m.request_history[0] |
493 | | - |
494 | | - assert history.text == "<tsRequest />" |
0 commit comments