blob: 6a141920e7c6c02bc3f95c1331bb1e7bd1c9bbf6 [file] [log] [blame]
James E. Blair3897a132016-12-22 18:23:42 -08001# Copyright 2015 Red Hat, Inc.
2#
3# Licensed under the Apache License, Version 2.0 (the "License"); you may
4# not use this file except in compliance with the License. You may obtain
5# a copy of the License at
6#
7# http://www.apache.org/licenses/LICENSE-2.0
8#
9# Unless required by applicable law or agreed to in writing, software
10# distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
11# WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
12# License for the specific language governing permissions and limitations
13# under the License.
14
15
16import time
17
18import zuul.zk
19import zuul.nodepool
20from zuul import model
21
22from tests.base import BaseTestCase, ChrootedKazooFixture, FakeNodepool
23
24
25class TestNodepool(BaseTestCase):
26 # Tests the Nodepool interface class using a fake nodepool and
27 # scheduler.
28
29 def setUp(self):
30 super(BaseTestCase, self).setUp()
31
32 self.zk_chroot_fixture = self.useFixture(ChrootedKazooFixture())
33 self.zk_config = zuul.zk.ZooKeeperConnectionConfig(
34 self.zk_chroot_fixture.zookeeper_host,
35 self.zk_chroot_fixture.zookeeper_port,
36 self.zk_chroot_fixture.zookeeper_chroot)
37
38 self.zk = zuul.zk.ZooKeeper()
39 self.zk.connect([self.zk_config])
40
41 self.provisioned_requests = []
42 # This class implements the scheduler methods zuul.nodepool
43 # needs, so we pass 'self' as the scheduler.
44 self.nodepool = zuul.nodepool.Nodepool(self)
45
46 self.fake_nodepool = FakeNodepool(self.zk_config.host,
47 self.zk_config.port,
48 self.zk_config.chroot)
49
50 def waitForRequests(self):
51 # Wait until all requests are complete.
52 while self.nodepool.requests:
53 time.sleep(0.1)
54
55 def onNodesProvisioned(self, request):
56 # This is a scheduler method that the nodepool class calls
57 # back when a request is provisioned.
58 self.provisioned_requests.append(request)
59
60 def test_node_request(self):
61 # Test a simple node request
62
63 nodeset = model.NodeSet()
64 nodeset.addNode(model.Node('controller', 'ubuntu-xenial'))
65 nodeset.addNode(model.Node('compute', 'ubuntu-xenial'))
66 job = model.Job('testjob')
67 job.nodeset = nodeset
68 request = self.nodepool.requestNodes(None, job)
69 self.waitForRequests()
70 self.assertEqual(len(self.provisioned_requests), 1)
71 self.assertEqual(request.state, 'fulfilled')