Changeset View
Changeset View
Standalone View
Standalone View
swh/indexer/tests/test_journal_client.py
- This file was added.
| # Copyright (C) 2019 The Software Heritage developers | |||||
| # See the AUTHORS file at the top-level directory of this distribution | |||||
| # License: GNU General Public License version 3, or any later version | |||||
| # See top-level LICENSE file for more information | |||||
| import unittest | |||||
| from unittest.mock import Mock | |||||
| from swh.indexer.journal_client import process_journal_objects | |||||
| class JournalClientTest(unittest.TestCase): | |||||
| def testOriginVisit(self): | |||||
| mock_scheduler = Mock() | |||||
| messages = { | |||||
| 'origin_visit': [ | |||||
| { | |||||
| b'status': b'full', | |||||
| b'origin': { | |||||
| b'url': 'file:///dev/zero', | |||||
| } | |||||
| } | |||||
| ] | |||||
| } | |||||
| process_journal_objects( | |||||
| messages, scheduler=mock_scheduler, | |||||
| task_names={'origin_metadata': 'task-name'}) | |||||
| self.assertTrue(mock_scheduler.create_tasks.called) | |||||
| call_args = mock_scheduler.create_tasks.call_args | |||||
| (args, kwargs) = call_args | |||||
| self.assertEqual(kwargs, {}) | |||||
| del args[0][0]['next_run'] | |||||
| self.assertEqual(args, ([ | |||||
| { | |||||
| 'arguments': { | |||||
| 'kwargs': {'policy_update': 'update-dups'}, | |||||
| 'args': (['file:///dev/zero'],) | |||||
| }, | |||||
| 'policy': 'oneshot', | |||||
| 'type': 'task-name' | |||||
| } | |||||
| ],)) | |||||