def setup(self):
xml_data = mock.Mock()
xml_data.get_name = mock.Mock(return_value='some-disk-image')
self.xml_state = mock.Mock()
self.xml_state.xml_data = xml_data
self.disk_format = DiskFormatBase(self.xml_state, 'root_dir',
'target_dir')
class TestDiskFormatBase(object):
def setup(self):
xml_data = mock.Mock()
xml_data.get_name = mock.Mock(return_value="some-disk-image")
self.xml_state = mock.Mock()
self.xml_state.xml_data = xml_data
self.disk_format = DiskFormatBase(self.xml_state, "root_dir", "target_dir")
@raises(NotImplementedError)
def test_create_image_format(self):
self.disk_format.create_image_format()
@raises(KiwiFormatSetupError)
def test_get_target_name_for_format_invalid_format(self):
self.disk_format.get_target_name_for_format("foo")
def test_post_init(self):
self.disk_format.post_init({"option": "unhandled"})
assert self.disk_format.custom_args == {}
def test_get_qemu_option_list(self):
custom_args = {"subformat=format": None, "adapter_type=type": None}
assert self.disk_format.get_qemu_option_list(custom_args) == [
"-o",
"adapter_type=type",
"-o",
"subformat=format",
]
def test_get_target_name_for_format(self):
assert self.disk_format.get_target_name_for_format("vhd") == "target_dir/some-disk-image.vhd"
@patch("kiwi.disk_format_base.Path.wipe")
@patch("os.path.exists")
def test_destructor(self, mock_exists, mock_wipe):
mock_exists.return_value = True
self.disk_format.temp_image_dir = "tmpdir"
self.disk_format.__del__()
self.disk_format.temp_image_dir = None
mock_wipe.assert_called_once_with("tmpdir")
class TestDiskFormatBase(object):
def setup(self):
xml_data = mock.Mock()
xml_data.get_name = mock.Mock(return_value='some-disk-image')
self.xml_state = mock.Mock()
self.xml_state.xml_data = xml_data
self.disk_format = DiskFormatBase(self.xml_state, 'root_dir',
'target_dir')
@raises(NotImplementedError)
def test_create_image_format(self):
self.disk_format.create_image_format()
@raises(KiwiFormatSetupError)
def test_get_target_name_for_format_invalid_format(self):
self.disk_format.get_target_name_for_format('foo')
def test_post_init(self):
self.disk_format.post_init({'option': 'unhandled'})
assert self.disk_format.custom_args == {}
def test_get_qemu_option_list(self):
custom_args = {'subformat=format': None, 'adapter_type=type': None}
assert self.disk_format.get_qemu_option_list(custom_args) == [
'-o', 'adapter_type=type', '-o', 'subformat=format'
]
def test_get_target_name_for_format(self):
assert self.disk_format.get_target_name_for_format('vhd') == \
'target_dir/some-disk-image.vhd'
@patch('kiwi.disk_format_base.Path.wipe')
@patch('os.path.exists')
def test_destructor(self, mock_exists, mock_wipe):
mock_exists.return_value = True
self.disk_format.temp_image_dir = 'tmpdir'
self.disk_format.__del__()
self.disk_format.temp_image_dir = None
mock_wipe.assert_called_once_with('tmpdir')
def setup(self):
xml_data = mock.Mock()
xml_data.get_name = mock.Mock(return_value="some-disk-image")
self.xml_state = mock.Mock()
self.xml_state.xml_data = xml_data
self.disk_format = DiskFormatBase(self.xml_state, "root_dir", "target_dir")
