def setUp(self):
self.request = mock.MagicMock(autospec=Request)
self.request.uri = "http://www.example.com/test"
self.request.path = self.path
self.request.requestHeaders = mock.MagicMock()
self.request.requestHeaders.hasHeader.return_value = True
self.request.requestHeaders.getRawHeaders.return_value = 'application/json'
self.coordinators = []
for i in xrange(3):
lvsservice = mock.MagicMock()
lvsservice.name = 'test_pool%d' % i
coord = mock.MagicMock()
coord.servers = {}
coord.lvsservice = lvsservice
for j in xrange(1, 11):
host = "mw10%02d" % j
ip = '192.168.10.%d' % j
port = 80
weight = 10
s = ServerStub(host, ip=ip, port=port, weight=weight,
lvsservice=lvsservice)
s.pool = True
s.up = True
coord.servers[host] = s
self.coordinators.append(coord)
PoolsRoot.addPool('test_pool0', self.coordinators[0])
def test_getChild(self):
"""Test case for `PoolsRoot.getChild`"""
r = PoolsRoot()
self.assertEquals(r, r.getChild("", self.request))
res = r.getChild('test_pool0', self.request)
self.assertIsInstance(res, PoolServers)
self.assertEquals(res.coordinator, self.coordinators[0])
notf = r.getChild('not_test', self.request)
self.assertIsInstance(notf, Resp404)
def test_render(self):
"""Test case for `PoolsRoot.render_GET`"""
r = PoolsRoot()
self.assertEquals(r.render_GET(self.request), '["test_pool0"]')
