def test_task_generate_ids_and_operate_on_payloads_no_action(
self,
get_repeat_record_ids_mock,
unused_1,
unused_2,
):
get_repeat_record_ids_mock.return_value = ['c0ffee', 'deadbeef']
response = task_generate_ids_and_operate_on_payloads(
payload_id='c0ffee',
repeater_id=None,
domain='test_domain',
action='',
use_sql=False,
)
self.assertEqual(response, {
'messages': {
'errors': [
"Could not perform action for repeat record (id=c0ffee): "
"Unknown action ''",
"Could not perform action for repeat record (id=deadbeef): "
"Unknown action ''",
],
'success': [],
'success_count_msg': '',
}
})
def test_generate_ids_and_operate_on_payloads_success(
self, mock_operate_on_payloads, mock__get_ids):
payload_id = 'c0ffee'
repeater_id = 'deadbeef'
task_generate_ids_and_operate_on_payloads(payload_id, repeater_id,
'test_domain', 'test_action')
mock__get_ids.assert_called_once()
mock__get_ids.assert_called_with('c0ffee', 'deadbeef', 'test_domain')
mock_record_ids = mock__get_ids('c0ffee', 'deadbeef', 'test_domain')
mock_operate_on_payloads.assert_called_once()
mock_operate_on_payloads.assert_called_with(
mock_record_ids,
'test_domain',
'test_action',
task=task_generate_ids_and_operate_on_payloads)
def test_task_generate_ids_and_operate_on_payloads_no_data(self):
response = task_generate_ids_and_operate_on_payloads(
payload_id=None,
repeater_id=None,
domain='test_domain',
action=''
)
self.assertEqual(response,
{'messages': {'errors': ['No payloads specified']}})
def test_task_generate_ids_and_operate_on_payloads_valid(
self, mock_task, mock_generate_ids):
payload = {
'data': ['payload_id'],
'domain': 'test_domain',
'action': 'test_action',
}
response = task_generate_ids_and_operate_on_payloads(**payload)
self.assertEqual(mock_generate_ids.call_count, 1)
mock_generate_ids.assert_called_with(['payload_id'], 'test_domain',
'test_action', mock_task)
def test_task_generate_ids_and_operate_on_payloads_no_data(
self, mock_generate_ids):
with patch('corehq.apps.data_interfaces.tasks._') as _:
response = task_generate_ids_and_operate_on_payloads(
data=[], domain='test_domain', action='')
expected_response = {
'messages': {
'errors': [_('No data is supplied')]
}
}
self.assertEqual(response, expected_response)
self.assertEqual(mock_generate_ids.call_count, 0)
def test_task_generate_ids_and_operate_on_payloads_no_action(
self,
get_repeat_record_ids_mock,
):
get_repeat_record_ids_mock.return_value = ['c0ffee', 'deadbeef']
response = task_generate_ids_and_operate_on_payloads(
payload_id='c0ffee',
repeater_id=None,
domain='test_domain',
action=''
)
self.assertEqual(response,
{'messages': {'errors': ['No action specified']}})
