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")