Differential D2881 Diff 10999 swh/web/client/tests/conftest.py

Changeset View

Standalone View

swh/web/client/tests/conftest.py

	# Copyright (C) 2020 The Software Heritage developers			# Copyright (C) 2020 The Software Heritage developers
	# See the AUTHORS file at the top-level directory of this distribution			# See the AUTHORS file at the top-level directory of this distribution
	# License: GNU General Public License version 3, or any later version			# License: GNU General Public License version 3, or any later version
	# See top-level LICENSE file for more information			# See top-level LICENSE file for more information

	import pytest			import pytest

	from .api_data import API_URL, API_DATA			from .api_data import API_URL, API_DATA
	from swh.web.client import WebAPIClient			from swh.web.client import WebAPIClient


	@pytest.fixture			@pytest.fixture
	def web_api_mock(requests_mock):			def web_api_mock(requests_mock):
				# monkey patch URLs that require a special response headers
	for api_call, data in API_DATA.items():			for api_call, data in API_DATA.items():
	headers = {}			headers = {}
	if api_call == "snapshot/cabcc7d7bf639bbe1cc3b41989e1806618dd5764/":			if api_call == "snapshot/cabcc7d7bf639bbe1cc3b41989e1806618dd5764/":
	# monkey patch the only URL that require a special response headers			# to make the client init and follow pagination
	# (to make the client init and follow pagination)
	headers = {			headers = {
	"Link": f'<{API_URL}/{api_call}?branches_count=1000&branches_from=refs/tags/v3.0-rc7>; rel="next"' # NoQA: E501			"Link": f'<{API_URL}/{api_call}?branches_count=1000&branches_from=refs/tags/v3.0-rc7>; rel="next"' # NoQA: E501
	}			}
				elif api_call == "origin/https://github.com/NixOS/nixpkgs/visits/?last_visit=50&per_page=10": # NoQA: E501
				# to make the client follow pagination
				zackUnsubmitted Not Done Inline Actions this is no longer "the only URL" :-) can you fix this comment and the one you copied this from, to simply state why each URL needs monkey patching? zack: this is no longer "the only URL" :-) can you fix this comment and the one you copied this from…
				headers = {
				"Link":
				f"<{API_URL}/origin/https://github.com/NixOS/nixpkgs/visits/?last_visit=40&per_page=10>; rel=\"next\"" # NoQA: E501
				}
	requests_mock.get(f"{API_URL}/{api_call}", text=data, headers=headers)			requests_mock.get(f"{API_URL}/{api_call}", text=data, headers=headers)
	return requests_mock			return requests_mock


	@pytest.fixture			@pytest.fixture
	def web_api_client():			def web_api_client():
	# use the fake base API URL that matches API data			# use the fake base API URL that matches API data
	return WebAPIClient(api_url=API_URL)			return WebAPIClient(api_url=API_URL)