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}