Enable usage of Timeout class without 'with' block

Signed-off-by: Andrea Cervesato <[email protected]>

Author: acerv

ltp/tests/test_utils.py

@@ -55,3 +55,11 @@ def test_timeout_custom_all(self):
             with Timeout(0.01) as timeout:
                 time.sleep(0.01)
                 timeout.check(err_msg='error message', exc=TimeoutError)
+
+    def test_no_with(self):
+        """
+        Test class without 'with' block.
+        """
+        timeout = Timeout(1)
+        time.sleep(0.01)
+        timeout.check()

ltp/utils.py

@@ -41,6 +41,9 @@ def check(self, err_msg: str = None, exc: Exception = None):
         """
         Check if time is out.
         """
+        if not self._end:
+            self._end = time.time() + self._timeout
+
         if self._end > time.time():
             return