Adding a test for job killing and sleep test task (tests/{test_kill.py,tasks/general.py)

Author: DasFranck

tests/tasks/general.py

@@ -83,6 +83,12 @@ def run(self, params):
             pass
 
 
+class Sleep(Task):
+    def run(self, params):
+        time.sleep(params.get("sleep", 100))
+        return True
+
+
 class Retry(Task):
 
     def run(self, params):

tests/test_kill.py

@@ -0,0 +1,35 @@
+from mrq.job import Job
+import time
+
+
+def test_kill_by_id(worker):
+    worker.start()
+
+    job_id1 = worker.send_task("tests.tasks.general.Sleep", {}, block=False)
+
+    Job(job_id1).kill()
+    time.sleep(1)
+
+    job1 = Job(job_id1).fetch().data
+
+    assert job1["status"] == "killed"
+
+
+# def test_kill_by_path(worker):
+#     worker.start()
+
+#     job_id1 = worker.send_task("tests.tasks.general.Sleep", {}, block=False)
+#     job_id2 = worker.send_task("tests.tasks.general.Sleep", {}, block=False)
+
+#     worker.send_task("mrq.basetasks.utils.JobAction", {
+#         "path": "tests.tasks.general.Sleep",
+#         "action": "kill"
+#     }, block=False)
+
+#     time.sleep(1)
+
+#     job1 = Job(job_id1).fetch().data
+#     job2 = Job(job_id2).fetch().data
+
+#     assert job1["status"] == "killed"
+#     assert job2["status"] == "killed"