def test_node_request(self): # Test a simple node request nodeset = model.NodeSet() nodeset.addNode(model.Node(['controller', 'foo'], 'ubuntu-xenial')) nodeset.addNode(model.Node(['compute'], 'ubuntu-xenial')) job = model.Job('testjob') job.nodeset = nodeset request = self.nodepool.requestNodes(None, job, 0) self.waitForRequests() self.assertEqual(len(self.provisioned_requests), 1) self.assertEqual(request.state, 'fulfilled') # Accept the nodes self.nodepool.acceptNodes(request, request.id) nodeset = request.nodeset for node in nodeset.getNodes(): self.assertIsNotNone(node.lock) self.assertEqual(node.state, 'ready') # Mark the nodes in use self.nodepool.useNodeSet(nodeset) for node in nodeset.getNodes(): self.assertEqual(node.state, 'in-use') # Return the nodes self.nodepool.returnNodeSet(nodeset) for node in nodeset.getNodes(): self.assertIsNone(node.lock) self.assertEqual(node.state, 'used')
def test_node_request_canceled(self): # Test that node requests can be canceled 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.pause() request = self.nodepool.requestNodes(None, job, 0) self.nodepool.cancelRequest(request) self.waitForRequests() self.assertEqual(len(self.provisioned_requests), 0)
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.pause() request = self.nodepool.requestNodes(None, job, 0) self.zk.client.stop() self.zk.client.start() self.fake_nodepool.unpause() self.waitForRequests() self.assertEqual(len(self.provisioned_requests), 1) self.assertEqual(request.state, 'fulfilled')
def test_node_request(self): # Test a simple node request nodeset = model.NodeSet() nodeset.addNode(model.Node(['controller'], 'fake-label')) job = model.Job('testjob') job.nodeset = nodeset request = self.nodepool.requestNodes(None, job, 0) self.waitForRequests() self.assertEqual(len(self.provisioned_requests), 1) self.assertEqual(request.state, model.STATE_FULFILLED) # Accept the nodes self.nodepool.acceptNodes(request, request.id) nodeset = request.nodeset for node in nodeset.getNodes(): self.assertIsNotNone(node.lock) self.assertEqual(node.state, model.STATE_READY) # Mark the nodes in use self.nodepool.useNodeSet(nodeset) for node in nodeset.getNodes(): self.assertEqual(node.state, model.STATE_IN_USE) # Return the nodes self.nodepool.returnNodeSet(nodeset) for node in nodeset.getNodes(): self.assertIsNone(node.lock) self.assertEqual(node.state, model.STATE_USED)
def test_node_request_priority(self): # Test that requests are satisfied in priority order nodeset = model.NodeSet() nodeset.addNode(model.Node(['controller', 'foo'], 'ubuntu-xenial')) nodeset.addNode(model.Node(['compute'], 'ubuntu-xenial')) job = model.Job('testjob') job.nodeset = nodeset self.fake_nodepool.pause() request1 = self.nodepool.requestNodes(None, job, 1) request2 = self.nodepool.requestNodes(None, job, 0) self.fake_nodepool.unpause() self.waitForRequests() self.assertEqual(len(self.provisioned_requests), 2) self.assertEqual(request1.state, 'fulfilled') self.assertEqual(request2.state, 'fulfilled') self.assertTrue(request2.state_time < request1.state_time)
def test_invalid_node_request(self): # Test requests with an invalid node type fail nodeset = model.NodeSet() nodeset.addNode(model.Node(['controller'], 'invalid-label')) job = model.Job('testjob') job.nodeset = nodeset request = self.nodepool.requestNodes(None, job, 0) self.waitForRequests() self.assertEqual(len(self.provisioned_requests), 1) self.assertEqual(request.state, model.STATE_FAILED)
def test_accept_nodes_resubmitted(self): # Test that a resubmitted request would not lock nodes nodeset = model.NodeSet() nodeset.addNode(model.Node(['controller'], 'ubuntu-xenial')) nodeset.addNode(model.Node(['compute'], 'ubuntu-xenial')) job = model.Job('testjob') job.nodeset = nodeset request = self.nodepool.requestNodes(None, job, 0) self.waitForRequests() self.assertEqual(len(self.provisioned_requests), 1) self.assertEqual(request.state, 'fulfilled') # Accept the nodes, passing a different ID self.nodepool.acceptNodes(request, "invalid") nodeset = request.nodeset for node in nodeset.getNodes(): self.assertIsNone(node.lock) self.assertEqual(node.state, 'ready')
def test_accept_nodes_lost_request(self): # Test that a lost request would not lock nodes nodeset = model.NodeSet() nodeset.addNode(model.Node(['controller'], 'ubuntu-xenial')) nodeset.addNode(model.Node(['compute'], 'ubuntu-xenial')) job = model.Job('testjob') job.nodeset = nodeset request = self.nodepool.requestNodes(None, job, 0) self.waitForRequests() self.assertEqual(len(self.provisioned_requests), 1) self.assertEqual(request.state, 'fulfilled') self.zk.deleteNodeRequest(request) # Accept the nodes self.nodepool.acceptNodes(request, request.id) nodeset = request.nodeset for node in nodeset.getNodes(): self.assertIsNone(node.lock) self.assertEqual(node.state, 'ready')