Adding more tests for unexpected exits during the job execution.

Author: cezarsa

tests/__init__.py

@@ -65,6 +65,14 @@ def perform(exit_code):
         sys.exit(exit_code)
         return "Never got here"
 
+class PrematureHardExitJob(object):
+    queue = 'basic'
+
+    @staticmethod
+    def perform(exit_code):
+        os._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, PrematureExitJob
+from tests import PyResTests, Basic, TestProcess, ErrorObject, RetryOnExceptionJob, TimeoutJob, CrashJob, PrematureExitJob, PrematureHardExitJob
 from pyres import ResQ
 from pyres.job import Job
 from pyres.scheduler import Scheduler
@@ -248,6 +248,30 @@ def test_detect_code_0_sys_exit_as_success(self):
         assert worker.job() == {}
         assert worker.get_failed() == 0
 
+    def test_detect_non_0_os_exit_as_failure(self):
+        worker = Worker(['basic'])
+        self.resq.enqueue(PrematureHardExitJob, 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_os_exit_as_success(self):
+        worker = Worker(['basic'])
+        self.resq.enqueue(PrematureHardExitJob, 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)