Changeset View
Changeset View
Standalone View
Standalone View
swh/search/tests/test_journal_client.py
| # Copyright (C) 2019-2020 The Software Heritage developers | # Copyright (C) 2019-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 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 functools | import functools | ||||
| from unittest.mock import MagicMock | from unittest.mock import MagicMock | ||||
| from swh.search.journal_client import process_journal_objects | from swh.search.journal_client import process_journal_objects | ||||
| Show All 17 Lines | def test_journal_client_origin_from_journal(): | ||||
| ) | ) | ||||
| def test_journal_client_origin_visit_from_journal(): | def test_journal_client_origin_visit_from_journal(): | ||||
| search_mock = MagicMock() | search_mock = MagicMock() | ||||
| worker_fn = functools.partial(process_journal_objects, search=search_mock,) | worker_fn = functools.partial(process_journal_objects, search=search_mock,) | ||||
| worker_fn({"origin_visit": [{"origin": {"url": "http://foobar.baz"},}]}) | worker_fn({"origin_visit": [{"origin": "http://foobar.baz", "type": "git"},]}) | ||||
| search_mock.origin_update.assert_called_once_with( | search_mock.origin_update.assert_called_once_with( | ||||
| [{"url": "http://foobar.baz"},] | [{"url": "http://foobar.baz", "visit_types": ["git"]},] | ||||
| ) | ) | ||||
| def test_journal_client_origin_visit_status_from_journal(): | def test_journal_client_origin_visit_status_from_journal(): | ||||
| search_mock = MagicMock() | search_mock = MagicMock() | ||||
| worker_fn = functools.partial(process_journal_objects, search=search_mock,) | worker_fn = functools.partial(process_journal_objects, search=search_mock,) | ||||
| ▲ Show 20 Lines • Show All 49 Lines • Show Last 20 Lines | |||||