commit 15be0e1e1112c8f2e0cc87961d7148492c8e3406
author James E. Blair <jeblair@redhat.com> Tue Jan 03 13:45:20 2017 -0800
committer James E. Blair <jeblair@redhat.com> Wed Jan 04 09:11:35 2017 -0800
tree 75a8241bb037f1e18b98ae2d1bfd0491fd43e938
parent 3897a13743955b0eef82d4a9f184a48efce8fc13

Re-submit node requests on ZooKeeper disconnect

Change-Id: I689bf812c713fa6f5f37958b7001b0d5fb0a254b

tests/base.py

diff --git a/tests/base.py b/tests/base.py
index cbb06b7..9ee5838 100755
--- a/tests/base.py
+++ b/tests/base.py
@@ -883,6 +883,7 @@
             hosts='%s:%s%s' % (host, port, chroot))
         self.client.start()
         self._running = True
+        self.paused = False
         self.thread = threading.Thread(target=self.run)
         self.thread.daemon = True
         self.thread.start()
@@ -899,6 +900,8 @@
             time.sleep(0.1)
 
     def _run(self):
+        if self.paused:
+            return
         for req in self.getNodeRequests():
             self.fulfillRequest(req)
 
@@ -1501,6 +1504,8 @@
         return True
 
     def areAllNodeRequestsComplete(self):
+        if self.fake_nodepool.paused:
+            return True
         if self.sched.nodepool.requests:
             return False
         return True