diff options
author | John L. Villalovos <john.l.villalovos@intel.com> | 2015-09-29 17:34:28 -0700 |
---|---|---|
committer | John L. Villalovos <john.l.villalovos@intel.com> | 2015-09-30 17:47:19 -0700 |
commit | ef6c57eb5f41472ad77b7e3b8500fc06cdae28f1 (patch) | |
tree | 013f8025c81830e776d335bfeaad5efe2dd0a7e8 /ironic/tests/unit/drivers/modules/test_inspector.py | |
parent | 32d3e98ca5fbd8120d04ca22f18d4bbfa6483e25 (diff) | |
download | ironic-ef6c57eb5f41472ad77b7e3b8500fc06cdae28f1.tar.gz |
Move unit tests to correct directory
Move unit tests that test ironic/drivers/modules/*
from ironic/tests/drivers/* to ironic/tests/drivers/modules/*
Update any unit test imports that were broken by move.
Change-Id: Ic5c2e4143ad76e9d910a4361483dc09145dc1dff
Diffstat (limited to 'ironic/tests/unit/drivers/modules/test_inspector.py')
-rw-r--r-- | ironic/tests/unit/drivers/modules/test_inspector.py | 239 |
1 files changed, 239 insertions, 0 deletions
diff --git a/ironic/tests/unit/drivers/modules/test_inspector.py b/ironic/tests/unit/drivers/modules/test_inspector.py new file mode 100644 index 000000000..349acb4b8 --- /dev/null +++ b/ironic/tests/unit/drivers/modules/test_inspector.py @@ -0,0 +1,239 @@ +# 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 eventlet +import ironic_inspector_client as client +import mock + +from ironic.common import driver_factory +from ironic.common import exception +from ironic.common import keystone +from ironic.common import states +from ironic.conductor import task_manager +from ironic.drivers.modules import inspector +from ironic.tests.unit.conductor import utils as mgr_utils +from ironic.tests.unit.db import base as db_base +from ironic.tests.unit.objects import utils as obj_utils + + +class DisabledTestCase(db_base.DbTestCase): + def setUp(self): + super(DisabledTestCase, self).setUp() + + def _do_mock(self): + # NOTE(dtantsur): fake driver always has inspection, using another one + mgr_utils.mock_the_extension_manager("pxe_ssh") + self.driver = driver_factory.get_driver("pxe_ssh") + + def test_disabled(self): + self.config(enabled=False, group='inspector') + self._do_mock() + self.assertIsNone(self.driver.inspect) + # NOTE(dtantsur): it's expected that fake_inspector fails to load + # in this case + self.assertRaises(exception.DriverLoadError, + mgr_utils.mock_the_extension_manager, + "fake_inspector") + + def test_enabled(self): + self.config(enabled=True, group='inspector') + self._do_mock() + self.assertIsNotNone(self.driver.inspect) + + @mock.patch.object(inspector, 'client', None) + def test_init_inspector_not_imported(self): + self.assertRaises(exception.DriverLoadError, + inspector.Inspector) + + def test_init_ok(self): + self.config(enabled=True, group='inspector') + inspector.Inspector() + + +class BaseTestCase(db_base.DbTestCase): + def setUp(self): + super(BaseTestCase, self).setUp() + self.config(enabled=True, group='inspector') + mgr_utils.mock_the_extension_manager("fake_inspector") + self.driver = driver_factory.get_driver("fake_inspector") + self.node = obj_utils.get_test_node(self.context) + self.task = mock.MagicMock(spec=task_manager.TaskManager) + self.task.context = mock.MagicMock(spec_set=['auth_token']) + self.task.shared = False + self.task.node = self.node + self.task.driver = self.driver + self.api_version = (1, 0) + + +class CommonFunctionsTestCase(BaseTestCase): + def test_validate_ok(self): + self.driver.inspect.validate(self.task) + + def test_get_properties(self): + res = self.driver.inspect.get_properties() + self.assertEqual({}, res) + + def test_create_if_enabled(self): + res = inspector.Inspector.create_if_enabled('driver') + self.assertIsInstance(res, inspector.Inspector) + + @mock.patch.object(inspector.LOG, 'info', autospec=True) + def test_create_if_enabled_disabled(self, warn_mock): + self.config(enabled=False, group='inspector') + res = inspector.Inspector.create_if_enabled('driver') + self.assertIsNone(res) + self.assertTrue(warn_mock.called) + + +@mock.patch.object(eventlet, 'spawn_n', lambda f, *a, **kw: f(*a, **kw)) +@mock.patch.object(client, 'introspect') +class InspectHardwareTestCase(BaseTestCase): + def test_ok(self, mock_introspect): + self.assertEqual(states.INSPECTING, + self.driver.inspect.inspect_hardware(self.task)) + mock_introspect.assert_called_once_with( + self.node.uuid, + api_version=self.api_version, + auth_token=self.task.context.auth_token) + + def test_url(self, mock_introspect): + self.config(service_url='meow', group='inspector') + self.assertEqual(states.INSPECTING, + self.driver.inspect.inspect_hardware(self.task)) + mock_introspect.assert_called_once_with( + self.node.uuid, + api_version=self.api_version, + auth_token=self.task.context.auth_token, + base_url='meow') + + @mock.patch.object(task_manager, 'acquire', autospec=True) + def test_error(self, mock_acquire, mock_introspect): + mock_introspect.side_effect = RuntimeError('boom') + self.driver.inspect.inspect_hardware(self.task) + mock_introspect.assert_called_once_with( + self.node.uuid, + api_version=self.api_version, + auth_token=self.task.context.auth_token) + task = mock_acquire.return_value.__enter__.return_value + self.assertIn('boom', task.node.last_error) + task.process_event.assert_called_once_with('fail') + + +@mock.patch.object(keystone, 'get_admin_auth_token', lambda: 'the token') +@mock.patch.object(client, 'get_status') +class CheckStatusTestCase(BaseTestCase): + def setUp(self): + super(CheckStatusTestCase, self).setUp() + self.node.provision_state = states.INSPECTING + + def test_not_inspecting(self, mock_get): + self.node.provision_state = states.MANAGEABLE + inspector._check_status(self.task) + self.assertFalse(mock_get.called) + + def test_not_inspector(self, mock_get): + self.task.driver.inspect = object() + inspector._check_status(self.task) + self.assertFalse(mock_get.called) + + def test_not_finished(self, mock_get): + mock_get.return_value = {} + inspector._check_status(self.task) + mock_get.assert_called_once_with(self.node.uuid, + api_version=self.api_version, + auth_token='the token') + self.assertFalse(self.task.process_event.called) + + def test_exception_ignored(self, mock_get): + mock_get.side_effect = RuntimeError('boom') + inspector._check_status(self.task) + mock_get.assert_called_once_with(self.node.uuid, + api_version=self.api_version, + auth_token='the token') + self.assertFalse(self.task.process_event.called) + + def test_status_ok(self, mock_get): + mock_get.return_value = {'finished': True} + inspector._check_status(self.task) + mock_get.assert_called_once_with(self.node.uuid, + api_version=self.api_version, + auth_token='the token') + self.task.process_event.assert_called_once_with('done') + + def test_status_error(self, mock_get): + mock_get.return_value = {'error': 'boom'} + inspector._check_status(self.task) + mock_get.assert_called_once_with(self.node.uuid, + api_version=self.api_version, + auth_token='the token') + self.task.process_event.assert_called_once_with('fail') + self.assertIn('boom', self.node.last_error) + + def test_service_url(self, mock_get): + self.config(service_url='meow', group='inspector') + mock_get.return_value = {'finished': True} + inspector._check_status(self.task) + mock_get.assert_called_once_with(self.node.uuid, + api_version=self.api_version, + auth_token='the token', + base_url='meow') + self.task.process_event.assert_called_once_with('done') + + def test_is_standalone(self, mock_get): + self.config(auth_strategy='noauth') + mock_get.return_value = {'finished': True} + inspector._check_status(self.task) + mock_get.assert_called_once_with( + self.node.uuid, + api_version=self.api_version, + auth_token=self.task.context.auth_token) + self.task.process_event.assert_called_once_with('done') + + def test_not_standalone(self, mock_get): + self.config(auth_strategy='keystone') + mock_get.return_value = {'finished': True} + inspector._check_status(self.task) + mock_get.assert_called_once_with(self.node.uuid, + api_version=self.api_version, + auth_token='the token') + self.task.process_event.assert_called_once_with('done') + + +@mock.patch.object(eventlet.greenthread, 'spawn_n', + lambda f, *a, **kw: f(*a, **kw)) +@mock.patch.object(task_manager, 'acquire', autospec=True) +@mock.patch.object(inspector, '_check_status', autospec=True) +class PeriodicTaskTestCase(BaseTestCase): + def test_ok(self, mock_check, mock_acquire): + mgr = mock.MagicMock(spec=['iter_nodes']) + mgr.iter_nodes.return_value = [('1', 'd1'), ('2', 'd2')] + tasks = [mock.sentinel.task1, mock.sentinel.task2] + mock_acquire.side_effect = ( + mock.MagicMock(__enter__=mock.MagicMock(return_value=task)) + for task in tasks + ) + inspector.Inspector()._periodic_check_result( + mgr, mock.sentinel.context) + mock_check.assert_any_call(tasks[0]) + mock_check.assert_any_call(tasks[1]) + self.assertEqual(2, mock_acquire.call_count) + + def test_node_locked(self, mock_check, mock_acquire): + iter_nodes_ret = [('1', 'd1'), ('2', 'd2')] + mock_acquire.side_effect = iter([exception.NodeLocked("boom")] * + len(iter_nodes_ret)) + mgr = mock.MagicMock(spec=['iter_nodes']) + mgr.iter_nodes.return_value = iter_nodes_ret + inspector.Inspector()._periodic_check_result( + mgr, mock.sentinel.context) + self.assertFalse(mock_check.called) + self.assertEqual(2, mock_acquire.call_count) |