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/test_nodepool.py

diff --git a/tests/test_nodepool.py b/tests/test_nodepool.py
index 6a14192..3fb0335 100644
--- a/tests/test_nodepool.py
+++ b/tests/test_nodepool.py
@@ -69,3 +69,20 @@
         self.waitForRequests()
         self.assertEqual(len(self.provisioned_requests), 1)
         self.assertEqual(request.state, 'fulfilled')
+
+    def test_node_request_disconnect(self):
+        # Test that node requests are re-submitted after disconnect
+
+        nodeset = model.NodeSet()
+        nodeset.addNode(model.Node('controller', 'ubuntu-xenial'))
+        nodeset.addNode(model.Node('compute', 'ubuntu-xenial'))
+        job = model.Job('testjob')
+        job.nodeset = nodeset
+        self.fake_nodepool.paused = True
+        request = self.nodepool.requestNodes(None, job)
+        self.zk.client.stop()
+        self.zk.client.start()
+        self.fake_nodepool.paused = False
+        self.waitForRequests()
+        self.assertEqual(len(self.provisioned_requests), 1)
+        self.assertEqual(request.state, 'fulfilled')