337 lines
13 KiB
Python
337 lines
13 KiB
Python
# Copyright 2015 Red Hat, 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 collections
|
|
import copy
|
|
import unittest
|
|
|
|
import mock
|
|
from oslo_concurrency import processutils
|
|
from oslo_config import cfg
|
|
import requests
|
|
import stevedore
|
|
|
|
from ironic_python_agent import errors
|
|
from ironic_python_agent import hardware
|
|
from ironic_python_agent import inspector
|
|
from ironic_python_agent import utils
|
|
|
|
|
|
CONF = cfg.CONF
|
|
|
|
|
|
class AcceptingFailure(mock.Mock):
|
|
def __call__(self, *args):
|
|
return super(mock.Mock, self).__call__(
|
|
*(copy.deepcopy(x) for x in args))
|
|
|
|
def assert_called_with_failure(self, expect_error=False):
|
|
self.assert_called_once_with({}, mock.ANY)
|
|
failure = self.call_args[0][1]
|
|
assert bool(failure) is expect_error, '%s is not %s' % (
|
|
failure, expect_error)
|
|
|
|
|
|
class TestMisc(unittest.TestCase):
|
|
def test_default_collector_loadable(self):
|
|
ext = inspector.extension_manager([inspector.DEFAULT_COLLECTOR])
|
|
self.assertIs(ext[inspector.DEFAULT_COLLECTOR].plugin,
|
|
inspector.collect_default)
|
|
|
|
def test_raise_on_wrong_collector(self):
|
|
self.assertRaisesRegexp(errors.InspectionError,
|
|
'foobar',
|
|
inspector.extension_manager,
|
|
['foobar'])
|
|
|
|
|
|
@mock.patch.object(inspector, 'setup_ipmi_credentials', autospec=True)
|
|
@mock.patch.object(inspector, 'call_inspector', new_callable=AcceptingFailure)
|
|
@mock.patch.object(stevedore, 'NamedExtensionManager', autospec=True)
|
|
class TestInspect(unittest.TestCase):
|
|
def setUp(self):
|
|
super(TestInspect, self).setUp()
|
|
CONF.set_override('inspection_callback_url', 'http://foo/bar')
|
|
CONF.set_override('inspection_collectors', '')
|
|
self.mock_collect = AcceptingFailure()
|
|
self.mock_ext = mock.Mock(spec=['plugin', 'name'],
|
|
plugin=self.mock_collect)
|
|
|
|
def test_ok(self, mock_ext_mgr, mock_call, mock_setup_ipmi):
|
|
mock_ext_mgr.return_value = [self.mock_ext]
|
|
mock_call.return_value = {'uuid': 'uuid1'}
|
|
|
|
result = inspector.inspect()
|
|
|
|
self.mock_collect.assert_called_with_failure()
|
|
mock_call.assert_called_with_failure()
|
|
self.assertEqual('uuid1', result)
|
|
mock_setup_ipmi.assert_called_once_with(mock_call.return_value)
|
|
|
|
def test_collectors_option(self, mock_ext_mgr, mock_call, mock_setup_ipmi):
|
|
CONF.set_override('inspection_collectors', 'foo,bar')
|
|
mock_ext_mgr.return_value = [
|
|
mock.Mock(spec=['name', 'plugin'], plugin=AcceptingFailure()),
|
|
mock.Mock(spec=['name', 'plugin'], plugin=AcceptingFailure()),
|
|
]
|
|
|
|
inspector.inspect()
|
|
|
|
for fake_ext in mock_ext_mgr.return_value:
|
|
fake_ext.plugin.assert_called_with_failure()
|
|
mock_call.assert_called_with_failure()
|
|
|
|
def test_collector_failed(self, mock_ext_mgr, mock_call, mock_setup_ipmi):
|
|
mock_ext_mgr.return_value = [self.mock_ext]
|
|
self.mock_collect.side_effect = RuntimeError('boom')
|
|
|
|
self.assertRaisesRegexp(errors.InspectionError,
|
|
'boom', inspector.inspect)
|
|
|
|
self.mock_collect.assert_called_with_failure()
|
|
mock_call.assert_called_with_failure(expect_error=True)
|
|
self.assertFalse(mock_setup_ipmi.called)
|
|
|
|
def test_extensions_failed(self, mock_ext_mgr, mock_call, mock_setup_ipmi):
|
|
CONF.set_override('inspection_collectors', 'foo,bar')
|
|
mock_ext_mgr.side_effect = RuntimeError('boom')
|
|
|
|
self.assertRaisesRegexp(RuntimeError, 'boom', inspector.inspect)
|
|
|
|
mock_call.assert_called_with_failure(expect_error=True)
|
|
self.assertFalse(mock_setup_ipmi.called)
|
|
|
|
def test_inspector_error(self, mock_ext_mgr, mock_call, mock_setup_ipmi):
|
|
mock_call.return_value = None
|
|
mock_ext_mgr.return_value = [self.mock_ext]
|
|
|
|
result = inspector.inspect()
|
|
|
|
self.mock_collect.assert_called_with_failure()
|
|
mock_call.assert_called_with_failure()
|
|
self.assertIsNone(result)
|
|
self.assertFalse(mock_setup_ipmi.called)
|
|
|
|
|
|
@mock.patch.object(requests, 'post', autospec=True)
|
|
class TestCallInspector(unittest.TestCase):
|
|
def setUp(self):
|
|
super(TestCallInspector, self).setUp()
|
|
CONF.set_override('inspection_callback_url', 'url')
|
|
|
|
def test_ok(self, mock_post):
|
|
failures = utils.AccumulatedFailures()
|
|
data = collections.OrderedDict(data=42)
|
|
mock_post.return_value.status_code = 200
|
|
|
|
res = inspector.call_inspector(data, failures)
|
|
|
|
mock_post.assert_called_once_with('url',
|
|
data='{"data": 42, "error": null}')
|
|
self.assertEqual(mock_post.return_value.json.return_value, res)
|
|
|
|
def test_send_failure(self, mock_post):
|
|
failures = mock.Mock(spec=utils.AccumulatedFailures)
|
|
failures.get_error.return_value = "boom"
|
|
data = collections.OrderedDict(data=42)
|
|
mock_post.return_value.status_code = 200
|
|
|
|
res = inspector.call_inspector(data, failures)
|
|
|
|
mock_post.assert_called_once_with('url',
|
|
data='{"data": 42, "error": "boom"}')
|
|
self.assertEqual(mock_post.return_value.json.return_value, res)
|
|
|
|
def test_inspector_error(self, mock_post):
|
|
failures = utils.AccumulatedFailures()
|
|
data = collections.OrderedDict(data=42)
|
|
mock_post.return_value.status_code = 400
|
|
|
|
res = inspector.call_inspector(data, failures)
|
|
|
|
mock_post.assert_called_once_with('url',
|
|
data='{"data": 42, "error": null}')
|
|
self.assertIsNone(res)
|
|
|
|
|
|
@mock.patch.object(utils, 'execute', autospec=True)
|
|
class TestSetupIpmiCredentials(unittest.TestCase):
|
|
def setUp(self):
|
|
super(TestSetupIpmiCredentials, self).setUp()
|
|
self.resp = {'ipmi_username': 'user',
|
|
'ipmi_password': 'pwd',
|
|
'ipmi_setup_credentials': True}
|
|
|
|
def test_disabled(self, mock_call):
|
|
del self.resp['ipmi_setup_credentials']
|
|
|
|
inspector.setup_ipmi_credentials(self.resp)
|
|
|
|
self.assertFalse(mock_call.called)
|
|
|
|
def test_ok(self, mock_call):
|
|
inspector.setup_ipmi_credentials(self.resp)
|
|
|
|
expected = [
|
|
mock.call('ipmitool', 'user', 'set', 'name', '2', 'user'),
|
|
mock.call('ipmitool', 'user', 'set', 'password', '2', 'pwd'),
|
|
mock.call('ipmitool', 'user', 'enable', '2'),
|
|
mock.call('ipmitool', 'channel', 'setaccess', '1', '2',
|
|
'link=on', 'ipmi=on', 'callin=on', 'privilege=4'),
|
|
]
|
|
self.assertEqual(expected, mock_call.call_args_list)
|
|
|
|
def test_user_failed(self, mock_call):
|
|
mock_call.side_effect = processutils.ProcessExecutionError()
|
|
|
|
self.assertRaises(errors.InspectionError,
|
|
inspector.setup_ipmi_credentials,
|
|
self.resp)
|
|
|
|
mock_call.assert_called_once_with('ipmitool', 'user', 'set', 'name',
|
|
'2', 'user')
|
|
|
|
def test_password_failed(self, mock_call):
|
|
mock_call.side_effect = iter((None,
|
|
processutils.ProcessExecutionError))
|
|
|
|
self.assertRaises(errors.InspectionError,
|
|
inspector.setup_ipmi_credentials,
|
|
self.resp)
|
|
|
|
expected = [
|
|
mock.call('ipmitool', 'user', 'set', 'name', '2', 'user'),
|
|
mock.call('ipmitool', 'user', 'set', 'password', '2', 'pwd')
|
|
]
|
|
self.assertEqual(expected, mock_call.call_args_list)
|
|
|
|
|
|
class BaseDiscoverTest(unittest.TestCase):
|
|
def setUp(self):
|
|
super(BaseDiscoverTest, self).setUp()
|
|
self.inventory = {
|
|
'interfaces': [
|
|
hardware.NetworkInterface(name='em1',
|
|
mac_addr='aa:bb:cc:dd:ee:ff',
|
|
ipv4_address='1.1.1.1'),
|
|
hardware.NetworkInterface(name='em2',
|
|
mac_addr='11:22:33:44:55:66',
|
|
ipv4_address=None),
|
|
],
|
|
'cpu': hardware.CPU(model_name='generic', frequency='3000',
|
|
count=4, architecture='x86_64'),
|
|
'memory': hardware.Memory(total=11998396 * 1024,
|
|
physical_mb=12288),
|
|
'disks': [
|
|
hardware.BlockDevice(name='/dev/sdc',
|
|
model='Disk 2',
|
|
size=500107862016,
|
|
rotational=False),
|
|
hardware.BlockDevice(name='/dev/sda',
|
|
model='Too Small Disk',
|
|
size=4294967295,
|
|
rotational=False),
|
|
hardware.BlockDevice(name='/dev/sdb',
|
|
model='Disk 1',
|
|
size=500107862016,
|
|
rotational=True)
|
|
],
|
|
'bmc_address': '1.2.3.4',
|
|
}
|
|
self.failures = utils.AccumulatedFailures()
|
|
self.data = {}
|
|
|
|
|
|
@mock.patch.object(utils, 'get_agent_params',
|
|
lambda: {'BOOTIF': 'boot:if'})
|
|
class TestDiscoverNetworkProperties(BaseDiscoverTest):
|
|
def test_no_network_interfaces(self):
|
|
self.inventory['interfaces'] = [
|
|
hardware.NetworkInterface(name='lo',
|
|
mac_addr='aa:bb:cc:dd:ee:ff',
|
|
ipv4_address='127.0.0.1')
|
|
]
|
|
|
|
inspector.discover_network_properties(self.inventory, self.data,
|
|
self.failures)
|
|
|
|
self.assertIn('no network interfaces found', self.failures.get_error())
|
|
self.assertFalse(self.data['interfaces'])
|
|
|
|
def test_ok(self):
|
|
inspector.discover_network_properties(self.inventory, self.data,
|
|
self.failures)
|
|
|
|
self.assertEqual({'em1': {'mac': 'aa:bb:cc:dd:ee:ff',
|
|
'ip': '1.1.1.1'},
|
|
'em2': {'mac': '11:22:33:44:55:66',
|
|
'ip': None}},
|
|
self.data['interfaces'])
|
|
self.assertEqual('1.2.3.4', self.data['ipmi_address'])
|
|
self.assertEqual('boot:if', self.data['boot_interface'])
|
|
self.assertFalse(self.failures)
|
|
|
|
def test_missing(self):
|
|
self.inventory['interfaces'] = [
|
|
hardware.NetworkInterface(name='em1',
|
|
mac_addr='aa:bb:cc:dd:ee:ff'),
|
|
hardware.NetworkInterface(name='em2',
|
|
mac_addr=None,
|
|
ipv4_address='1.2.1.2'),
|
|
]
|
|
|
|
inspector.discover_network_properties(self.inventory, self.data,
|
|
self.failures)
|
|
|
|
self.assertEqual({'em1': {'mac': 'aa:bb:cc:dd:ee:ff', 'ip': None}},
|
|
self.data['interfaces'])
|
|
self.assertFalse(self.failures)
|
|
|
|
|
|
class TestDiscoverSchedulingProperties(BaseDiscoverTest):
|
|
def test_ok(self):
|
|
inspector.discover_scheduling_properties(self.inventory, self.data)
|
|
|
|
self.assertEqual({'cpus': 4, 'cpu_arch': 'x86_64', 'local_gb': 464,
|
|
'memory_mb': 12288}, self.data)
|
|
|
|
def test_no_local_gb(self):
|
|
# Some DRAC servers do not have any visible hard drive until RAID is
|
|
# built
|
|
self.inventory['disks'] = []
|
|
|
|
inspector.discover_scheduling_properties(self.inventory, self.data)
|
|
|
|
self.assertEqual({'cpus': 4, 'cpu_arch': 'x86_64', 'memory_mb': 12288},
|
|
self.data)
|
|
|
|
|
|
@mock.patch.object(inspector, 'discover_scheduling_properties', autospec=True)
|
|
@mock.patch.object(inspector, 'discover_network_properties', autospec=True)
|
|
@mock.patch.object(hardware, 'dispatch_to_managers', autospec=True)
|
|
class TestCollectDefault(BaseDiscoverTest):
|
|
def test_ok(self, mock_dispatch, mock_discover_net, mock_discover_sched):
|
|
mock_dispatch.return_value = self.inventory
|
|
|
|
inspector.collect_default(self.data, self.failures)
|
|
|
|
mock_dispatch.assert_called_once_with('list_hardware_info')
|
|
mock_discover_net.assert_called_once_with(self.inventory, self.data,
|
|
self.failures)
|
|
mock_discover_sched.assert_called_once_with(self.inventory, self.data)
|
|
|
|
for key in ('memory', 'interfaces', 'cpu', 'disks'):
|
|
self.assertTrue(self.data['inventory'][key])
|