Adding tests for calling sys.exit() from job's perform method. (Currently failing)

Author: cezarsa

tests/__init__.py

@@ -56,6 +56,15 @@ def perform():
         ctypes.string_at(1)
         return "Never got here"
 
+class PrematureExitJob(object):
+    queue = 'basic'
+
+    @staticmethod
+    def perform(exit_code):
+        import sys
+        sys.exit(exit_code)
+        return "Never got here"
+
 class TestProcess(object):
     queue = 'high'
 

tests/test_worker.py

@@ -1,4 +1,4 @@
-from tests import PyResTests, Basic, TestProcess, ErrorObject, RetryOnExceptionJob, TimeoutJob, CrashJob
+from tests import PyResTests, Basic, TestProcess, ErrorObject, RetryOnExceptionJob, TimeoutJob, CrashJob, PrematureExitJob
 from pyres import ResQ
 from pyres.job import Job
 from pyres.scheduler import Scheduler
@@ -224,6 +224,30 @@ def test_detect_crashed_workers_as_failures(self):
         assert worker.job() == {}
         assert worker.get_failed() == 1
 
+    def test_detect_non_0_sys_exit_as_failure(self):
+        worker = Worker(['basic'])
+        self.resq.enqueue(PrematureExitJob, 9)
+
+        assert worker.job() == {}
+        assert worker.get_failed() == 0
+
+        worker.fork_worker(worker.reserve())
+
+        assert worker.job() == {}
+        assert worker.get_failed() == 1
+
+    def test_detect_code_0_sys_exit_as_success(self):
+        worker = Worker(['basic'])
+        self.resq.enqueue(PrematureExitJob, 0)
+
+        assert worker.job() == {}
+        assert worker.get_failed() == 0
+
+        worker.fork_worker(worker.reserve())
+
+        assert worker.job() == {}
+        assert worker.get_failed() == 0
+
     def test_retries_give_up_eventually(self):
         now = datetime.datetime.now()
         self.set_current_time(now)