577 lines
22 KiB
Python
577 lines
22 KiB
Python
# Copyright 2012 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 calendar
|
|
import datetime
|
|
|
|
import iso8601
|
|
import mock
|
|
from oslo.utils import timeutils
|
|
import webob.exc
|
|
|
|
from nova.api.openstack.compute.contrib import services
|
|
from nova.api.openstack import extensions
|
|
from nova import availability_zones
|
|
from nova.compute import cells_api
|
|
from nova import context
|
|
from nova import db
|
|
from nova import exception
|
|
from nova.servicegroup.drivers import db as db_driver
|
|
from nova import test
|
|
from nova.tests.unit.api.openstack import fakes
|
|
from nova.tests.unit.objects import test_service
|
|
|
|
|
|
fake_services_list = [
|
|
dict(test_service.fake_service,
|
|
binary='nova-scheduler',
|
|
host='host1',
|
|
id=1,
|
|
disabled=True,
|
|
topic='scheduler',
|
|
updated_at=datetime.datetime(2012, 10, 29, 13, 42, 2),
|
|
created_at=datetime.datetime(2012, 9, 18, 2, 46, 27),
|
|
disabled_reason='test1'),
|
|
dict(test_service.fake_service,
|
|
binary='nova-compute',
|
|
host='host1',
|
|
id=2,
|
|
disabled=True,
|
|
topic='compute',
|
|
updated_at=datetime.datetime(2012, 10, 29, 13, 42, 5),
|
|
created_at=datetime.datetime(2012, 9, 18, 2, 46, 27),
|
|
disabled_reason='test2'),
|
|
dict(test_service.fake_service,
|
|
binary='nova-scheduler',
|
|
host='host2',
|
|
id=3,
|
|
disabled=False,
|
|
topic='scheduler',
|
|
updated_at=datetime.datetime(2012, 9, 19, 6, 55, 34),
|
|
created_at=datetime.datetime(2012, 9, 18, 2, 46, 28),
|
|
disabled_reason=None),
|
|
dict(test_service.fake_service,
|
|
binary='nova-compute',
|
|
host='host2',
|
|
id=4,
|
|
disabled=True,
|
|
topic='compute',
|
|
updated_at=datetime.datetime(2012, 9, 18, 8, 3, 38),
|
|
created_at=datetime.datetime(2012, 9, 18, 2, 46, 28),
|
|
disabled_reason='test4'),
|
|
]
|
|
|
|
|
|
class FakeRequest(object):
|
|
environ = {"nova.context": context.get_admin_context()}
|
|
GET = {}
|
|
|
|
|
|
class FakeRequestWithService(object):
|
|
environ = {"nova.context": context.get_admin_context()}
|
|
GET = {"binary": "nova-compute"}
|
|
|
|
|
|
class FakeRequestWithHost(object):
|
|
environ = {"nova.context": context.get_admin_context()}
|
|
GET = {"host": "host1"}
|
|
|
|
|
|
class FakeRequestWithHostService(object):
|
|
environ = {"nova.context": context.get_admin_context()}
|
|
GET = {"host": "host1", "binary": "nova-compute"}
|
|
|
|
|
|
def fake_service_get_all(services):
|
|
def service_get_all(context, filters=None, set_zones=False):
|
|
if set_zones or 'availability_zone' in filters:
|
|
return availability_zones.set_availability_zones(context,
|
|
services)
|
|
return services
|
|
return service_get_all
|
|
|
|
|
|
def fake_db_api_service_get_all(context, disabled=None):
|
|
return fake_services_list
|
|
|
|
|
|
def fake_db_service_get_by_host_binary(services):
|
|
def service_get_by_host_binary(context, host, binary):
|
|
for service in services:
|
|
if service['host'] == host and service['binary'] == binary:
|
|
return service
|
|
raise exception.HostBinaryNotFound(host=host, binary=binary)
|
|
return service_get_by_host_binary
|
|
|
|
|
|
def fake_service_get_by_host_binary(context, host, binary):
|
|
fake = fake_db_service_get_by_host_binary(fake_services_list)
|
|
return fake(context, host, binary)
|
|
|
|
|
|
def _service_get_by_id(services, value):
|
|
for service in services:
|
|
if service['id'] == value:
|
|
return service
|
|
return None
|
|
|
|
|
|
def fake_db_service_update(services):
|
|
def service_update(context, service_id, values):
|
|
service = _service_get_by_id(services, service_id)
|
|
if service is None:
|
|
raise exception.ServiceNotFound(service_id=service_id)
|
|
return service
|
|
return service_update
|
|
|
|
|
|
def fake_service_update(context, service_id, values):
|
|
fake = fake_db_service_update(fake_services_list)
|
|
return fake(context, service_id, values)
|
|
|
|
|
|
def fake_utcnow():
|
|
return datetime.datetime(2012, 10, 29, 13, 42, 11)
|
|
|
|
|
|
fake_utcnow.override_time = None
|
|
|
|
|
|
def fake_utcnow_ts():
|
|
d = fake_utcnow()
|
|
return calendar.timegm(d.utctimetuple())
|
|
|
|
|
|
class ServicesTest(test.TestCase):
|
|
|
|
def setUp(self):
|
|
super(ServicesTest, self).setUp()
|
|
|
|
self.ext_mgr = extensions.ExtensionManager()
|
|
self.ext_mgr.extensions = {}
|
|
self.controller = services.ServiceController(self.ext_mgr)
|
|
|
|
self.stubs.Set(timeutils, "utcnow", fake_utcnow)
|
|
self.stubs.Set(timeutils, "utcnow_ts", fake_utcnow_ts)
|
|
|
|
self.stubs.Set(self.controller.host_api, "service_get_all",
|
|
fake_service_get_all(fake_services_list))
|
|
|
|
self.stubs.Set(db, "service_get_by_args",
|
|
fake_db_service_get_by_host_binary(fake_services_list))
|
|
self.stubs.Set(db, "service_update",
|
|
fake_db_service_update(fake_services_list))
|
|
|
|
def test_services_list(self):
|
|
req = FakeRequest()
|
|
res_dict = self.controller.index(req)
|
|
|
|
response = {'services': [
|
|
{'binary': 'nova-scheduler',
|
|
'host': 'host1',
|
|
'zone': 'internal',
|
|
'status': 'disabled',
|
|
'state': 'up',
|
|
'updated_at': datetime.datetime(2012, 10, 29, 13, 42, 2)},
|
|
{'binary': 'nova-compute',
|
|
'host': 'host1',
|
|
'zone': 'nova',
|
|
'status': 'disabled',
|
|
'state': 'up',
|
|
'updated_at': datetime.datetime(2012, 10, 29, 13, 42, 5)},
|
|
{'binary': 'nova-scheduler',
|
|
'host': 'host2',
|
|
'zone': 'internal',
|
|
'status': 'enabled',
|
|
'state': 'down',
|
|
'updated_at': datetime.datetime(2012, 9, 19, 6, 55, 34)},
|
|
{'binary': 'nova-compute',
|
|
'host': 'host2',
|
|
'zone': 'nova',
|
|
'status': 'disabled',
|
|
'state': 'down',
|
|
'updated_at': datetime.datetime(2012, 9, 18, 8, 3, 38)}]}
|
|
self.assertEqual(res_dict, response)
|
|
|
|
def test_services_list_with_host(self):
|
|
req = FakeRequestWithHost()
|
|
res_dict = self.controller.index(req)
|
|
|
|
response = {'services': [
|
|
{'binary': 'nova-scheduler',
|
|
'host': 'host1',
|
|
'zone': 'internal',
|
|
'status': 'disabled',
|
|
'state': 'up',
|
|
'updated_at': datetime.datetime(2012, 10, 29, 13, 42, 2)},
|
|
{'binary': 'nova-compute',
|
|
'host': 'host1',
|
|
'zone': 'nova',
|
|
'status': 'disabled',
|
|
'state': 'up',
|
|
'updated_at': datetime.datetime(2012, 10, 29, 13, 42, 5)}]}
|
|
self.assertEqual(res_dict, response)
|
|
|
|
def test_services_list_with_service(self):
|
|
req = FakeRequestWithService()
|
|
res_dict = self.controller.index(req)
|
|
|
|
response = {'services': [
|
|
{'binary': 'nova-compute',
|
|
'host': 'host1',
|
|
'zone': 'nova',
|
|
'status': 'disabled',
|
|
'state': 'up',
|
|
'updated_at': datetime.datetime(2012, 10, 29, 13, 42, 5)},
|
|
{'binary': 'nova-compute',
|
|
'host': 'host2',
|
|
'zone': 'nova',
|
|
'status': 'disabled',
|
|
'state': 'down',
|
|
'updated_at': datetime.datetime(2012, 9, 18, 8, 3, 38)}]}
|
|
self.assertEqual(res_dict, response)
|
|
|
|
def test_services_list_with_host_service(self):
|
|
req = FakeRequestWithHostService()
|
|
res_dict = self.controller.index(req)
|
|
|
|
response = {'services': [
|
|
{'binary': 'nova-compute',
|
|
'host': 'host1',
|
|
'zone': 'nova',
|
|
'status': 'disabled',
|
|
'state': 'up',
|
|
'updated_at': datetime.datetime(2012, 10, 29, 13, 42, 5)}]}
|
|
self.assertEqual(res_dict, response)
|
|
|
|
def test_services_detail(self):
|
|
self.ext_mgr.extensions['os-extended-services'] = True
|
|
req = FakeRequest()
|
|
res_dict = self.controller.index(req)
|
|
response = {'services': [
|
|
{'binary': 'nova-scheduler',
|
|
'host': 'host1',
|
|
'zone': 'internal',
|
|
'status': 'disabled',
|
|
'state': 'up',
|
|
'updated_at': datetime.datetime(2012, 10, 29, 13, 42, 2),
|
|
'disabled_reason': 'test1'},
|
|
{'binary': 'nova-compute',
|
|
'host': 'host1',
|
|
'zone': 'nova',
|
|
'status': 'disabled',
|
|
'state': 'up',
|
|
'updated_at': datetime.datetime(2012, 10, 29, 13, 42, 5),
|
|
'disabled_reason': 'test2'},
|
|
{'binary': 'nova-scheduler',
|
|
'host': 'host2',
|
|
'zone': 'internal',
|
|
'status': 'enabled',
|
|
'state': 'down',
|
|
'updated_at': datetime.datetime(2012, 9, 19, 6, 55, 34),
|
|
'disabled_reason': None},
|
|
{'binary': 'nova-compute',
|
|
'host': 'host2',
|
|
'zone': 'nova',
|
|
'status': 'disabled',
|
|
'state': 'down',
|
|
'updated_at': datetime.datetime(2012, 9, 18, 8, 3, 38),
|
|
'disabled_reason': 'test4'}]}
|
|
self.assertEqual(res_dict, response)
|
|
|
|
def test_service_detail_with_host(self):
|
|
self.ext_mgr.extensions['os-extended-services'] = True
|
|
req = FakeRequestWithHost()
|
|
res_dict = self.controller.index(req)
|
|
response = {'services': [
|
|
{'binary': 'nova-scheduler',
|
|
'host': 'host1',
|
|
'zone': 'internal',
|
|
'status': 'disabled',
|
|
'state': 'up',
|
|
'updated_at': datetime.datetime(2012, 10, 29, 13, 42, 2),
|
|
'disabled_reason': 'test1'},
|
|
{'binary': 'nova-compute',
|
|
'host': 'host1',
|
|
'zone': 'nova',
|
|
'status': 'disabled',
|
|
'state': 'up',
|
|
'updated_at': datetime.datetime(2012, 10, 29, 13, 42, 5),
|
|
'disabled_reason': 'test2'}]}
|
|
self.assertEqual(res_dict, response)
|
|
|
|
def test_service_detail_with_service(self):
|
|
self.ext_mgr.extensions['os-extended-services'] = True
|
|
req = FakeRequestWithService()
|
|
res_dict = self.controller.index(req)
|
|
response = {'services': [
|
|
{'binary': 'nova-compute',
|
|
'host': 'host1',
|
|
'zone': 'nova',
|
|
'status': 'disabled',
|
|
'state': 'up',
|
|
'updated_at': datetime.datetime(2012, 10, 29, 13, 42, 5),
|
|
'disabled_reason': 'test2'},
|
|
{'binary': 'nova-compute',
|
|
'host': 'host2',
|
|
'zone': 'nova',
|
|
'status': 'disabled',
|
|
'state': 'down',
|
|
'updated_at': datetime.datetime(2012, 9, 18, 8, 3, 38),
|
|
'disabled_reason': 'test4'}]}
|
|
self.assertEqual(res_dict, response)
|
|
|
|
def test_service_detail_with_host_service(self):
|
|
self.ext_mgr.extensions['os-extended-services'] = True
|
|
req = FakeRequestWithHostService()
|
|
res_dict = self.controller.index(req)
|
|
response = {'services': [
|
|
{'binary': 'nova-compute',
|
|
'host': 'host1',
|
|
'zone': 'nova',
|
|
'status': 'disabled',
|
|
'state': 'up',
|
|
'updated_at': datetime.datetime(2012, 10, 29, 13, 42, 5),
|
|
'disabled_reason': 'test2'}]}
|
|
self.assertEqual(res_dict, response)
|
|
|
|
def test_services_detail_with_delete_extension(self):
|
|
self.ext_mgr.extensions['os-extended-services-delete'] = True
|
|
req = FakeRequest()
|
|
res_dict = self.controller.index(req)
|
|
response = {'services': [
|
|
{'binary': 'nova-scheduler',
|
|
'host': 'host1',
|
|
'id': 1,
|
|
'zone': 'internal',
|
|
'status': 'disabled',
|
|
'state': 'up',
|
|
'updated_at': datetime.datetime(2012, 10, 29, 13, 42, 2)},
|
|
{'binary': 'nova-compute',
|
|
'host': 'host1',
|
|
'id': 2,
|
|
'zone': 'nova',
|
|
'status': 'disabled',
|
|
'state': 'up',
|
|
'updated_at': datetime.datetime(2012, 10, 29, 13, 42, 5)},
|
|
{'binary': 'nova-scheduler',
|
|
'host': 'host2',
|
|
'id': 3,
|
|
'zone': 'internal',
|
|
'status': 'enabled',
|
|
'state': 'down',
|
|
'updated_at': datetime.datetime(2012, 9, 19, 6, 55, 34)},
|
|
{'binary': 'nova-compute',
|
|
'host': 'host2',
|
|
'id': 4,
|
|
'zone': 'nova',
|
|
'status': 'disabled',
|
|
'state': 'down',
|
|
'updated_at': datetime.datetime(2012, 9, 18, 8, 3, 38)}]}
|
|
self.assertEqual(res_dict, response)
|
|
|
|
def test_services_enable(self):
|
|
def _service_update(context, service_id, values):
|
|
self.assertIsNone(values['disabled_reason'])
|
|
return dict(test_service.fake_service, id=service_id, **values)
|
|
|
|
self.stubs.Set(db, "service_update", _service_update)
|
|
|
|
body = {'host': 'host1', 'binary': 'nova-compute'}
|
|
req = fakes.HTTPRequest.blank('/v2/fake/os-services/enable')
|
|
|
|
res_dict = self.controller.update(req, "enable", body)
|
|
self.assertEqual(res_dict['service']['status'], 'enabled')
|
|
self.assertNotIn('disabled_reason', res_dict['service'])
|
|
|
|
def test_services_enable_with_invalid_host(self):
|
|
body = {'host': 'invalid', 'binary': 'nova-compute'}
|
|
req = fakes.HTTPRequest.blank('/v2/fake/os-services/enable')
|
|
self.assertRaises(webob.exc.HTTPNotFound,
|
|
self.controller.update,
|
|
req,
|
|
"enable",
|
|
body)
|
|
|
|
def test_services_enable_with_invalid_binary(self):
|
|
body = {'host': 'host1', 'binary': 'invalid'}
|
|
req = fakes.HTTPRequest.blank('/v2/fake/os-services/enable')
|
|
self.assertRaises(webob.exc.HTTPNotFound,
|
|
self.controller.update,
|
|
req,
|
|
"enable",
|
|
body)
|
|
|
|
# This test is just to verify that the servicegroup API gets used when
|
|
# calling this API.
|
|
def test_services_with_exception(self):
|
|
def dummy_is_up(self, dummy):
|
|
raise KeyError()
|
|
|
|
self.stubs.Set(db_driver.DbDriver, 'is_up', dummy_is_up)
|
|
req = FakeRequestWithHostService()
|
|
self.assertRaises(KeyError, self.controller.index, req)
|
|
|
|
def test_services_disable(self):
|
|
req = fakes.HTTPRequest.blank('/v2/fake/os-services/disable')
|
|
body = {'host': 'host1', 'binary': 'nova-compute'}
|
|
res_dict = self.controller.update(req, "disable", body)
|
|
|
|
self.assertEqual(res_dict['service']['status'], 'disabled')
|
|
self.assertNotIn('disabled_reason', res_dict['service'])
|
|
|
|
def test_services_disable_with_invalid_host(self):
|
|
body = {'host': 'invalid', 'binary': 'nova-compute'}
|
|
req = fakes.HTTPRequest.blank('/v2/fake/os-services/disable')
|
|
self.assertRaises(webob.exc.HTTPNotFound,
|
|
self.controller.update,
|
|
req,
|
|
"disable",
|
|
body)
|
|
|
|
def test_services_disable_with_invalid_binary(self):
|
|
body = {'host': 'host1', 'binary': 'invalid'}
|
|
req = fakes.HTTPRequestV3.blank('/v2/fake/os-services/disable')
|
|
self.assertRaises(webob.exc.HTTPNotFound,
|
|
self.controller.update,
|
|
req,
|
|
"disable",
|
|
body)
|
|
|
|
def test_services_disable_log_reason(self):
|
|
self.ext_mgr.extensions['os-extended-services'] = True
|
|
req = \
|
|
fakes.HTTPRequest.blank('v2/fakes/os-services/disable-log-reason')
|
|
body = {'host': 'host1',
|
|
'binary': 'nova-compute',
|
|
'disabled_reason': 'test-reason',
|
|
}
|
|
res_dict = self.controller.update(req, "disable-log-reason", body)
|
|
|
|
self.assertEqual(res_dict['service']['status'], 'disabled')
|
|
self.assertEqual(res_dict['service']['disabled_reason'], 'test-reason')
|
|
|
|
def test_mandatory_reason_field(self):
|
|
self.ext_mgr.extensions['os-extended-services'] = True
|
|
req = \
|
|
fakes.HTTPRequest.blank('v2/fakes/os-services/disable-log-reason')
|
|
body = {'host': 'host1',
|
|
'binary': 'nova-compute',
|
|
}
|
|
self.assertRaises(webob.exc.HTTPBadRequest,
|
|
self.controller.update, req, "disable-log-reason", body)
|
|
|
|
def test_invalid_reason_field(self):
|
|
reason = ' '
|
|
self.assertFalse(self.controller._is_valid_as_reason(reason))
|
|
reason = 'a' * 256
|
|
self.assertFalse(self.controller._is_valid_as_reason(reason))
|
|
reason = 'it\'s a valid reason.'
|
|
self.assertTrue(self.controller._is_valid_as_reason(reason))
|
|
|
|
def test_services_delete(self):
|
|
self.ext_mgr.extensions['os-extended-services-delete'] = True
|
|
|
|
request = fakes.HTTPRequest.blank('/v2/fakes/os-services/1',
|
|
use_admin_context=True)
|
|
request.method = 'DELETE'
|
|
|
|
with mock.patch.object(self.controller.host_api,
|
|
'service_delete') as service_delete:
|
|
self.controller.delete(request, '1')
|
|
service_delete.assert_called_once_with(
|
|
request.environ['nova.context'], '1')
|
|
self.assertEqual(self.controller.delete.wsgi_code, 204)
|
|
|
|
def test_services_delete_not_found(self):
|
|
self.ext_mgr.extensions['os-extended-services-delete'] = True
|
|
|
|
request = fakes.HTTPRequest.blank('/v2/fakes/os-services/abc',
|
|
use_admin_context=True)
|
|
request.method = 'DELETE'
|
|
self.assertRaises(webob.exc.HTTPNotFound,
|
|
self.controller.delete, request, 'abc')
|
|
|
|
def test_services_delete_not_enabled(self):
|
|
request = fakes.HTTPRequest.blank('/v2/fakes/os-services/300',
|
|
use_admin_context=True)
|
|
request.method = 'DELETE'
|
|
self.assertRaises(webob.exc.HTTPMethodNotAllowed,
|
|
self.controller.delete, request, '300')
|
|
|
|
|
|
class ServicesCellsTest(test.TestCase):
|
|
def setUp(self):
|
|
super(ServicesCellsTest, self).setUp()
|
|
|
|
host_api = cells_api.HostAPI()
|
|
|
|
self.ext_mgr = extensions.ExtensionManager()
|
|
self.ext_mgr.extensions = {}
|
|
self.controller = services.ServiceController(self.ext_mgr)
|
|
self.controller.host_api = host_api
|
|
|
|
self.stubs.Set(timeutils, "utcnow", fake_utcnow)
|
|
self.stubs.Set(timeutils, "utcnow_ts", fake_utcnow_ts)
|
|
|
|
services_list = []
|
|
for service in fake_services_list:
|
|
service = service.copy()
|
|
service['id'] = 'cell1@%d' % service['id']
|
|
services_list.append(service)
|
|
|
|
self.stubs.Set(host_api.cells_rpcapi, "service_get_all",
|
|
fake_service_get_all(services_list))
|
|
|
|
def test_services_detail(self):
|
|
self.ext_mgr.extensions['os-extended-services-delete'] = True
|
|
req = FakeRequest()
|
|
res_dict = self.controller.index(req)
|
|
utc = iso8601.iso8601.Utc()
|
|
response = {'services': [
|
|
{'id': 'cell1@1',
|
|
'binary': 'nova-scheduler',
|
|
'host': 'host1',
|
|
'zone': 'internal',
|
|
'status': 'disabled',
|
|
'state': 'up',
|
|
'updated_at': datetime.datetime(2012, 10, 29, 13, 42, 2,
|
|
tzinfo=utc)},
|
|
{'id': 'cell1@2',
|
|
'binary': 'nova-compute',
|
|
'host': 'host1',
|
|
'zone': 'nova',
|
|
'status': 'disabled',
|
|
'state': 'up',
|
|
'updated_at': datetime.datetime(2012, 10, 29, 13, 42, 5,
|
|
tzinfo=utc)},
|
|
{'id': 'cell1@3',
|
|
'binary': 'nova-scheduler',
|
|
'host': 'host2',
|
|
'zone': 'internal',
|
|
'status': 'enabled',
|
|
'state': 'down',
|
|
'updated_at': datetime.datetime(2012, 9, 19, 6, 55, 34,
|
|
tzinfo=utc)},
|
|
{'id': 'cell1@4',
|
|
'binary': 'nova-compute',
|
|
'host': 'host2',
|
|
'zone': 'nova',
|
|
'status': 'disabled',
|
|
'state': 'down',
|
|
'updated_at': datetime.datetime(2012, 9, 18, 8, 3, 38,
|
|
tzinfo=utc)}]}
|
|
self.assertEqual(res_dict, response)
|