Differential D5660 Diff 20220 swh/web/tests/api/views/test_origin_save.py

Changeset View

Standalone View

swh/web/tests/api/views/test_origin_save.py

# Copyright (C) 2018-2021 The Software Heritage developers		# Copyright (C) 2018-2021 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 Affero General Public License version 3, or any later version		# License: GNU Affero 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

from datetime import datetime, timedelta		from datetime import datetime, timedelta

import pytest		import pytest

		from django.core.exceptions import ObjectDoesNotExist
from django.utils import timezone		from django.utils import timezone

		from swh.web.auth.utils import SWH_AMBASSADOR_PERMISSION
from swh.web.common.models import (		from swh.web.common.models import (
SAVE_REQUEST_ACCEPTED,		SAVE_REQUEST_ACCEPTED,
SAVE_REQUEST_PENDING,		SAVE_REQUEST_PENDING,
SAVE_REQUEST_REJECTED,		SAVE_REQUEST_REJECTED,
SAVE_TASK_FAILED,		SAVE_TASK_FAILED,
SAVE_TASK_NOT_CREATED,		SAVE_TASK_NOT_CREATED,
SAVE_TASK_NOT_YET_SCHEDULED,		SAVE_TASK_NOT_YET_SCHEDULED,
SAVE_TASK_SCHEDULED,		SAVE_TASK_SCHEDULED,
▲ Show 20 Lines • Show All 408 Lines • ▼ Show 20 Lines	def test_save_request_form_server_error(api_client, mocker):
create_save_origin_request.side_effect = Exception("Server error")		create_save_origin_request.side_effect = Exception("Server error")

url = reverse(		url = reverse(
"api-1-save-origin",		"api-1-save-origin",
url_args={"visit_type": _visit_type, "origin_url": _origin_url},		url_args={"visit_type": _visit_type, "origin_url": _origin_url},
)		)

check_api_post_responses(api_client, url, status_code=500)		check_api_post_responses(api_client, url, status_code=500)


		@pytest.fixture
		def origin_to_review():
		return "https://git.example.org/user/project"


		def test_create_save_request_pending_review_anonymous_user(
		api_client, origin_to_review, requests_mock
		):

		# see swh.web.common.origin_save.origin_exists
		requests_mock.head(
		origin_to_review, status_code=200,
		)

		url = reverse(
		"api-1-save-origin",
		url_args={"visit_type": "git", "origin_url": origin_to_review},
		)

		response = check_api_post_responses(api_client, url, status_code=200)

		assert response.data["save_request_status"] == SAVE_REQUEST_PENDING

		with pytest.raises(ObjectDoesNotExist):
		SaveAuthorizedOrigin.objects.get(url=origin_to_review)


		def test_create_save_request_accepted_ambassador_user(
		api_client, origin_to_review, requests_mock, keycloak_oidc, mocker
		):

		keycloak_oidc.realm_permissions += [SWH_AMBASSADOR_PERMISSION]
		oidc_profile = keycloak_oidc.login()
		api_client.credentials(HTTP_AUTHORIZATION=f"Bearer {oidc_profile['refresh_token']}")

		# see swh.web.common.origin_save.origin_exists
		requests_mock.head(
		origin_to_review, status_code=200,
		)

		check_created_save_request_status(
		api_client,
		mocker,
		origin_to_review,
		expected_request_status=SAVE_REQUEST_ACCEPTED,
		expected_task_status=SAVE_TASK_NOT_YET_SCHEDULED,
		)

		assert SaveAuthorizedOrigin.objects.get(url=origin_to_review)