118 lines
4.4 KiB
Python
118 lines
4.4 KiB
Python
# Copyright 2014 IBM Corp.
|
|
#
|
|
# Licensed under the Apache License, Version 2.0 (the "License"); you may
|
|
# not use this file except in compliance with the License. You may obtain
|
|
# a copy of the License at
|
|
#
|
|
# http://www.apache.org/licenses/LICENSE-2.0
|
|
#
|
|
# Unless required by applicable law or agreed to in writing, software
|
|
# distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
|
|
# WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
|
|
# License for the specific language governing permissions and limitations
|
|
# under the License.
|
|
|
|
import copy
|
|
|
|
import mock
|
|
|
|
from nova.api.openstack.compute import hypervisors \
|
|
as hypervisors_v21
|
|
from nova.api.openstack.compute.legacy_v2.contrib import hypervisors \
|
|
as hypervisors_v2
|
|
from nova.api.openstack import extensions
|
|
from nova import exception
|
|
from nova import objects
|
|
from nova import test
|
|
from nova.tests.unit.api.openstack.compute import test_hypervisors
|
|
from nova.tests.unit.api.openstack import fakes
|
|
|
|
|
|
def fake_compute_node_get(context, compute_id):
|
|
for hyper in test_hypervisors.TEST_HYPERS_OBJ:
|
|
if hyper.id == int(compute_id):
|
|
return hyper
|
|
raise exception.ComputeHostNotFound(host=compute_id)
|
|
|
|
|
|
def fake_compute_node_get_all(context):
|
|
return test_hypervisors.TEST_HYPERS_OBJ
|
|
|
|
|
|
@classmethod
|
|
def fake_service_get_by_compute_host(cls, context, host):
|
|
for service in test_hypervisors.TEST_SERVICES:
|
|
if service.host == host:
|
|
return service
|
|
|
|
|
|
class ExtendedHypervisorsTestV21(test.NoDBTestCase):
|
|
DETAIL_HYPERS_DICTS = copy.deepcopy(test_hypervisors.TEST_HYPERS)
|
|
del DETAIL_HYPERS_DICTS[0]['service_id']
|
|
del DETAIL_HYPERS_DICTS[1]['service_id']
|
|
del DETAIL_HYPERS_DICTS[0]['host']
|
|
del DETAIL_HYPERS_DICTS[1]['host']
|
|
DETAIL_HYPERS_DICTS[0].update({'state': 'up',
|
|
'status': 'enabled',
|
|
'service': dict(id=1, host='compute1',
|
|
disabled_reason=None)})
|
|
DETAIL_HYPERS_DICTS[1].update({'state': 'up',
|
|
'status': 'enabled',
|
|
'service': dict(id=2, host='compute2',
|
|
disabled_reason=None)})
|
|
|
|
def _set_up_controller(self):
|
|
self.controller = hypervisors_v21.HypervisorsController()
|
|
self.controller.servicegroup_api.service_is_up = mock.MagicMock(
|
|
return_value=True)
|
|
|
|
def _get_request(self):
|
|
return fakes.HTTPRequest.blank('/v2/fake/os-hypervisors/detail',
|
|
use_admin_context=True)
|
|
|
|
def setUp(self):
|
|
super(ExtendedHypervisorsTestV21, self).setUp()
|
|
self._set_up_controller()
|
|
|
|
self.stubs.Set(self.controller.host_api, 'compute_node_get_all',
|
|
fake_compute_node_get_all)
|
|
self.stubs.Set(self.controller.host_api, 'compute_node_get',
|
|
fake_compute_node_get)
|
|
self.stubs.Set(objects.Service, 'get_by_compute_host',
|
|
fake_service_get_by_compute_host)
|
|
|
|
def test_view_hypervisor_detail_noservers(self):
|
|
result = self.controller._view_hypervisor(
|
|
test_hypervisors.TEST_HYPERS_OBJ[0],
|
|
test_hypervisors.TEST_SERVICES[0], True)
|
|
|
|
self.assertEqual(result, self.DETAIL_HYPERS_DICTS[0])
|
|
|
|
def test_detail(self):
|
|
req = self._get_request()
|
|
result = self.controller.detail(req)
|
|
|
|
self.assertEqual(result, dict(hypervisors=self.DETAIL_HYPERS_DICTS))
|
|
|
|
def test_show_withid(self):
|
|
req = self._get_request()
|
|
result = self.controller.show(req, '1')
|
|
|
|
self.assertEqual(result, dict(hypervisor=self.DETAIL_HYPERS_DICTS[0]))
|
|
|
|
|
|
class ExtendedHypervisorsTestV2(ExtendedHypervisorsTestV21):
|
|
DETAIL_HYPERS_DICTS = copy.deepcopy(test_hypervisors.TEST_HYPERS)
|
|
del DETAIL_HYPERS_DICTS[0]['service_id']
|
|
del DETAIL_HYPERS_DICTS[1]['service_id']
|
|
del DETAIL_HYPERS_DICTS[0]['host']
|
|
del DETAIL_HYPERS_DICTS[1]['host']
|
|
DETAIL_HYPERS_DICTS[0].update({'service': dict(id=1, host='compute1')})
|
|
DETAIL_HYPERS_DICTS[1].update({'service': dict(id=2, host='compute2')})
|
|
|
|
def _set_up_controller(self):
|
|
self.ext_mgr = extensions.ExtensionManager()
|
|
self.ext_mgr.extensions = {}
|
|
self.ext_mgr.extensions['os-extended-hypervisors'] = True
|
|
self.controller = hypervisors_v2.HypervisorsController(self.ext_mgr)
|