Changeset View
Changeset View
Standalone View
Standalone View
swh/deposit/tests/loader/test_checker.py
| # Copyright (C) 2017-2019 The Software Heritage developers | # Copyright (C) 2017-2019 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 | ||||
| from django.urls import reverse | |||||
| from unittest.mock import patch | from unittest.mock import patch | ||||
| from swh.deposit.config import PRIVATE_CHECK_DEPOSIT | |||||
| def test_check_deposit_ready(swh_config, requests_mock_datadir, deposit_checker): | def test_check_deposit_ready(swh_config, requests_mock_datadir, deposit_checker): | ||||
| """Check on a valid 'deposited' deposit should result in 'verified' | """Check on a valid 'deposited' deposit should result in 'verified' | ||||
| """ | """ | ||||
| deposit_check_url = reverse(PRIVATE_CHECK_DEPOSIT, args=["test", 1]) | actual_result = deposit_checker.check(collection="test", deposit_id=1) | ||||
| actual_result = deposit_checker.check(deposit_check_url=deposit_check_url) | |||||
| assert actual_result == {"status": "eventful"} | assert actual_result == {"status": "eventful"} | ||||
| def test_check_deposit_rejected(swh_config, requests_mock_datadir, deposit_checker): | def test_check_deposit_rejected(swh_config, requests_mock_datadir, deposit_checker): | ||||
| """Check on invalid 'deposited' deposit should result in 'rejected' | """Check on invalid 'deposited' deposit should result in 'rejected' | ||||
| """ | """ | ||||
| deposit_check_url = reverse(PRIVATE_CHECK_DEPOSIT, args=["test", 2]) | actual_result = deposit_checker.check(collection="test", deposit_id=2) | ||||
| actual_result = deposit_checker.check(deposit_check_url=deposit_check_url) | |||||
| assert actual_result == {"status": "failed"} | assert actual_result == {"status": "failed"} | ||||
| @patch("swh.deposit.client.requests.get") | @patch("swh.deposit.client.requests.get") | ||||
| def test_check_deposit_rejected_exception(mock_requests, swh_config, deposit_checker): | def test_check_deposit_rejected_exception(mock_requests, swh_config, deposit_checker): | ||||
| """Check on invalid 'deposited' deposit should result in 'rejected' | """Check on invalid 'deposited' deposit should result in 'rejected' | ||||
| """ | """ | ||||
| mock_requests.side_effect = ValueError("simulated problem when checking") | mock_requests.side_effect = ValueError("simulated problem when checking") | ||||
| deposit_check_url = reverse(PRIVATE_CHECK_DEPOSIT, args=["test", 3]) | actual_result = deposit_checker.check(collection="test", deposit_id=3) | ||||
| actual_result = deposit_checker.check(deposit_check_url=deposit_check_url) | |||||
| assert actual_result == {"status": "failed"} | assert actual_result == {"status": "failed"} | ||||