From b86ad744be25c8a1f17546a8861f32abd4253a71 Mon Sep 17 00:00:00 2001
From: "Ivan A. Melnikov" <imelnikov@griddynamics.com>
Date: Wed, 26 Mar 2014 13:46:29 +0400
Subject: [PATCH] Add timeout to WaitForOneFromTask

We should never wait forever for anything.

Change-Id: Icaa61febbf4745206abbe5bf9255c9401f341e71
---
 taskflow/tests/utils.py | 8 +++++++-
 1 file changed, 7 insertions(+), 1 deletion(-)

diff --git a/taskflow/tests/utils.py b/taskflow/tests/utils.py
index 64147cf3..20fc3758 100644
--- a/taskflow/tests/utils.py
+++ b/taskflow/tests/utils.py
@@ -307,7 +307,13 @@ class WaitForOneFromTask(SaveOrderTask):
         self.event = threading.Event()
 
     def execute(self):
-        self.event.wait()
+        # NOTE(imelnikov): if test was not complete within
+        # 5 minutes, something is terribly wrong
+        self.event.wait(300)
+        if not self.event.is_set():
+            raise RuntimeError('Timeout occurred while waiting '
+                               'for %s to change state to %s'
+                               % (self.wait_for, self.wait_states))
         return super(WaitForOneFromTask, self).execute()
 
     def callback(self, state, details):