Source code for ironicclient.tests.unit.v1.test_create_resources

#   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 jsonschema
import mock
import six.moves.builtins as __builtin__

from ironicclient import exc
from ironicclient.tests.unit import utils
from ironicclient.v1 import create_resources

valid_json = {
    "chassis": [{
        "description": "testing resources import",
        "nodes": [{
            "driver": "agent_ssh",
            "extra": {
                "kv1": True,
                "vk1": None
            },
            "properties": {
                "a": "c"
            },
            "ports": [{
                "address": "00:00:00:00:00:01",
                "extra": {
                    "a": "b"
                }
            }]
        }]
    }],
    "nodes": [{
        "driver": "fake",
        "driver_info": {
            "fake_key": "fake",
            "dict_prop": {
                "a": "b"
            },
            "arr_prop": [
                1, 2, 3
            ]
        },
        "chassis_uuid": "10f99593-b8c2-4fcb-8858-494c1a47cee6"
    }]
}

ironic_pov_invalid_json = {
    "nodes": [{
        "driver": "non_existent",
        "ports": [{
            "address": "invalid_address"
        }]
    }]
}

schema_pov_invalid_json = {"meow": "woof!"}


[docs]class CreateSchemaTest(utils.BaseTestCase):
[docs] def test_schema(self): schema = create_resources._CREATE_SCHEMA jsonschema.validate(valid_json, schema) jsonschema.validate(ironic_pov_invalid_json, schema) self.assertRaises(jsonschema.ValidationError, jsonschema.validate, schema_pov_invalid_json, schema)
[docs]class CreateResourcesTest(utils.BaseTestCase):
[docs] def setUp(self): super(CreateResourcesTest, self).setUp() self.client = mock.MagicMock()
@mock.patch.object(create_resources, 'create_nodes', autospec=True) @mock.patch.object(create_resources, 'create_chassis', autospec=True) @mock.patch.object(jsonschema, 'validate', autospec=True) @mock.patch.object(create_resources, 'load_from_file', side_effect=[valid_json], autospec=True)
[docs] def test_create_resources( self, mock_load, mock_validate, mock_chassis, mock_nodes): resources_files = ['file.json'] create_resources.create_resources(self.client, resources_files) mock_load.assert_has_calls([ mock.call('file.json') ]) mock_validate.assert_called_once_with(valid_json, mock.ANY) mock_chassis.assert_called_once_with(self.client, valid_json['chassis']) mock_nodes.assert_called_once_with(self.client, valid_json['nodes'])
@mock.patch.object(create_resources, 'create_nodes', autospec=True) @mock.patch.object(create_resources, 'create_chassis', autospec=True) @mock.patch.object(jsonschema, 'validate', autospec=True) @mock.patch.object(create_resources, 'load_from_file', side_effect=exc.ClientException, autospec=True)
[docs] def test_create_resources_cannot_read_schema( self, mock_load, mock_validate, mock_chassis, mock_nodes): resources_files = ['file.json'] self.assertRaises(exc.ClientException, create_resources.create_resources, self.client, resources_files) mock_load.assert_called_once_with('file.json') self.assertFalse(mock_validate.called) self.assertFalse(mock_chassis.called) self.assertFalse(mock_nodes.called)
@mock.patch.object(create_resources, 'create_nodes', autospec=True) @mock.patch.object(create_resources, 'create_chassis', autospec=True) @mock.patch.object(jsonschema, 'validate', side_effect=jsonschema.ValidationError(''), autospec=True) @mock.patch.object(create_resources, 'load_from_file', side_effect=[schema_pov_invalid_json], autospec=True)
[docs] def test_create_resources_validation_fails( self, mock_load, mock_validate, mock_chassis, mock_nodes): resources_files = ['file.json'] self.assertRaises(exc.ClientException, create_resources.create_resources, self.client, resources_files) mock_load.assert_has_calls([ mock.call('file.json') ]) mock_validate.assert_called_once_with(schema_pov_invalid_json, mock.ANY) self.assertFalse(mock_chassis.called) self.assertFalse(mock_nodes.called)
@mock.patch.object(create_resources, 'create_nodes', autospec=True) @mock.patch.object(create_resources, 'create_chassis', autospec=True) @mock.patch.object(jsonschema, 'validate', side_effect=[None, jsonschema.ValidationError('')], autospec=True) @mock.patch.object(create_resources, 'load_from_file', side_effect=[valid_json, schema_pov_invalid_json], autospec=True)
[docs] def test_create_resources_validation_fails_multiple( self, mock_load, mock_validate, mock_chassis, mock_nodes): resources_files = ['file.json', 'file2.json'] self.assertRaises(exc.ClientException, create_resources.create_resources, self.client, resources_files) mock_load.assert_has_calls([ mock.call('file.json'), mock.call('file2.json') ]) mock_validate.assert_has_calls([ mock.call(valid_json, mock.ANY), mock.call(schema_pov_invalid_json, mock.ANY) ]) self.assertFalse(mock_chassis.called) self.assertFalse(mock_nodes.called)
@mock.patch.object(create_resources, 'create_nodes', autospec=True) @mock.patch.object(create_resources, 'create_chassis', autospec=True) @mock.patch.object(jsonschema, 'validate', autospec=True) @mock.patch.object(create_resources, 'load_from_file', side_effect=[ironic_pov_invalid_json], autospec=True)
[docs] def test_create_resources_ironic_fails_to_create( self, mock_load, mock_validate, mock_chassis, mock_nodes): mock_nodes.return_value = [exc.ClientException('cannot create that')] mock_chassis.return_value = [] resources_files = ['file.json'] self.assertRaises(exc.ClientException, create_resources.create_resources, self.client, resources_files) mock_load.assert_has_calls([ mock.call('file.json') ]) mock_validate.assert_called_once_with(ironic_pov_invalid_json, mock.ANY) mock_chassis.assert_called_once_with(self.client, []) mock_nodes.assert_called_once_with( self.client, ironic_pov_invalid_json['nodes'])
[docs]class LoadFromFileTest(utils.BaseTestCase): @mock.patch.object(__builtin__, 'open', mock.mock_open(read_data='{"a": "b"}'))
[docs] def test_load_json(self): fname = 'abc.json' res = create_resources.load_from_file(fname) self.assertEqual({'a': 'b'}, res)
@mock.patch.object(__builtin__, 'open', mock.mock_open(read_data='{"a": "b"}'))
[docs] def test_load_unknown_extension(self): fname = 'abc' self.assertRaisesRegex(exc.ClientException, 'must have .json or .yaml extension', create_resources.load_from_file, fname)
@mock.patch.object(__builtin__, 'open')
[docs] def test_load_ioerror(self, mock_open): mock_open.side_effect = IOError('file does not exist') fname = 'abc.json' self.assertRaisesRegex(exc.ClientException, 'Cannot read file', create_resources.load_from_file, fname)
@mock.patch.object(__builtin__, 'open', mock.mock_open(read_data='{{bbb'))
[docs] def test_load_incorrect_json(self): fname = 'abc.json' self.assertRaisesRegex( exc.ClientException, 'File "%s" is invalid' % fname, create_resources.load_from_file, fname)
@mock.patch.object(__builtin__, 'open', mock.mock_open(read_data='---\na: b'))
[docs] def test_load_yaml(self): fname = 'abc.yaml' res = create_resources.load_from_file(fname) self.assertEqual({'a': 'b'}, res)
@mock.patch.object(__builtin__, 'open', mock.mock_open(read_data='---\na-: - b'))
[docs] def test_load_incorrect_yaml(self): fname = 'abc.yaml' self.assertRaisesRegex( exc.ClientException, 'File "%s" is invalid' % fname, create_resources.load_from_file, fname)
[docs]class CreateMethodsTest(utils.BaseTestCase):
[docs] def setUp(self): super(CreateMethodsTest, self).setUp() self.client = mock.MagicMock()
[docs] def test_create_single_node(self): params = {'driver': 'fake'} self.client.node.create.return_value = mock.Mock(uuid='uuid') self.assertEqual( ('uuid', None), create_resources.create_single_node(self.client, **params) ) self.client.node.create.assert_called_once_with(driver='fake')
[docs] def test_create_single_node_with_ports(self): params = {'driver': 'fake', 'ports': ['some ports here']} self.client.node.create.return_value = mock.Mock(uuid='uuid') self.assertEqual( ('uuid', None), create_resources.create_single_node(self.client, **params) ) self.client.node.create.assert_called_once_with(driver='fake')
[docs] def test_create_single_node_raises_client_exception(self): params = {'driver': 'fake'} e = exc.ClientException('foo') self.client.node.create.side_effect = e res, err = create_resources.create_single_node(self.client, **params) self.assertIsNone(res) self.assertIsInstance(err, exc.ClientException) self.assertIn('Unable to create the node', str(err)) self.client.node.create.assert_called_once_with(driver='fake')
[docs] def test_create_single_node_raises_invalid_exception(self): params = {'driver': 'fake'} e = exc.InvalidAttribute('foo') self.client.node.create.side_effect = e res, err = create_resources.create_single_node(self.client, **params) self.assertIsNone(res) self.assertIsInstance(err, exc.InvalidAttribute) self.assertIn('Cannot create the node with attributes', str(err)) self.client.node.create.assert_called_once_with(driver='fake')
[docs] def test_create_single_port(self): params = {'address': 'fake-address', 'node_uuid': 'fake-node-uuid'} self.client.port.create.return_value = mock.Mock(uuid='fake-port-uuid') self.assertEqual( ('fake-port-uuid', None), create_resources.create_single_port(self.client, **params) ) self.client.port.create.assert_called_once_with(**params)
[docs] def test_create_single_chassis(self): self.client.chassis.create.return_value = mock.Mock(uuid='uuid') self.assertEqual( ('uuid', None), create_resources.create_single_chassis(self.client) ) self.client.chassis.create.assert_called_once_with()
[docs] def test_create_single_chassis_with_nodes(self): params = {'nodes': ['some nodes here']} self.client.chassis.create.return_value = mock.Mock(uuid='uuid') self.assertEqual( ('uuid', None), create_resources.create_single_chassis(self.client, **params) ) self.client.chassis.create.assert_called_once_with()
[docs] def test_create_ports(self): port = {'address': 'fake-address'} port_with_node_uuid = port.copy() port_with_node_uuid.update(node_uuid='fake-node-uuid') self.client.port.create.return_value = mock.Mock(uuid='uuid') self.assertEqual([], create_resources.create_ports(self.client, [port], 'fake-node-uuid')) self.client.port.create.assert_called_once_with(**port_with_node_uuid)
[docs] def test_create_ports_two_node_uuids(self): port = {'address': 'fake-address', 'node_uuid': 'node-uuid-1'} self.client.port.create.return_value = mock.Mock(uuid='uuid') errs = create_resources.create_ports(self.client, [port], 'node-uuid-2') self.assertIsInstance(errs[0], exc.ClientException) self.assertEqual(1, len(errs))
@mock.patch.object(create_resources, 'create_ports', autospec=True)
[docs] def test_create_nodes(self, mock_create_ports): node = {'driver': 'fake', 'ports': ['list of ports']} self.client.node.create.return_value = mock.Mock(uuid='uuid') self.assertEqual([], create_resources.create_nodes(self.client, [node])) self.client.node.create.assert_called_once_with(driver='fake') mock_create_ports.assert_called_once_with( self.client, ['list of ports'], node_uuid='uuid')
@mock.patch.object(create_resources, 'create_ports', autospec=True)
[docs] def test_create_nodes_exception(self, mock_create_ports): node = {'driver': 'fake', 'ports': ['list of ports']} self.client.node.create.side_effect = exc.ClientException('bar') errs = create_resources.create_nodes(self.client, [node]) self.assertIsInstance(errs[0], exc.ClientException) self.assertEqual(1, len(errs)) self.client.node.create.assert_called_once_with(driver='fake') self.assertFalse(mock_create_ports.called)
@mock.patch.object(create_resources, 'create_ports', autospec=True)
[docs] def test_create_nodes_two_chassis_uuids(self, mock_create_ports): node = {'driver': 'fake', 'ports': ['list of ports'], 'chassis_uuid': 'chassis-uuid-1'} errs = create_resources.create_nodes(self.client, [node], chassis_uuid='chassis-uuid-2') self.assertFalse(self.client.node.create.called) self.assertFalse(mock_create_ports.called) self.assertEqual(1, len(errs)) self.assertIsInstance(errs[0], exc.ClientException)
@mock.patch.object(create_resources, 'create_ports', autospec=True)
[docs] def test_create_nodes_no_ports(self, mock_create_ports): node = {'driver': 'fake'} self.client.node.create.return_value = mock.Mock(uuid='uuid') self.assertEqual([], create_resources.create_nodes(self.client, [node])) self.client.node.create.assert_called_once_with(driver='fake') self.assertFalse(mock_create_ports.called)
@mock.patch.object(create_resources, 'create_nodes', autospec=True)
[docs] def test_create_chassis(self, mock_create_nodes): chassis = {'description': 'fake', 'nodes': ['list of nodes']} self.client.chassis.create.return_value = mock.Mock(uuid='uuid') self.assertEqual([], create_resources.create_chassis(self.client, [chassis])) self.client.chassis.create.assert_called_once_with(description='fake') mock_create_nodes.assert_called_once_with( self.client, ['list of nodes'], chassis_uuid='uuid')
@mock.patch.object(create_resources, 'create_nodes', autospec=True)
[docs] def test_create_chassis_exception(self, mock_create_nodes): chassis = {'description': 'fake', 'nodes': ['list of nodes']} self.client.chassis.create.side_effect = exc.ClientException('bar') errs = create_resources.create_chassis(self.client, [chassis]) self.client.chassis.create.assert_called_once_with(description='fake') self.assertFalse(mock_create_nodes.called) self.assertEqual(1, len(errs)) self.assertIsInstance(errs[0], exc.ClientException)
@mock.patch.object(create_resources, 'create_nodes', autospec=True)
[docs] def test_create_chassis_no_nodes(self, mock_create_nodes): chassis = {'description': 'fake'} self.client.chassis.create.return_value = mock.Mock(uuid='uuid') self.assertEqual([], create_resources.create_chassis(self.client, [chassis])) self.client.chassis.create.assert_called_once_with(description='fake') self.assertFalse(mock_create_nodes.called)

Project Source