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}