def test_lb_get_index(self, mock_api, mock_extra): mock_api.return_value = self.dict_list mock_extra.return_value = {'id': 1, 'virtualIps': 1} res = api.lb_get_index(self.conf, self.tenant_id) self.assertTrue(mock_api.called) self.assertTrue(mock_extra.called) self.assertEquals(res[0], {'id': 1})
def test_lb_get_index(self, mock_api, mock_extra): mock_api.return_value = [{'id': 1, 'virtualIps': 'foo'}, {'id': 2}] mock_extra.return_value = 'foo' resp = api.lb_get_index(self.conf, self.tenant_id) self.assertTrue(mock_api.called) self.assertTrue(mock_extra.called) self.assertTrue(mock_extra.call_count == 2) self.assertEqual(resp, ['foo', 'foo']) mock_api.assert_called_once_with(self.conf, self.tenant_id) mock_extra.assert_any_call({'id': 1, 'virtualIps': 'foo'}) mock_extra.assert_any_call({'id': 2})
def index(self, req): logger.debug("Got index request. Request: %s", req) tenant_id = req.headers.get('X-Tenant-Id', "") result = core_api.lb_get_index(self.conf, tenant_id) return {'loadbalancers': result}
def test_lb_get_index(self, mock_api): api.lb_get_index(self.conf, self.tenant_id) self.assertTrue(mock_api.called)
def index(self, req, tenant_id): LOG.debug("Got index request. Request: %s", req) result = core_api.lb_get_index(self.conf, tenant_id) return {'loadbalancers': result}