# 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, shared_task


@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"


@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


@shared_task(name="swh.scheduler.tests.tasks.error")
def not_implemented():
    raise NotImplementedError("Nope")


@shared_task(name="swh.scheduler.tests.tasks.add")
def add(x, y):
    return x + y


@shared_task(name="swh.scheduler.tests.tasks.echo")
def echo(**kw):
    "Does nothing, just return the given kwargs dict"
    return kw