Skip to content
Snippets Groups Projects
Commit f63b8326 authored by David Douard's avatar David Douard
Browse files

Add tests for npm tasks

parent 264e9ea5
No related branches found
No related tags found
No related merge requests found
......@@ -13,6 +13,7 @@ def celery_includes():
'swh.lister.debian.tasks',
'swh.lister.github.tasks',
'swh.lister.gitlab.tasks',
'swh.lister.npm.tasks',
]
......
from swh.lister.core.tests.conftest import * # noqa
from contextlib import contextmanager
from unittest.mock import patch
@contextmanager
def mock_save(lister):
yield
def test_ping(swh_app, celery_session_worker):
res = swh_app.send_task(
'swh.lister.npm.tasks.ping')
assert res
res.wait()
assert res.successful()
assert res.result == 'OK'
@patch('swh.lister.npm.tasks.save_registry_state')
@patch('swh.lister.npm.tasks.NpmLister')
def test_lister(lister, save, swh_app, celery_session_worker):
# setup the mocked NpmLister
lister.return_value = lister
lister.run.return_value = None
save.side_effect = mock_save
res = swh_app.send_task('swh.lister.npm.tasks.NpmListerTask')
assert res
res.wait()
assert res.successful()
lister.assert_called_once_with()
lister.run.assert_called_once_with()
@patch('swh.lister.npm.tasks.save_registry_state')
@patch('swh.lister.npm.tasks.get_last_update_seq')
@patch('swh.lister.npm.tasks.NpmIncrementalLister')
def test_incremental(lister, seq, save, swh_app, celery_session_worker):
# setup the mocked NpmLister
lister.return_value = lister
lister.run.return_value = None
lister.request_headers.return_value = []
seq.return_value = 42
save.side_effect = mock_save
res = swh_app.send_task(
'swh.lister.npm.tasks.NpmIncrementalListerTask')
assert res
res.wait()
assert res.successful()
lister.assert_called_once_with()
seq.assert_called_once_with(lister)
lister.run.assert_called_once_with(min_bound=42)
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment