296 lines
12 KiB
Python
296 lines
12 KiB
Python
# Copyright 2015 Huawei Technologies Co.,LTD.
|
|
#
|
|
# 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_serialization import jsonutils
|
|
|
|
from magnum.conductor import k8s_monitor
|
|
from magnum.conductor import mesos_monitor
|
|
from magnum.conductor import monitors
|
|
from magnum.conductor import swarm_monitor
|
|
from magnum import objects
|
|
from magnum.tests import base
|
|
from magnum.tests.unit.db import utils
|
|
from magnum.tests.unit.objects import utils as obj_utils
|
|
|
|
|
|
class MonitorsTestCase(base.TestCase):
|
|
|
|
test_metrics_spec = {
|
|
'metric1': {
|
|
'unit': 'metric1_unit',
|
|
'func': 'metric1_func',
|
|
},
|
|
'metric2': {
|
|
'unit': 'metric2_unit',
|
|
'func': 'metric2_func',
|
|
},
|
|
}
|
|
|
|
def setUp(self):
|
|
super(MonitorsTestCase, self).setUp()
|
|
|
|
bay = utils.get_test_bay(node_addresses=['1.2.3.4'],
|
|
api_address='https://5.6.7.8:2376',
|
|
master_addresses=['10.0.0.6'])
|
|
self.bay = objects.Bay(self.context, **bay)
|
|
self.monitor = swarm_monitor.SwarmMonitor(self.context, self.bay)
|
|
self.k8s_monitor = k8s_monitor.K8sMonitor(self.context, self.bay)
|
|
self.mesos_monitor = mesos_monitor.MesosMonitor(self.context,
|
|
self.bay)
|
|
p = mock.patch('magnum.conductor.swarm_monitor.SwarmMonitor.'
|
|
'metrics_spec', new_callable=mock.PropertyMock)
|
|
self.mock_metrics_spec = p.start()
|
|
self.mock_metrics_spec.return_value = self.test_metrics_spec
|
|
self.addCleanup(p.stop)
|
|
|
|
def test_create_monitor_success(self):
|
|
self.bay.baymodel = obj_utils.get_test_baymodel(
|
|
self.context, uuid=self.bay.baymodel_id, coe='swarm')
|
|
monitor = monitors.create_monitor(self.context, self.bay)
|
|
self.assertIsInstance(monitor, swarm_monitor.SwarmMonitor)
|
|
|
|
def test_create_monitor_k8s_bay(self):
|
|
self.bay.baymodel = obj_utils.get_test_baymodel(
|
|
self.context, uuid=self.bay.baymodel_id, coe='kubernetes')
|
|
monitor = monitors.create_monitor(self.context, self.bay)
|
|
self.assertIsInstance(monitor, k8s_monitor.K8sMonitor)
|
|
|
|
def test_create_monitor_mesos_bay(self):
|
|
self.bay.baymodel = obj_utils.get_test_baymodel(
|
|
self.context, uuid=self.bay.baymodel_id, coe='mesos')
|
|
monitor = monitors.create_monitor(self.context, self.bay)
|
|
self.assertIsInstance(monitor, mesos_monitor.MesosMonitor)
|
|
|
|
@mock.patch('magnum.common.docker_utils.docker_for_bay')
|
|
def test_swarm_monitor_pull_data_success(self, mock_docker_for_bay):
|
|
mock_docker = mock.MagicMock()
|
|
mock_docker.info.return_value = {'DriverStatus': [[
|
|
u' \u2514 Reserved Memory', u'0 B / 1 GiB']]}
|
|
mock_docker.containers.return_value = [mock.MagicMock()]
|
|
mock_docker.inspect_container.return_value = 'test_container'
|
|
mock_docker_for_bay.return_value.__enter__.return_value = mock_docker
|
|
|
|
self.monitor.pull_data()
|
|
|
|
self.assertEqual([{'MemTotal': 1073741824.0}],
|
|
self.monitor.data['nodes'])
|
|
self.assertEqual(['test_container'], self.monitor.data['containers'])
|
|
|
|
@mock.patch('magnum.common.docker_utils.docker_for_bay')
|
|
def test_swarm_monitor_pull_data_raise(self, mock_docker_for_bay):
|
|
mock_container = mock.MagicMock()
|
|
mock_docker = mock.MagicMock()
|
|
mock_docker.info.return_value = {'DriverStatus': [[
|
|
u' \u2514 Reserved Memory', u'0 B / 1 GiB']]}
|
|
mock_docker.containers.return_value = [mock_container]
|
|
mock_docker.inspect_container.side_effect = Exception("inspect error")
|
|
mock_docker_for_bay.return_value.__enter__.return_value = mock_docker
|
|
|
|
self.monitor.pull_data()
|
|
|
|
self.assertEqual([{'MemTotal': 1073741824.0}],
|
|
self.monitor.data['nodes'])
|
|
self.assertEqual([mock_container], self.monitor.data['containers'])
|
|
|
|
def test_swarm_monitor_get_metric_names(self):
|
|
names = self.monitor.get_metric_names()
|
|
self.assertEqual(sorted(['metric1', 'metric2']), sorted(names))
|
|
|
|
def test_swarm_monitor_get_metric_unit(self):
|
|
unit = self.monitor.get_metric_unit('metric1')
|
|
self.assertEqual('metric1_unit', unit)
|
|
|
|
def test_swarm_monitor_compute_metric_value(self):
|
|
mock_func = mock.MagicMock()
|
|
mock_func.return_value = 'metric1_value'
|
|
self.monitor.metric1_func = mock_func
|
|
value = self.monitor.compute_metric_value('metric1')
|
|
self.assertEqual('metric1_value', value)
|
|
|
|
def test_swarm_monitor_compute_memory_util(self):
|
|
test_data = {
|
|
'nodes': [
|
|
{
|
|
'Name': 'node',
|
|
'MemTotal': 20,
|
|
},
|
|
],
|
|
'containers': [
|
|
{
|
|
'Name': 'container',
|
|
'HostConfig': {
|
|
'Memory': 10,
|
|
},
|
|
},
|
|
],
|
|
}
|
|
self.monitor.data = test_data
|
|
mem_util = self.monitor.compute_memory_util()
|
|
self.assertEqual(50, mem_util)
|
|
|
|
test_data = {
|
|
'nodes': [],
|
|
'containers': [],
|
|
}
|
|
self.monitor.data = test_data
|
|
mem_util = self.monitor.compute_memory_util()
|
|
self.assertEqual(0, mem_util)
|
|
|
|
@mock.patch('magnum.conductor.k8s_api.create_k8s_api')
|
|
def test_k8s_monitor_pull_data_success(self, mock_k8s_api):
|
|
mock_nodes = mock.MagicMock()
|
|
mock_node = mock.MagicMock()
|
|
mock_node.status = mock.MagicMock()
|
|
mock_node.status.capacity = "{'memory': '2000Ki'}"
|
|
mock_nodes.items = [mock_node]
|
|
mock_k8s_api.return_value.list_namespaced_node.return_value = (
|
|
mock_nodes)
|
|
mock_pods = mock.MagicMock()
|
|
mock_pod = mock.MagicMock()
|
|
mock_pod.spec = mock.MagicMock()
|
|
mock_container = mock.MagicMock()
|
|
mock_container.resources = mock.MagicMock()
|
|
mock_container.resources.limits = "{'memory':'100Mi'}"
|
|
mock_pod.spec.containers = [mock_container]
|
|
mock_pods.items = [mock_pod]
|
|
mock_k8s_api.return_value.list_namespaced_pod.return_value = mock_pods
|
|
|
|
self.k8s_monitor.pull_data()
|
|
self.assertEqual(self.k8s_monitor.data['nodes'],
|
|
[{'Memory': 2048000.0}])
|
|
self.assertEqual(self.k8s_monitor.data['pods'],
|
|
[{'Memory': 104857600.0}])
|
|
|
|
def test_k8s_monitor_get_metric_names(self):
|
|
k8s_metric_spec = 'magnum.conductor.k8s_monitor.K8sMonitor.'\
|
|
'metrics_spec'
|
|
with mock.patch(k8s_metric_spec,
|
|
new_callable=mock.PropertyMock) as mock_k8s_metric:
|
|
mock_k8s_metric.return_value = self.test_metrics_spec
|
|
names = self.k8s_monitor.get_metric_names()
|
|
self.assertEqual(sorted(['metric1', 'metric2']), sorted(names))
|
|
|
|
def test_k8s_monitor_get_metric_unit(self):
|
|
k8s_metric_spec = 'magnum.conductor.k8s_monitor.K8sMonitor.' \
|
|
'metrics_spec'
|
|
with mock.patch(k8s_metric_spec,
|
|
new_callable=mock.PropertyMock) as mock_k8s_metric:
|
|
mock_k8s_metric.return_value = self.test_metrics_spec
|
|
unit = self.k8s_monitor.get_metric_unit('metric1')
|
|
self.assertEqual('metric1_unit', unit)
|
|
|
|
def test_k8s_monitor_compute_memory_util(self):
|
|
test_data = {
|
|
'nodes': [
|
|
{
|
|
'Memory': 20,
|
|
},
|
|
],
|
|
'pods': [
|
|
{
|
|
'Memory': 10,
|
|
},
|
|
],
|
|
}
|
|
self.k8s_monitor.data = test_data
|
|
mem_util = self.k8s_monitor.compute_memory_util()
|
|
self.assertEqual(50, mem_util)
|
|
|
|
test_data = {
|
|
'nodes': [],
|
|
'pods': [],
|
|
}
|
|
self.k8s_monitor.data = test_data
|
|
mem_util = self.k8s_monitor.compute_memory_util()
|
|
self.assertEqual(0, mem_util)
|
|
|
|
def _test_mesos_monitor_pull_data(
|
|
self, mock_url_get, state_json, expected_mem_total,
|
|
expected_mem_used):
|
|
state_json = jsonutils.dumps(state_json)
|
|
mock_url_get.return_value = state_json
|
|
self.mesos_monitor.pull_data()
|
|
self.assertEqual(self.mesos_monitor.data['mem_total'],
|
|
expected_mem_total)
|
|
self.assertEqual(self.mesos_monitor.data['mem_used'],
|
|
expected_mem_used)
|
|
|
|
@mock.patch('magnum.common.urlfetch.get')
|
|
def test_mesos_monitor_pull_data_success(self, mock_url_get):
|
|
state_json = {
|
|
'leader': 'master@10.0.0.6:5050',
|
|
'pid': 'master@10.0.0.6:5050',
|
|
'slaves': [{
|
|
'resources': {
|
|
'mem': 100
|
|
},
|
|
'used_resources': {
|
|
'mem': 50
|
|
}
|
|
}]
|
|
}
|
|
self._test_mesos_monitor_pull_data(mock_url_get, state_json, 100, 50)
|
|
|
|
@mock.patch('magnum.common.urlfetch.get')
|
|
def test_mesos_monitor_pull_data_success_not_leader(self, mock_url_get):
|
|
state_json = {
|
|
'leader': 'master@10.0.0.6:5050',
|
|
'pid': 'master@1.1.1.1:5050',
|
|
'slaves': []
|
|
}
|
|
self._test_mesos_monitor_pull_data(mock_url_get, state_json, 0, 0)
|
|
|
|
@mock.patch('magnum.common.urlfetch.get')
|
|
def test_mesos_monitor_pull_data_success_no_master(self, mock_url_get):
|
|
self.bay.master_addresses = []
|
|
self._test_mesos_monitor_pull_data(mock_url_get, {}, 0, 0)
|
|
|
|
def test_mesos_monitor_get_metric_names(self):
|
|
mesos_metric_spec = 'magnum.conductor.mesos_monitor.MesosMonitor.'\
|
|
'metrics_spec'
|
|
with mock.patch(mesos_metric_spec,
|
|
new_callable=mock.PropertyMock) as mock_mesos_metric:
|
|
mock_mesos_metric.return_value = self.test_metrics_spec
|
|
names = self.mesos_monitor.get_metric_names()
|
|
self.assertEqual(sorted(['metric1', 'metric2']), sorted(names))
|
|
|
|
def test_mesos_monitor_get_metric_unit(self):
|
|
mesos_metric_spec = 'magnum.conductor.mesos_monitor.MesosMonitor.' \
|
|
'metrics_spec'
|
|
with mock.patch(mesos_metric_spec,
|
|
new_callable=mock.PropertyMock) as mock_mesos_metric:
|
|
mock_mesos_metric.return_value = self.test_metrics_spec
|
|
unit = self.mesos_monitor.get_metric_unit('metric1')
|
|
self.assertEqual('metric1_unit', unit)
|
|
|
|
def test_mesos_monitor_compute_memory_util(self):
|
|
test_data = {
|
|
'mem_total': 100,
|
|
'mem_used': 50
|
|
}
|
|
self.mesos_monitor.data = test_data
|
|
mem_util = self.mesos_monitor.compute_memory_util()
|
|
self.assertEqual(50, mem_util)
|
|
|
|
test_data = {
|
|
'mem_total': 0,
|
|
'pods': 0,
|
|
}
|
|
self.mesos_monitor.data = test_data
|
|
mem_util = self.mesos_monitor.compute_memory_util()
|
|
self.assertEqual(0, mem_util)
|