def test_assert_enqueue_called_with_same_kwargs(self, queue):
kwargs = {'key': 1}
api_index_enqueue(self.function, **kwargs)
queue().enqueue_call.assert_called_with(self.function,
tuple(),
kwargs,
timeout=self.timeout)
def index_catalog(node: Node, task, read_local=False, force=False):
"""Ejecuta el pipeline de lectura, guardado e indexado de datos
y metadatos sobre cada distribución del catálogo especificado
"""
try:
catalog = DataJson(node.catalog_url,
catalog_format=node.catalog_format)
node.catalog = json.dumps(catalog)
node.save()
except Exception as e:
IndexDataTask.info(task, READ_ERROR.format(node.catalog_id, e))
return
distributions = Distribution.objects.filter(
present=True,
dataset__indexable=True,
dataset__catalog__identifier=node.catalog_id)
for distribution in distributions:
api_index_enqueue(index_distribution,
distribution.identifier,
node.id,
task.id,
read_local,
force=force)
def test_enqueue_called_with_timeout_from_config(self, queue):
api_index_enqueue(self.function)
timeout = queue().enqueue_call.call_args[1]['timeout']
self.assertEqual(timeout, self.timeout)
def test_assert_enqueue_called_with_same_args(self, queue):
args = (1, )
api_index_enqueue(self.function, *args)
queue().enqueue_call.assert_called_with(self.function,
args, {},
timeout=self.timeout)
def test_queue_called(self, queue):
api_index_enqueue(self.function)
queue().enqueue_call.assert_called_once()