D2162.diff
No OneTemporary
Actions

Size

1 KB

Subscribers

None

D2162.diff
View Options

	diff --git a/swh/scheduler/tests/tasks.py b/swh/scheduler/tests/tasks.py
	--- a/swh/scheduler/tests/tasks.py
	+++ b/swh/scheduler/tests/tasks.py
	@@ -3,12 +3,10 @@
	# 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')
	@@ -20,7 +18,7 @@
	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))
	@@ -28,11 +26,11 @@
	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