def test__get_ids_payload_id_in_data(
self, mock_iter_repeat_records_by_repeater,
mock_get_repeat_records_by_payload_id):
payload_id = Mock()
_get_repeat_record_ids(payload_id, None, 'test_domain')
self.assertEqual(mock_get_repeat_records_by_payload_id.call_count, 1)
mock_get_repeat_records_by_payload_id.assert_called_with(
'test_domain', payload_id)
self.assertEqual(mock_iter_repeat_records_by_repeater.call_count, 0)
def test__get_ids_payload_id_not_in_data(
self,
iter_by_repeater,
get_by_payload_id,
):
REPEATER_ID = 'c0ffee'
_get_repeat_record_ids(None, REPEATER_ID, 'test_domain', False)
get_by_payload_id.assert_not_called()
iter_by_repeater.assert_called_with('test_domain', REPEATER_ID)
self.assertEqual(iter_by_repeater.call_count, 1)
def test__get_ids_payload_id_in_data(
self,
iter_by_repeater,
get_by_payload_id,
):
payload_id = Mock()
_get_repeat_record_ids(payload_id, None, 'test_domain', False)
self.assertEqual(get_by_payload_id.call_count, 1)
get_by_payload_id.assert_called_with('test_domain', payload_id)
self.assertEqual(iter_by_repeater.call_count, 0)
def test__get_ids_no_data(self):
response = _get_repeat_record_ids(None, None, 'test_domain')
self.assertEqual(response, [])
def test_repeater_id_couch(self):
result = _get_repeat_record_ids(payload_id=None,
repeater_id=self.repeater._id,
domain=DOMAIN,
use_sql=False)
self.assertEqual(set(result), {r._id for r in self.couch_records})
def test_repeater_id_sql(self):
result = _get_repeat_record_ids(payload_id=None,
repeater_id=self.repeater._id,
domain=DOMAIN,
use_sql=True)
self.assertEqual(set(result), {r.pk for r in self.sql_records})
def test_no_payload_id_no_repeater_id_couch(self):
result = _get_repeat_record_ids(payload_id=None,
repeater_id=None,
domain=DOMAIN,
use_sql=False)
self.assertEqual(result, [])