def test_updates(self, monkeypatch, data, fields_to_update):
"""
Test if the update_company task is called with the right parameters for all the records
spread across pages.
"""
mock_get_company_update_page = mock.Mock(
side_effect=lambda _, next_page: data[next_page],
)
monkeypatch.setattr(
'datahub.dnb_api.tasks.update.get_company_update_page',
mock_get_company_update_page,
)
mock_update_company = mock.Mock()
monkeypatch.setattr(
'datahub.dnb_api.tasks.update.update_company_from_dnb_data',
mock_update_company,
)
task_result = get_company_updates.apply_async(
kwargs={'fields_to_update': fields_to_update}, )
assert mock_get_company_update_page.call_count == 2
mock_get_company_update_page.assert_any_call(
'2019-01-01T00:00:00',
None,
)
mock_get_company_update_page.assert_any_call(
'2019-01-01T00:00:00',
'http://foo.bar/companies?cursor=page2',
)
assert mock_update_company.apply_async.call_count == 3
expected_kwargs = {
'fields_to_update': fields_to_update,
'update_descriptor':
f'celery:get_company_updates:{task_result.id}',
}
mock_update_company.apply_async.assert_any_call(
args=({
'foo': 1
}, ),
kwargs=expected_kwargs,
)
mock_update_company.apply_async.assert_any_call(
args=({
'bar': 2
}, ),
kwargs=expected_kwargs,
)
mock_update_company.apply_async.assert_any_call(
args=({
'baz': 3
}, ),
kwargs=expected_kwargs,
)
def test_updates_with_update_company_from_dnb_data(
self,
mocked_log_to_sentry,
mocked_send_realtime_message,
caplog,
monkeypatch,
dnb_company_updates_response_uk,
):
"""
Test full integration for the `get_company_updates` task with the
`update_company_from_dnb_data` task when all fields are updated.
"""
caplog.set_level('INFO')
company = CompanyFactory(duns_number='123456789')
mock_get_company_update_page = mock.Mock(
return_value=dnb_company_updates_response_uk,
)
monkeypatch.setattr(
'datahub.dnb_api.tasks.update.get_company_update_page',
mock_get_company_update_page,
)
task_result = get_company_updates.apply_async()
company.refresh_from_db()
dnb_company = dnb_company_updates_response_uk['results'][0]
assert company.name == dnb_company['primary_name']
expected_gu_number = dnb_company['global_ultimate_duns_number']
assert company.global_ultimate_duns_number == expected_gu_number
mocked_log_to_sentry.assert_called_with(
'get_company_updates task completed.',
extra={
'success_count': 1,
'failure_count': 0,
'updated_company_ids': [str(company.pk)],
'producer_task_id': task_result.id,
'start_time': '2019-01-02T02:00:00+00:00',
'end_time': '2019-01-02T02:00:00+00:00',
},
)
expected_message = (
'datahub.dnb_api.tasks.update.get_company_updates '
'updated: 1; failed to update: 0'
)
mocked_send_realtime_message.assert_called_once_with(expected_message)
assert 'get_company_updates total update count: 1' in caplog.text
def test_updates_max_update_limit(self, monkeypatch, data):
"""
Test if the update_company task is called with the
right parameters for all the records spread across
pages.
"""
mock_get_company_update_page = mock.Mock(
side_effect=lambda _, next_page: data[next_page],
)
monkeypatch.setattr(
'datahub.dnb_api.tasks.update.get_company_update_page',
mock_get_company_update_page,
)
mock_update_company = mock.Mock()
monkeypatch.setattr(
'datahub.dnb_api.tasks.update.update_company_from_dnb_data',
mock_update_company,
)
task_result = get_company_updates.apply_async()
assert mock_update_company.apply_async.call_count == 2
expected_kwargs = {
'fields_to_update': None,
'update_descriptor':
f'celery:get_company_updates:{task_result.id}',
}
mock_update_company.apply_async.assert_any_call(
args=({
'foo': 1
}, ),
kwargs=expected_kwargs,
)
mock_update_company.apply_async.assert_any_call(
args=({
'bar': 2
}, ),
kwargs=expected_kwargs,
)