91 lines
3.5 KiB
Python
91 lines
3.5 KiB
Python
# Copyright 2014 Intel 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 import objects
|
|
from nova import test
|
|
from nova.tests.unit.api.openstack.compute import test_hypervisors
|
|
from nova.tests.unit.api.openstack import fakes
|
|
|
|
TEST_HYPER = test_hypervisors.TEST_HYPERS_OBJ[0].obj_clone()
|
|
TEST_SERVICE = objects.Service(id=1,
|
|
host="compute1",
|
|
binary="nova-compute",
|
|
topic="compute_topic",
|
|
report_count=5,
|
|
disabled=False,
|
|
disabled_reason=None,
|
|
availability_zone="nova")
|
|
|
|
|
|
class HypervisorStatusTestV21(test.NoDBTestCase):
|
|
def _prepare_extension(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 test_view_hypervisor_service_status(self):
|
|
self._prepare_extension()
|
|
req = self._get_request()
|
|
|
|
result = self.controller._view_hypervisor(
|
|
TEST_HYPER, TEST_SERVICE, False, req)
|
|
self.assertEqual('enabled', result['status'])
|
|
self.assertEqual('up', result['state'])
|
|
self.assertEqual('enabled', result['status'])
|
|
|
|
self.controller.servicegroup_api.service_is_up.return_value = False
|
|
result = self.controller._view_hypervisor(
|
|
TEST_HYPER, TEST_SERVICE, False, req)
|
|
self.assertEqual('down', result['state'])
|
|
|
|
hyper = copy.deepcopy(TEST_HYPER)
|
|
service = copy.deepcopy(TEST_SERVICE)
|
|
service.disabled = True
|
|
result = self.controller._view_hypervisor(hyper, service, False, req)
|
|
self.assertEqual('disabled', result['status'])
|
|
|
|
def test_view_hypervisor_detail_status(self):
|
|
self._prepare_extension()
|
|
req = self._get_request()
|
|
|
|
result = self.controller._view_hypervisor(
|
|
TEST_HYPER, TEST_SERVICE, True, req)
|
|
|
|
self.assertEqual('enabled', result['status'])
|
|
self.assertEqual('up', result['state'])
|
|
self.assertIsNone(result['service']['disabled_reason'])
|
|
|
|
self.controller.servicegroup_api.service_is_up.return_value = False
|
|
result = self.controller._view_hypervisor(
|
|
TEST_HYPER, TEST_SERVICE, True, req)
|
|
self.assertEqual('down', result['state'])
|
|
|
|
hyper = copy.deepcopy(TEST_HYPER)
|
|
service = copy.deepcopy(TEST_SERVICE)
|
|
service.disabled = True
|
|
service.disabled_reason = "fake"
|
|
result = self.controller._view_hypervisor(hyper, service, True, req)
|
|
self.assertEqual('disabled', result['status'],)
|
|
self.assertEqual('fake', result['service']['disabled_reason'])
|