No OneTemporary
Actions

Size

1 KB

Subscribers

None

View Options

	diff --git a/swh/scheduler/tests/tasks.py b/swh/scheduler/tests/tasks.py
	index 93ee0f1..be7628d 100644
	--- a/swh/scheduler/tests/tasks.py
	+++ b/swh/scheduler/tests/tasks.py
	@@ -1,38 +1,36 @@
	# Copyright (C) 2018-2019 The Software Heritage developers
	# See the AUTHORS file at the top-level directory of this distribution
	# License: GNU General Public License version 3, or any later version
	# See top-level LICENSE file for more information

	-from celery import group
	+from celery import group, shared_task

	-from swh.scheduler.celery_backend.config import app

	-
	-@app.task(name='swh.scheduler.tests.tasks.ping', bind=True)
	+@shared_task(name='swh.scheduler.tests.tasks.ping', bind=True)
	def ping(self, **kw):
	# check this is a SWHTask
	assert hasattr(self, 'log')
	assert not hasattr(self, 'run_task')
	assert 'SWHTask' in [x.__name__ for x in self.__class__.__mro__]
	self.log.debug(self.name)
	if kw:
	return 'OK (kw=%s)' % kw
	return 'OK'


	-@app.task(name='swh.scheduler.tests.tasks.multiping', bind=True)
	+@shared_task(name='swh.scheduler.tests.tasks.multiping', bind=True)
	def multiping(self, n=10):
	promise = group(ping.s(i=i) for i in range(n))()
	self.log.debug('%s OK (spawned %s subtasks)' % (self.name, n))
	promise.save()
	return promise.id


	-@app.task(name='swh.scheduler.tests.tasks.error')
	+@shared_task(name='swh.scheduler.tests.tasks.error')
	def not_implemented():
	raise NotImplementedError('Nope')


	-@app.task(name='swh.scheduler.tests.tasks.add')
	+@shared_task(name='swh.scheduler.tests.tasks.add')
	def add(x, y):
	return x + y