307 lines
13 KiB
Python
307 lines
13 KiB
Python
# Copyright 2015 Intel, Inc
|
|
#
|
|
# 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 mock
|
|
|
|
from oslo_config import cfg
|
|
|
|
from magnum.common import context
|
|
from magnum.common.rpc_service import CONF
|
|
from magnum.db.sqlalchemy import api as dbapi
|
|
from magnum import objects
|
|
from magnum.objects.fields import BayStatus as bay_status
|
|
from magnum.service import periodic
|
|
from magnum.tests import base
|
|
from magnum.tests.unit.db import utils
|
|
|
|
periodic_opts = [
|
|
cfg.BoolOpt('periodic_enable',
|
|
default=True,
|
|
help='Enable periodic tasks.'),
|
|
cfg.IntOpt('periodic_interval_max',
|
|
default=60,
|
|
help='Max interval size between periodic tasks execution in '
|
|
'seconds.'),
|
|
]
|
|
|
|
|
|
class fake_stack(object):
|
|
def __init__(self, **kw):
|
|
for key, val in kw.items():
|
|
setattr(self, key, val)
|
|
|
|
|
|
class PeriodicTestCase(base.TestCase):
|
|
|
|
def setUp(self):
|
|
super(PeriodicTestCase, self).setUp()
|
|
|
|
ctx = context.make_admin_context()
|
|
|
|
bay1 = utils.get_test_bay(id=1, stack_id='11',
|
|
status=bay_status.CREATE_IN_PROGRESS)
|
|
bay2 = utils.get_test_bay(id=2, stack_id='22',
|
|
status=bay_status.DELETE_IN_PROGRESS)
|
|
bay3 = utils.get_test_bay(id=3, stack_id='33',
|
|
status=bay_status.UPDATE_IN_PROGRESS)
|
|
bay4 = utils.get_test_bay(id=4, stack_id='44',
|
|
status=bay_status.CREATE_COMPLETE)
|
|
|
|
self.bay1 = objects.Bay(ctx, **bay1)
|
|
self.bay2 = objects.Bay(ctx, **bay2)
|
|
self.bay3 = objects.Bay(ctx, **bay3)
|
|
self.bay4 = objects.Bay(ctx, **bay4)
|
|
|
|
mock_magnum_service_refresh = mock.Mock()
|
|
|
|
class FakeMS(object):
|
|
report_state_up = mock_magnum_service_refresh
|
|
|
|
self.fake_ms = FakeMS()
|
|
self.fake_ms_refresh = mock_magnum_service_refresh
|
|
|
|
@mock.patch.object(objects.Bay, 'list')
|
|
@mock.patch('magnum.common.clients.OpenStackClients')
|
|
@mock.patch.object(dbapi.Connection, 'destroy_bay')
|
|
@mock.patch.object(dbapi.Connection, 'update_bay')
|
|
def test_sync_bay_status_changes(self, mock_db_update, mock_db_destroy,
|
|
mock_oscc, mock_bay_list):
|
|
mock_heat_client = mock.MagicMock()
|
|
stack1 = fake_stack(id='11', stack_status=bay_status.CREATE_COMPLETE,
|
|
stack_status_reason='fake_reason_11')
|
|
stack3 = fake_stack(id='33', stack_status=bay_status.UPDATE_COMPLETE,
|
|
stack_status_reason='fake_reason_33')
|
|
mock_heat_client.stacks.list.return_value = [stack1, stack3]
|
|
mock_osc = mock_oscc.return_value
|
|
mock_osc.heat.return_value = mock_heat_client
|
|
mock_bay_list.return_value = [self.bay1, self.bay2, self.bay3]
|
|
|
|
mock_keystone_client = mock.MagicMock()
|
|
mock_keystone_client.client.project_id = "fake_project"
|
|
mock_osc.keystone.return_value = mock_keystone_client
|
|
|
|
periodic.MagnumPeriodicTasks(CONF,
|
|
'fake-conductor').sync_bay_status(None)
|
|
|
|
self.assertEqual(self.bay1.status, bay_status.CREATE_COMPLETE)
|
|
self.assertEqual(self.bay1.status_reason, 'fake_reason_11')
|
|
mock_db_destroy.assert_called_once_with(self.bay2.uuid)
|
|
self.assertEqual(self.bay3.status, bay_status.UPDATE_COMPLETE)
|
|
self.assertEqual(self.bay3.status_reason, 'fake_reason_33')
|
|
|
|
@mock.patch.object(objects.Bay, 'list')
|
|
@mock.patch('magnum.common.clients.OpenStackClients')
|
|
def test_sync_bay_status_not_changes(self, mock_oscc, mock_bay_list):
|
|
mock_heat_client = mock.MagicMock()
|
|
stack1 = fake_stack(id='11',
|
|
stack_status=bay_status.CREATE_IN_PROGRESS)
|
|
stack2 = fake_stack(id='22',
|
|
stack_status=bay_status.DELETE_IN_PROGRESS)
|
|
stack3 = fake_stack(id='33',
|
|
stack_status=bay_status.UPDATE_IN_PROGRESS)
|
|
mock_heat_client.stacks.list.return_value = [stack1, stack2, stack3]
|
|
mock_osc = mock_oscc.return_value
|
|
mock_osc.heat.return_value = mock_heat_client
|
|
mock_bay_list.return_value = [self.bay1, self.bay2, self.bay3]
|
|
periodic.MagnumPeriodicTasks(CONF,
|
|
'fake-conductor').sync_bay_status(None)
|
|
|
|
self.assertEqual(self.bay1.status, bay_status.CREATE_IN_PROGRESS)
|
|
self.assertEqual(self.bay2.status, bay_status.DELETE_IN_PROGRESS)
|
|
self.assertEqual(self.bay3.status, bay_status.UPDATE_IN_PROGRESS)
|
|
|
|
@mock.patch.object(objects.Bay, 'list')
|
|
@mock.patch('magnum.common.clients.OpenStackClients')
|
|
@mock.patch.object(dbapi.Connection, 'destroy_bay')
|
|
@mock.patch.object(dbapi.Connection, 'update_bay')
|
|
def test_sync_bay_status_heat_not_found(self, mock_db_update,
|
|
mock_db_destroy, mock_oscc,
|
|
mock_bay_list):
|
|
mock_heat_client = mock.MagicMock()
|
|
mock_heat_client.stacks.list.return_value = []
|
|
mock_osc = mock_oscc.return_value
|
|
mock_osc.heat.return_value = mock_heat_client
|
|
mock_bay_list.return_value = [self.bay1, self.bay2, self.bay3]
|
|
|
|
mock_keystone_client = mock.MagicMock()
|
|
mock_keystone_client.client.project_id = "fake_project"
|
|
mock_osc.keystone.return_value = mock_keystone_client
|
|
|
|
periodic.MagnumPeriodicTasks(CONF,
|
|
'fake-conductor').sync_bay_status(None)
|
|
|
|
self.assertEqual(self.bay1.status, bay_status.CREATE_FAILED)
|
|
self.assertEqual(self.bay1.status_reason, 'Stack with id 11 not '
|
|
'found in Heat.')
|
|
mock_db_destroy.assert_called_once_with(self.bay2.uuid)
|
|
self.assertEqual(self.bay3.status, bay_status.UPDATE_FAILED)
|
|
self.assertEqual(self.bay3.status_reason, 'Stack with id 33 not '
|
|
'found in Heat.')
|
|
|
|
@mock.patch.object(objects.MagnumService, 'get_by_host_and_binary')
|
|
@mock.patch.object(objects.MagnumService, 'create')
|
|
@mock.patch.object(objects.MagnumService, 'report_state_up')
|
|
def test_update_magnum_service_firsttime(self,
|
|
mock_ms_refresh,
|
|
mock_ms_create,
|
|
mock_ms_get
|
|
):
|
|
periodic_a = periodic.MagnumPeriodicTasks(CONF, 'fake-conductor')
|
|
mock_ms_get.return_value = None
|
|
|
|
periodic_a.update_magnum_service(None)
|
|
|
|
mock_ms_get.assert_called_once_with(mock.ANY, periodic_a.host,
|
|
periodic_a.binary)
|
|
mock_ms_create.assert_called_once_with(mock.ANY)
|
|
mock_ms_refresh.assert_called_once_with(mock.ANY)
|
|
|
|
@mock.patch.object(objects.MagnumService, 'get_by_host_and_binary')
|
|
@mock.patch.object(objects.MagnumService, 'create')
|
|
def test_update_magnum_service_on_restart(self,
|
|
mock_ms_create,
|
|
mock_ms_get):
|
|
periodic_a = periodic.MagnumPeriodicTasks(CONF, 'fake-conductor')
|
|
mock_ms_get.return_value = self.fake_ms
|
|
|
|
periodic_a.update_magnum_service(None)
|
|
|
|
mock_ms_get.assert_called_once_with(mock.ANY, periodic_a.host,
|
|
periodic_a.binary)
|
|
self.fake_ms_refresh.assert_called_once_with(mock.ANY)
|
|
|
|
def test_update_magnum_service_regular(self):
|
|
periodic_a = periodic.MagnumPeriodicTasks(CONF, 'fake-conductor')
|
|
periodic_a.magnum_service_ref = self.fake_ms
|
|
|
|
periodic_a.update_magnum_service(None)
|
|
|
|
self.fake_ms_refresh.assert_called_once_with(mock.ANY)
|
|
|
|
@mock.patch('magnum.conductor.monitors.create_monitor')
|
|
@mock.patch('magnum.objects.Bay.list')
|
|
@mock.patch('magnum.common.rpc.get_notifier')
|
|
@mock.patch('magnum.common.context.make_admin_context')
|
|
def test_send_bay_metrics(self, mock_make_admin_context, mock_get_notifier,
|
|
mock_bay_list, mock_create_monitor):
|
|
"""Test if RPC notifier receives the expected message"""
|
|
mock_make_admin_context.return_value = self.context
|
|
notifier = mock.MagicMock()
|
|
mock_get_notifier.return_value = notifier
|
|
mock_bay_list.return_value = [self.bay1, self.bay2, self.bay3,
|
|
self.bay4]
|
|
monitor = mock.MagicMock()
|
|
monitor.get_metric_names.return_value = ['metric1', 'metric2']
|
|
monitor.compute_metric_value.return_value = 30
|
|
monitor.get_metric_unit.return_value = '%'
|
|
mock_create_monitor.return_value = monitor
|
|
|
|
periodic.MagnumPeriodicTasks(
|
|
CONF, 'fake-conductor')._send_bay_metrics(self.context)
|
|
|
|
expected_event_type = 'magnum.bay.metrics.update'
|
|
expected_metrics = [
|
|
{
|
|
'name': 'metric1',
|
|
'value': 30,
|
|
'unit': '%',
|
|
},
|
|
{
|
|
'name': 'metric2',
|
|
'value': 30,
|
|
'unit': '%',
|
|
},
|
|
]
|
|
expected_msg = {
|
|
'user_id': self.bay4.user_id,
|
|
'project_id': self.bay4.project_id,
|
|
'resource_id': self.bay4.uuid,
|
|
'metrics': expected_metrics
|
|
}
|
|
|
|
self.assertEqual(mock_create_monitor.call_count, 1)
|
|
notifier.info.assert_called_once_with(
|
|
self.context, expected_event_type, expected_msg)
|
|
|
|
@mock.patch('magnum.conductor.monitors.create_monitor')
|
|
@mock.patch('magnum.objects.Bay.list')
|
|
@mock.patch('magnum.common.rpc.get_notifier')
|
|
@mock.patch('magnum.common.context.make_admin_context')
|
|
def test_send_bay_metrics_compute_metric_raise(
|
|
self, mock_make_admin_context, mock_get_notifier, mock_bay_list,
|
|
mock_create_monitor):
|
|
mock_make_admin_context.return_value = self.context
|
|
notifier = mock.MagicMock()
|
|
mock_get_notifier.return_value = notifier
|
|
mock_bay_list.return_value = [self.bay4]
|
|
monitor = mock.MagicMock()
|
|
monitor.get_metric_names.return_value = ['metric1', 'metric2']
|
|
monitor.compute_metric_value.side_effect = Exception(
|
|
"error on computing metric")
|
|
mock_create_monitor.return_value = monitor
|
|
|
|
periodic.MagnumPeriodicTasks(
|
|
CONF, 'fake-conductor')._send_bay_metrics(self.context)
|
|
|
|
expected_event_type = 'magnum.bay.metrics.update'
|
|
expected_msg = {
|
|
'user_id': self.bay4.user_id,
|
|
'project_id': self.bay4.project_id,
|
|
'resource_id': self.bay4.uuid,
|
|
'metrics': []
|
|
}
|
|
self.assertEqual(mock_create_monitor.call_count, 1)
|
|
notifier.info.assert_called_once_with(
|
|
self.context, expected_event_type, expected_msg)
|
|
|
|
@mock.patch('magnum.conductor.monitors.create_monitor')
|
|
@mock.patch('magnum.objects.Bay.list')
|
|
@mock.patch('magnum.common.rpc.get_notifier')
|
|
@mock.patch('magnum.common.context.make_admin_context')
|
|
def test_send_bay_metrics_pull_data_raise(
|
|
self, mock_make_admin_context, mock_get_notifier, mock_bay_list,
|
|
mock_create_monitor):
|
|
mock_make_admin_context.return_value = self.context
|
|
notifier = mock.MagicMock()
|
|
mock_get_notifier.return_value = notifier
|
|
mock_bay_list.return_value = [self.bay4]
|
|
monitor = mock.MagicMock()
|
|
monitor.pull_data.side_effect = Exception("error on pulling data")
|
|
mock_create_monitor.return_value = monitor
|
|
|
|
periodic.MagnumPeriodicTasks(
|
|
CONF, 'fake-conductor')._send_bay_metrics(self.context)
|
|
|
|
self.assertEqual(mock_create_monitor.call_count, 1)
|
|
self.assertEqual(notifier.info.call_count, 0)
|
|
|
|
@mock.patch('magnum.conductor.monitors.create_monitor')
|
|
@mock.patch('magnum.objects.Bay.list')
|
|
@mock.patch('magnum.common.rpc.get_notifier')
|
|
@mock.patch('magnum.common.context.make_admin_context')
|
|
def test_send_bay_metrics_monitor_none(
|
|
self, mock_make_admin_context, mock_get_notifier, mock_bay_list,
|
|
mock_create_monitor):
|
|
mock_make_admin_context.return_value = self.context
|
|
notifier = mock.MagicMock()
|
|
mock_get_notifier.return_value = notifier
|
|
mock_bay_list.return_value = [self.bay4]
|
|
mock_create_monitor.return_value = None
|
|
|
|
periodic.MagnumPeriodicTasks(
|
|
CONF, 'fake-conductor')._send_bay_metrics(self.context)
|
|
|
|
self.assertEqual(mock_create_monitor.call_count, 1)
|
|
self.assertEqual(notifier.info.call_count, 0)
|