Fixed issue with PicklingError causes the timeout to never be reached.

4 files changed, 21 insertions, 7 deletions

diff --git a/.gitignore b/.gitignore
index 280c69c..c0ffae1 100644
--- a/.gitignore
+++ b/.gitignore
@@ -28,6 +28,7 @@ pip-log.txt
 # Unit test / coverage reports
 .coverage
 .tox
+.cache
 nosetests.xml
 
 # Translations
diff --git a/CHANGES.rst b/CHANGES.rst
index dd67619..1fd3175 100644
--- a/CHANGES.rst
+++ b/CHANGES.rst
@@ -1,6 +1,10 @@
 Changelog
 =========
 
+0.3.1
+-----
+- Fixed issue with PicklingError causes the timeout to never be reached.
+
 0.3.0
 -----
 
diff --git a/tests/test_timeout_decorator.py b/tests/test_timeout_decorator.py
index 12d5389..e8e9fbf 100644
--- a/tests/test_timeout_decorator.py
+++ b/tests/test_timeout_decorator.py
@@ -48,3 +48,15 @@ def test_function_name(use_signals):
         pass
 
     assert func_name.__name__ == 'func_name'
+
+
+def test_timeout_pickle_error():
+    """Test that when a pickle error occurs a timeout error is raised."""
+    @timeout(seconds=1, use_signals=False)
+    def f():
+        time.sleep(0.1)
+        class Test(object):
+            pass
+        return Test()
+    with pytest.raises(TimeoutError):
+        f()
diff --git a/timeout_decorator/timeout_decorator.py b/timeout_decorator/timeout_decorator.py
index 014abc0..17798e3 100644
--- a/timeout_decorator/timeout_decorator.py
+++ b/timeout_decorator/timeout_decorator.py
@@ -118,7 +118,6 @@ class _Timeout:
         True, the "value" property may then be checked for returned data.
         """
         self.__limit = kwargs.pop('timeout', self.__limit)
-        self.cancel()
         self.__queue = multiprocessing.Queue(1)
         args = (self.__queue, self.__function) + args
         self.__process = multiprocessing.Process(target=_target,
@@ -135,17 +134,15 @@ class _Timeout:
         """Terminate any possible execution of the embedded function."""
         if self.__process.is_alive():
             self.__process.terminate()
-            raise TimeoutError()
+        raise TimeoutError()
 
     @property
     def ready(self):
         """Read-only property indicating status of "value" property."""
-        if self.__queue.full():
-            return True
-        elif not self.__queue.empty():
-            return True
-        elif self.__timeout < time.time():
+        if self.__timeout < time.time():
             self.cancel()
+        if self.__queue.full() and not self.__queue.empty():
+            return True
         else:
             return False