lister.tests: Add missing task_type for package listers

The scheduler module no longer initializes itself those task_type.
This commit is contained in:
Antoine R. Dumont (@ardumont) 2019-11-21 13:33:16 +01:00
parent 5191d291da
commit 1cf7c8e86b
No known key found for this signature in database
GPG key ID: 52E2E9840D10C3B8
10 changed files with 104 additions and 22 deletions

View file

@ -60,6 +60,7 @@ class CRANLister(SimpleLister):
policy = kwargs.get('policy', 'oneshot')
package = kwargs.get('name')
version = kwargs.get('version')
assert origin_type == 'tar'
return create_task_dict(
'load-%s' % origin_type,
policy, package, origin_url, version,

View file

@ -1 +1,23 @@
# 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 pytest
from swh.lister.core.tests.conftest import * # noqa
@pytest.fixture
def lister_cran(swh_listers):
lister = swh_listers['cran']
# Add the load-deb-package in the scheduler backend
lister.scheduler.create_task_type({
'type': 'load-tar',
'description': 'Load archive files',
'backend_name': 'swh.loader.package.tasks.LoadArchive',
'default_interval': '1 day',
})
return lister

View file

@ -28,18 +28,16 @@ def test_cran_compute_package_url_failure():
@patch('swh.lister.cran.lister.read_cran_data')
def test_cran_lister_cran(mock_cran, datadir, swh_listers):
lister = swh_listers['cran']
def test_cran_lister_cran(mock_cran, datadir, lister_cran):
with open(path.join(datadir, 'list-r-packages.json')) as f:
data = json.loads(f.read())
mock_cran.return_value = data
assert len(data) == 6
lister.run()
lister_cran.run()
r = lister.scheduler.search_tasks(task_type='load-tar')
r = lister_cran.scheduler.search_tasks(task_type='load-tar')
assert len(r) == 6
for row in r: