Merge pull request #121 from guilhermef/master

Registers worker again after a Redis flush
binarymatt · Nov 29, 2012 · 7d2843c · 7d2843c
2 parents c00c160 + c7af5db
commit 7d2843c
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 1 deletion.
diff --git a/pyres/worker.py b/pyres/worker.py
@@ -138,6 +138,8 @@ def work(self, interval=5):
                 logger.info('shutdown scheduled')
                 break
 
+            self.register_worker()
+
             job = self.reserve(interval)
 
             if job:

diff --git a/tests/test_worker.py b/tests/test_worker.py
@@ -38,7 +38,7 @@ def test_unregister(self):
         assert self.redis.sismember('resque:workers',name)
         worker.unregister_worker()
         assert name not in self.redis.smembers('resque:workers')
-    
+
     def test_working_on(self):
         name = "%s:%s:%s" % (os.uname()[1],os.getpid(),'basic')
         self.resq.enqueue(Basic,"test1")