def findLBforVM(self, req, **args):
     logger.debug("Got index request. Request: %s", req)
     tenant_id = req.headers.get('X-Tenant-Id', "")
     params = {}
     params['vm_id'] = args['vm_id']
     params['tenant_id'] = tenant_id
     result = core_api.lb_find_for_vm(self.conf, **params)
     return {'loadbalancers': result}
 def test_lb_find_for_vm(self, mock_api, mock_extra):
     vm_id = mock.Mock()
     mock_api.return_value = ['foo']
     mock_extra.return_value = 'foo'
     resp = api.lb_find_for_vm(self.conf, vm_id, self.tenant_id)
     mock_api.assert_called_once_with(self.conf, vm_id, self.tenant_id)
     mock_extra.assert_called_once_with('foo')
     self.assertTrue(mock_api.called)
     self.assertEqual(resp, ['foo'])
 def test_lb_find_for_vm(self, mock_api):
     vm_id = mock.Mock()
     api.lb_find_for_vm(self.conf, vm_id, self.tenant_id)
     self.assertTrue(mock_api.called)
 def findLBforVM(self, req, tenant_id, vm_id):
     LOG.debug("Got index request. Request: %s", req)
     result = core_api.lb_find_for_vm(self.conf, tenant_id, vm_id)
     return {'loadbalancers': result}