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"]')