Merge pull request #51 from seomoz/add-requeue

Add requeue.

api.lua

@@ -148,6 +148,12 @@ QlessAPI.put = function(now, me, queue, jid, klass, data, delay, ...)
   return Qless.queue(queue):put(now, me, jid, klass, data, delay, unpack(arg))
 end
 
+QlessAPI.requeue = function(now, me, queue, jid, ...)
+  local job = Qless.job(jid)
+  assert(job:exists(), 'Requeue(): Job ' .. jid .. ' does not exist')
+  return QlessAPI.put(now, me, queue, jid, unpack(arg))
+end
+
 QlessAPI.unfail = function(now, queue, group, count)
   return Qless.queue(queue):unfail(now, group, count)
 end

test/test_job.py

@@ -40,6 +40,19 @@ def test_history(self):
             {'q': 'queue', 'what': 'put', 'when': 98},
             {'q': 'queue', 'what': 'put', 'when': 99}])
 
+class TestRequeue(TestQless):
+    def test_requeue_existing_job(self):
+        '''Requeueing an existing job is identical to `put`'''
+        self.lua('put', 0, 'worker', 'queue', 'jid', 'klass', {}, 0)
+        self.lua('requeue', 1, 'worker', 'queue-2', 'jid', 'klass', {}, 0)
+        self.assertEqual(self.lua('get', 0, 'jid')['queue'], 'queue-2')
+
+    def test_requeue_cancelled_job(self):
+        '''Requeueing a cancelled (or non-existent) job fails'''
+        self.lua('put', 0, 'worker', 'queue', 'jid', 'klass', {}, 0)
+        self.lua('cancel', 1, 'jid')
+        self.assertRaisesRegexp(redis.ResponseError, r'does not exist',
+            self.lua, 'requeue', 2, 'worker', 'queue-2', 'jid', 'klass', {}, 0)
 
 class TestComplete(TestQless):
     '''Test how we complete jobs'''