Changeset View
Changeset View
Standalone View
Standalone View
swh/scheduler/tests/test_api_client.py
Show All 11 Lines | |||||
# tests are executed using imported class (TestScheduler) using overloaded | # tests are executed using imported class (TestScheduler) using overloaded | ||||
# swh_scheduler fixture below | # swh_scheduler fixture below | ||||
# the Flask app used as server in these tests | # the Flask app used as server in these tests | ||||
@pytest.fixture | @pytest.fixture | ||||
def app(swh_db_scheduler): | def app(swh_db_scheduler): | ||||
assert hasattr(server, "scheduler") | |||||
server.scheduler = swh_db_scheduler | server.scheduler = swh_db_scheduler | ||||
yield server.app | yield server.app | ||||
# the RPCClient class used as client used in these tests | # the RPCClient class used as client used in these tests | ||||
@pytest.fixture | @pytest.fixture | ||||
def swh_rpc_client_class(): | def swh_rpc_client_class(): | ||||
return RemoteScheduler | return RemoteScheduler | ||||
@pytest.fixture | @pytest.fixture | ||||
def swh_scheduler(swh_rpc_client, app): | def swh_scheduler(swh_rpc_client, app): | ||||
yield swh_rpc_client | yield swh_rpc_client | ||||
def test_site_map(flask_app_client): | def test_site_map(flask_app_client): | ||||
sitemap = flask_app_client.get(url_for("site_map")) | sitemap = flask_app_client.get(url_for("site_map")) | ||||
assert sitemap.headers["Content-Type"] == "application/json" | assert sitemap.headers["Content-Type"] == "application/json" | ||||
rules = set(x["rule"] for x in sitemap.json) | rules = set(x["rule"] for x in sitemap.json) | ||||
# we expect at least these rules | # we expect at least these rules | ||||
expected_rules = set( | expected_rules = set( | ||||
"/" + rule | "/" + rule | ||||
for rule in ( | for rule in ( | ||||
"set_status_tasks", | "priority_ratios/get", | ||||
"create_task_type", | "task/create", | ||||
"get_task_type", | "task/delete_archived", | ||||
"get_task_types", | "task/disable", | ||||
"create_tasks", | "task/filter_for_archive", | ||||
"disable_tasks", | "task/get", | ||||
"get_tasks", | "task/grab_ready", | ||||
"search_tasks", | "task/peek_ready", | ||||
"get_task_runs", | "task/search", | ||||
"peek_ready_tasks", | "task/set_status", | ||||
"grab_ready_tasks", | "task_run/end", | ||||
"schedule_task_run", | "task_run/get", | ||||
"mass_schedule_task_runs", | "task_run/schedule", | ||||
"start_task_run", | "task_run/schedule_one", | ||||
"end_task_run", | "task_run/start", | ||||
"filter_task_to_archive", | "task_type/create", | ||||
"delete_archived_tasks", | "task_type/get", | ||||
"get_priority_ratios", | "task_type/get_all", | ||||
) | ) | ||||
) | ) | ||||
assert rules == expected_rules | assert rules == expected_rules |