Skip to content

Commit 301ea25

Browse files
jorwoodsjacalata
authored andcommitted
chore: pytestify schedule (#1697)
* fix: black ci errors * chore: pytestify schedule * chore: remove unused imports --------- Co-authored-by: Jordan Woods <13803242+jorwoods@users.noreply.github.com>
1 parent f29abc4 commit 301ea25

File tree

1 file changed

+1
-72
lines changed

1 file changed

+1
-72
lines changed

test/test_schedule.py

Lines changed: 1 addition & 72 deletions
Original file line numberDiff line numberDiff line change
@@ -15,7 +15,6 @@
1515
GET_DAILY_ID_XML = TEST_ASSET_DIR / "schedule_get_daily_id.xml"
1616
GET_MONTHLY_ID_XML = TEST_ASSET_DIR / "schedule_get_monthly_id.xml"
1717
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"
1918
GET_EMPTY_XML = TEST_ASSET_DIR / "schedule_get_empty.xml"
2019
CREATE_HOURLY_XML = TEST_ASSET_DIR / "schedule_create_hourly.xml"
2120
CREATE_DAILY_XML = TEST_ASSET_DIR / "schedule_create_daily.xml"
@@ -27,7 +26,7 @@
2726
ADD_DATASOURCE_TO_SCHEDULE = TEST_ASSET_DIR / "schedule_add_datasource.xml"
2827
ADD_FLOW_TO_SCHEDULE = TEST_ASSET_DIR / "schedule_add_flow.xml"
2928
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"
3130

3231
WORKBOOK_GET_BY_ID_XML = TEST_ASSET_DIR / "workbook_get_by_id.xml"
3332
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:
179178
assert ("Monday", "First") == schedule.interval_item.interval
180179

181180

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-
197181
def test_delete(server: TSC.Server) -> None:
198182
with requests_mock.mock() as m:
199183
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:
437421
assert isinstance(extracts[0], list)
438422
assert 2 == len(extracts[0])
439423
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

Comments
 (0)