Changeset View
Changeset View
Standalone View
Standalone View
swh/lister/pubdev/tests/test_lister.py
| # Copyright (C) 2022 The Software Heritage developers | # Copyright (C) 2022 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 swh.lister.pubdev.lister import USER_AGENT, PubDevLister | from swh.lister import USER_AGENT_TEMPLATE | ||||
| from swh.lister.pubdev.lister import PubDevLister | |||||
| expected_origins = { | expected_origins = { | ||||
| "https://pub.dev/packages/Autolinker", | "https://pub.dev/packages/Autolinker", | ||||
| "https://pub.dev/packages/Babylon", | "https://pub.dev/packages/Babylon", | ||||
| } | } | ||||
| def test_pubdev_lister(datadir, requests_mock_datadir, swh_scheduler): | def test_pubdev_lister(datadir, requests_mock_datadir, swh_scheduler): | ||||
| Show All 9 Lines | def test_pubdev_lister(datadir, requests_mock_datadir, swh_scheduler): | ||||
| for origin in scheduler_origins: | for origin in scheduler_origins: | ||||
| assert origin.visit_type == "pubdev" | assert origin.visit_type == "pubdev" | ||||
| assert origin.url in expected_origins | assert origin.url in expected_origins | ||||
| assert origin.last_update is not None | assert origin.last_update is not None | ||||
| def _match_request(request): | def _match_request(request): | ||||
| return request.headers.get("User-Agent") == USER_AGENT | return ( | ||||
| request.headers.get("User-Agent") | |||||
| == USER_AGENT_TEMPLATE % PubDevLister.LISTER_NAME | |||||
| ) | |||||
| def test_pubdev_lister_skip_package( | def test_pubdev_lister_skip_package( | ||||
| datadir, requests_mock_datadir, swh_scheduler, requests_mock | datadir, requests_mock_datadir, swh_scheduler, requests_mock | ||||
| ): | ): | ||||
| requests_mock.get( | requests_mock.get( | ||||
| "https://pub.dev/api/packages/Autolinker", | "https://pub.dev/api/packages/Autolinker", | ||||
| Show All 9 Lines | |||||