Differential D3257 Diff 11557 swh/scheduler/tests/test_api_client.py

Changeset 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):
	server.scheduler = swh_db_scheduler			assert hasattr(server, "_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