def test_load_config_files_multiple_files(self, mock_os_path_isfile,
mock_yaml_load):
mock_yaml_load.side_effect = [{
'disk_images': [{
'arch': 'amd64',
'imagename': 'overcloud',
'distro': 'some_awesome_distro',
'type': 'qcow2',
'elements': ['image_element'],
'environment': {
'test_env': '1'
},
}]
}, {
'disk_images': [{
'imagename': 'overcloud',
'elements': ['another_image_element'],
'packages': ['a_package'],
'otherkey': 'some_other_key',
'environment': {
'test_env2': '0'
},
}]
}]
mock_os_path_isfile.return_value = True
mock_open_context = mock.mock_open()
mock_open_context().read.return_value = "YAML"
with mock.patch('six.moves.builtins.open', mock_open_context):
base_manager = BaseImageManager(['yamlfile1', 'yamlfile2'])
disk_images = base_manager.load_config_files('disk_images')
self.assertEqual(2, mock_yaml_load.call_count)
self.assertEqual(
[{
'arch': 'amd64',
'distro': 'some_awesome_distro',
'imagename': 'overcloud',
'type': 'qcow2',
'elements': ['image_element', 'another_image_element'],
'packages': ['a_package'],
'otherkey': 'some_other_key',
'environment': {
'test_env': '1',
'test_env2': '0'
},
}], disk_images)
def test_load_config_files(self, mock_os_path_isfile, mock_yaml_load):
mock_yaml_load.return_value = fakes.create_disk_images()
mock_os_path_isfile.return_value = True
mock_open_context = mock.mock_open()
mock_open_context().read.return_value = "YAML"
with mock.patch('builtins.open', mock_open_context):
base_manager = BaseImageManager(['yamlfile'])
disk_images = base_manager.load_config_files('disk_images')
mock_yaml_load.assert_called_once_with("YAML")
self.assertEqual([{
'arch': 'amd64',
'distro': 'some_awesome_os',
'imagename': 'overcloud',
'type': 'qcow2',
'elements': ['image_element']
}], disk_images)
def test_load_config_files(self, mock_os_path_isfile, mock_yaml_load):
mock_yaml_load.return_value = fakes.create_disk_images()
mock_os_path_isfile.return_value = True
mock_open_context = mock.mock_open()
mock_open_context().read.return_value = "YAML"
with mock.patch('six.moves.builtins.open', mock_open_context):
base_manager = BaseImageManager(['yamlfile'])
disk_images = base_manager.load_config_files('disk_images')
mock_yaml_load.assert_called_once_with("YAML")
self.assertEqual([{
'arch': 'amd64',
'distro': 'some_awesome_os',
'imagename': 'overcloud',
'type': 'qcow2',
'elements': ['image_element']
}], disk_images)
def test_load_config_files_multiple_files(self, mock_os_path_isfile,
mock_yaml_load):
mock_yaml_load.side_effect = [{
'disk_images': [{
'arch': 'amd64',
'imagename': 'overcloud',
'distro': 'some_awesome_distro',
'type': 'qcow2',
'elements': ['image_element'],
'environment': {'test_env': '1'},
}]},
{
'disk_images': [{
'imagename': 'overcloud',
'elements': ['another_image_element'],
'packages': ['a_package'],
'otherkey': 'some_other_key',
'environment': {'test_env2': '0'},
}]}]
mock_os_path_isfile.return_value = True
mock_open_context = mock.mock_open()
mock_open_context().read.return_value = "YAML"
with mock.patch('six.moves.builtins.open', mock_open_context):
base_manager = BaseImageManager(['yamlfile1', 'yamlfile2'])
disk_images = base_manager.load_config_files('disk_images')
self.assertEqual(2, mock_yaml_load.call_count)
self.assertEqual([{
'arch': 'amd64',
'distro': 'some_awesome_distro',
'imagename': 'overcloud',
'type': 'qcow2',
'elements': ['image_element', 'another_image_element'],
'packages': ['a_package'],
'otherkey': 'some_other_key',
'environment': {'test_env': '1', 'test_env2': '0'},
}], disk_images)
def test_load_config_files_single_image(self, mock_os_path_isfile,
mock_yaml_load):
mock_yaml_load.side_effect = [{
'disk_images': [
{
'arch': 'amd64',
'imagename': 'overcloud',
'distro': 'some_awesome_distro',
'type': 'qcow2',
'elements': ['image_element']
},
{
'arch': 'amd64',
'imagename': 'not-overcloud',
'distro': 'some_other_distro',
'type': 'qcow2',
'elements': ['other_element']
}
]}]
mock_os_path_isfile.return_value = True
mock_open_context = mock.mock_open()
mock_open_context().read.return_value = "YAML"
with mock.patch('builtins.open', mock_open_context):
base_manager = BaseImageManager(['yamlfile1'],
images=['not-overcloud'])
disk_images = base_manager.load_config_files('disk_images')
self.assertEqual(1, mock_yaml_load.call_count)
self.assertEqual([{
'arch': 'amd64',
'distro': 'some_other_distro',
'imagename': 'not-overcloud',
'type': 'qcow2',
'elements': ['other_element'],
}], disk_images)
def test_load_config_files_single_image(self, mock_os_path_isfile,
mock_yaml_load):
mock_yaml_load.side_effect = [{
'disk_images': [
{
'arch': 'amd64',
'imagename': 'overcloud',
'distro': 'some_awesome_distro',
'type': 'qcow2',
'elements': ['image_element']
},
{
'arch': 'amd64',
'imagename': 'not-overcloud',
'distro': 'some_other_distro',
'type': 'qcow2',
'elements': ['other_element']
}
]}]
mock_os_path_isfile.return_value = True
mock_open_context = mock.mock_open()
mock_open_context().read.return_value = "YAML"
with mock.patch('six.moves.builtins.open', mock_open_context):
base_manager = BaseImageManager(['yamlfile1'],
images=['not-overcloud'])
disk_images = base_manager.load_config_files('disk_images')
self.assertEqual(1, mock_yaml_load.call_count)
self.assertEqual([{
'arch': 'amd64',
'distro': 'some_other_distro',
'imagename': 'not-overcloud',
'type': 'qcow2',
'elements': ['other_element'],
}], disk_images)
def test_load_config_files_missing_image_name(self, mock_os_path_isfile,
mock_yaml_load):
mock_yaml_load.return_value = {
'disk_images': [{
'arch': 'amd64',
'imagename': 'overcloud',
'type': 'qcow2',
'elements': ['image_element']
}, {
'arch': 'amd64',
'type': 'qcow2',
}]
}
mock_os_path_isfile.return_value = True
mock_open_context = mock.mock_open()
mock_open_context().read.return_value = "YAML"
with mock.patch('builtins.open', mock_open_context):
base_manager = BaseImageManager(['yamlfile'])
self.assertRaises(ImageSpecificationException,
base_manager.load_config_files, 'disk_images')
def test_load_config_files_not_found(self):
base_manager = BaseImageManager(['file/does/not/exist'])
self.assertRaises(IOError, base_manager.load_config_files,
'disk_images')