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, [])