def setup(self, mock_boot, mock_filesystem): self.setup = mock.Mock() kiwi.builder.pxe.SystemSetup = mock.Mock(return_value=self.setup) self.boot_image_task = mock.MagicMock() self.boot_image_task.boot_root_directory = 'initrd_dir' mock_boot.return_value = self.boot_image_task self.filesystem = mock.MagicMock() self.filesystem.filename = 'myimage.fs' mock_filesystem.return_value = self.filesystem self.xml_state = mock.Mock() self.xml_state.get_image_version = mock.Mock(return_value='1.2.3') self.xml_state.xml_data.get_name = mock.Mock(return_value='some-image') kernel_type = namedtuple('kernel', ['filename', 'version']) xen_type = namedtuple('xen', ['filename', 'name']) self.kernel = mock.Mock() self.kernel.get_kernel = mock.Mock( return_value=kernel_type(filename='some-kernel', version='42')) self.kernel.get_xen_hypervisor = mock.Mock( return_value=xen_type(filename='hypervisor', name='xen.gz')) kiwi.builder.pxe.Kernel = mock.Mock(return_value=self.kernel) self.pxe = PxeBuilder( self.xml_state, 'target_dir', 'root_dir', custom_args={'signing_keys': ['key_file_a', 'key_file_b']}) self.machine = mock.Mock() self.machine.get_domain = mock.Mock(return_value='dom0') self.pxe.machine = self.machine self.pxe.image_name = 'myimage'
def setup(self, mock_boot, mock_filesystem): self.setup = Mock() kiwi.builder.pxe.SystemSetup = Mock(return_value=self.setup) self.boot_image_task = MagicMock() self.boot_image_task.boot_root_directory = 'initrd_dir' self.boot_image_task.initrd_filename = 'initrd_file_name' mock_boot.return_value = self.boot_image_task self.filesystem = MagicMock() self.filesystem.filename = 'myimage.fs' self.filesystem.root_uuid = 'some_uuid' mock_filesystem.return_value = self.filesystem self.xml_state = Mock() self.xml_state.get_image_version = Mock(return_value='1.2.3') self.xml_state.get_initrd_system = Mock(return_value='dracut') self.xml_state.xml_data.get_name = Mock(return_value='some-image') self.xml_state.build_type = Mock() self.xml_state.build_type.get_kernelcmdline = Mock( return_value='console=hvc0') kernel_type = namedtuple('kernel', ['filename', 'version']) xen_type = namedtuple('xen', ['filename', 'name']) self.kernel = Mock() self.kernel.get_kernel = Mock( return_value=kernel_type(filename='some-kernel', version='42')) self.kernel.get_xen_hypervisor = Mock( return_value=xen_type(filename='hypervisor', name='xen.gz')) kiwi.builder.pxe.Kernel = Mock(return_value=self.kernel) self.pxe = PxeBuilder( self.xml_state, 'target_dir', 'root_dir', custom_args={'signing_keys': ['key_file_a', 'key_file_b']}) self.pxe.image_name = 'myimage' self.pxe.compressed = True
def __new__(self, xml_state, target_dir, root_dir, custom_args=None): requested_image_type = xml_state.get_build_type_name() if requested_image_type in Defaults.get_filesystem_image_types(): return FileSystemBuilder( xml_state, target_dir, root_dir ) elif requested_image_type in Defaults.get_disk_image_types(): return DiskBuilder( xml_state, target_dir, root_dir, custom_args ) elif requested_image_type in Defaults.get_live_image_types(): return LiveImageBuilder( xml_state, target_dir, root_dir, custom_args ) elif requested_image_type in Defaults.get_network_image_types(): return PxeBuilder( xml_state, target_dir, root_dir, custom_args ) elif requested_image_type in Defaults.get_archive_image_types(): return ArchiveBuilder( xml_state, target_dir, root_dir, custom_args ) elif requested_image_type in Defaults.get_container_image_types(): return ContainerBuilder( xml_state, target_dir, root_dir, custom_args ) else: raise KiwiRequestedTypeError( 'requested image type %s not supported' % requested_image_type )
def setup(self, mock_boot, mock_filesystem): self.setup = mock.Mock() kiwi.builder.pxe.SystemSetup = mock.Mock( return_value=self.setup ) self.boot_image_task = mock.MagicMock() self.boot_image_task.boot_root_directory = 'initrd_dir' mock_boot.return_value = self.boot_image_task self.filesystem = mock.MagicMock() self.filesystem.filename = 'myimage' mock_filesystem.return_value = self.filesystem self.xml_state = mock.Mock() self.xml_state.get_image_version = mock.Mock( return_value='1.2.3' ) self.xml_state.xml_data.get_name = mock.Mock( return_value='some-image' ) kernel_type = namedtuple( 'kernel', ['filename', 'version'] ) xen_type = namedtuple( 'xen', ['filename', 'name'] ) self.kernel = mock.Mock() self.kernel.get_kernel = mock.Mock( return_value=kernel_type(filename='some-kernel', version='42') ) self.kernel.get_xen_hypervisor = mock.Mock( return_value=xen_type(filename='hypervisor', name='xen.gz') ) kiwi.builder.pxe.Kernel = mock.Mock( return_value=self.kernel ) self.pxe = PxeBuilder( self.xml_state, 'target_dir', 'root_dir' ) self.machine = mock.Mock() self.machine.get_domain = mock.Mock( return_value='dom0' ) self.pxe.machine = self.machine self.pxe.image_name = 'myimage'
class TestPxeBuilder: @patch('kiwi.builder.pxe.FileSystemBuilder') @patch('kiwi.builder.pxe.BootImage') def setup(self, mock_boot, mock_filesystem): self.setup = Mock() kiwi.builder.pxe.SystemSetup = Mock( return_value=self.setup ) self.boot_image_task = MagicMock() self.boot_image_task.boot_root_directory = 'initrd_dir' self.boot_image_task.initrd_filename = 'initrd_file_name' mock_boot.return_value = self.boot_image_task self.filesystem = MagicMock() self.filesystem.filename = 'myimage.fs' mock_filesystem.return_value = self.filesystem self.xml_state = Mock() self.xml_state.get_image_version = Mock( return_value='1.2.3' ) self.xml_state.xml_data.get_name = Mock( return_value='some-image' ) kernel_type = namedtuple( 'kernel', ['filename', 'version'] ) xen_type = namedtuple( 'xen', ['filename', 'name'] ) self.kernel = Mock() self.kernel.get_kernel = Mock( return_value=kernel_type(filename='some-kernel', version='42') ) self.kernel.get_xen_hypervisor = Mock( return_value=xen_type(filename='hypervisor', name='xen.gz') ) kiwi.builder.pxe.Kernel = Mock( return_value=self.kernel ) self.pxe = PxeBuilder( self.xml_state, 'target_dir', 'root_dir', custom_args={'signing_keys': ['key_file_a', 'key_file_b']} ) self.pxe.image_name = 'myimage' self.pxe.compressed = True @patch('kiwi.builder.pxe.Checksum') @patch('kiwi.builder.pxe.Compress') @patch('kiwi.logger.log.warning') @patch('kiwi.builder.pxe.ArchiveTar') @patch('os.rename') def test_create( self, mock_rename, mock_tar, mock_log_warn, mock_compress, mock_checksum ): tar = Mock() mock_tar.return_value = tar compress = Mock() mock_compress.return_value = compress compress.compressed_filename = 'compressed-file-name' checksum = Mock() mock_checksum.return_value = checksum self.boot_image_task.required = Mock( return_value=True ) self.pxe.create() self.filesystem.create.assert_called_once_with() mock_rename.assert_called_once_with( 'myimage.fs', 'myimage' ) compress.xz.assert_called_once_with(None) checksum.md5.assert_called_once_with( 'target_dir/some-image.x86_64-1.2.3.md5' ) self.boot_image_task.prepare.assert_called_once_with() self.setup.export_modprobe_setup.assert_called_once_with( 'initrd_dir' ) self.boot_image_task.create_initrd.assert_called_once_with() self.setup.export_package_list.assert_called_once_with( 'target_dir' ) self.setup.export_package_verification.assert_called_once_with( 'target_dir' ) tar.create.assert_called_once_with('target_dir') self.pxe.compressed = False self.pxe.create() tar.create_xz_compressed.assert_called_once_with( 'target_dir', xz_options=['--threads=0'] ) @patch('kiwi.builder.pxe.Checksum') @patch('kiwi.builder.pxe.Compress') @patch('os.rename') def test_create_no_kernel_found( self, mock_rename, mock_compress, mock_checksum ): compress = Mock() mock_compress.return_value = compress compress.compressed_filename = 'compressed-file-name' self.kernel.get_kernel.return_value = False with raises(KiwiPxeBootImageError): self.pxe.create() @patch('kiwi.builder.pxe.Checksum') @patch('kiwi.builder.pxe.Compress') @patch('os.rename') def test_create_no_hypervisor_found( self, mock_rename, mock_compress, mock_checksum ): compress = Mock() mock_compress.return_value = compress compress.compressed_filename = 'compressed-file-name' self.kernel.get_xen_hypervisor.return_value = False with raises(KiwiPxeBootImageError): self.pxe.create()
class TestPxeBuilder(object): @patch('kiwi.builder.pxe.FileSystemBuilder') @patch('kiwi.builder.pxe.BootImage') def setup(self, mock_boot, mock_filesystem): self.setup = mock.Mock() kiwi.builder.pxe.SystemSetup = mock.Mock( return_value=self.setup ) self.boot_image_task = mock.MagicMock() self.boot_image_task.boot_root_directory = 'initrd_dir' self.boot_image_task.initrd_filename = 'initrd_file_name' mock_boot.return_value = self.boot_image_task self.filesystem = mock.MagicMock() self.filesystem.filename = 'myimage.fs' mock_filesystem.return_value = self.filesystem self.xml_state = mock.Mock() self.xml_state.get_image_version = mock.Mock( return_value='1.2.3' ) self.xml_state.xml_data.get_name = mock.Mock( return_value='some-image' ) kernel_type = namedtuple( 'kernel', ['filename', 'version'] ) xen_type = namedtuple( 'xen', ['filename', 'name'] ) self.kernel = mock.Mock() self.kernel.get_kernel = mock.Mock( return_value=kernel_type(filename='some-kernel', version='42') ) self.kernel.get_xen_hypervisor = mock.Mock( return_value=xen_type(filename='hypervisor', name='xen.gz') ) kiwi.builder.pxe.Kernel = mock.Mock( return_value=self.kernel ) self.pxe = PxeBuilder( self.xml_state, 'target_dir', 'root_dir', custom_args={'signing_keys': ['key_file_a', 'key_file_b']} ) self.pxe.image_name = 'myimage' @patch('kiwi.builder.pxe.Checksum') @patch('kiwi.builder.pxe.Compress') @patch('kiwi.logger.log.warning') @patch('kiwi.builder.pxe.Command.run') @patch('os.rename') def test_create( self, mock_rename, mock_command, mock_log_warn, mock_compress, mock_checksum ): compress = mock.Mock() mock_compress.return_value = compress compress.compressed_filename = 'compressed-file-name' checksum = mock.Mock() mock_checksum.return_value = checksum self.boot_image_task.required = mock.Mock( return_value=True ) self.pxe.create() self.filesystem.create.assert_called_once_with() mock_rename.assert_called_once_with( 'myimage.fs', 'myimage' ) compress.xz.assert_called_once_with(None) checksum.md5.assert_called_once_with('compressed-file-name.md5') self.boot_image_task.prepare.assert_called_once_with() self.setup.export_modprobe_setup.assert_called_once_with( 'initrd_dir' ) self.boot_image_task.create_initrd.assert_called_once_with() self.setup.export_package_list.assert_called_once_with( 'target_dir' ) self.setup.export_package_verification.assert_called_once_with( 'target_dir' ) mock_command.assert_called_once_with( [ 'bash', '-c', 'tar -C target_dir -c --to-stdout ' + 'myimage-42.kernel ' + 'initrd_file_name ' + 'compressed-file-name ' + 'compressed-file-name.md5 ' + '| xz -f --threads=0 > ' + 'target_dir/some-image.x86_64-1.2.3.tar.xz' ] ) # warning for not implemented pxedeploy handling assert mock_log_warn.called @patch('kiwi.builder.pxe.Checksum') @patch('kiwi.builder.pxe.Compress') @patch('os.rename') @raises(KiwiPxeBootImageError) def test_create_no_kernel_found( self, mock_rename, mock_compress, mock_checksum ): compress = mock.Mock() mock_compress.return_value = compress compress.compressed_filename = 'compressed-file-name' self.kernel.get_kernel.return_value = False self.pxe.create() @patch('kiwi.builder.pxe.Checksum') @patch('kiwi.builder.pxe.Compress') @patch('os.rename') @raises(KiwiPxeBootImageError) def test_create_no_hypervisor_found( self, mock_rename, mock_compress, mock_checksum ): compress = mock.Mock() mock_compress.return_value = compress compress.compressed_filename = 'compressed-file-name' self.kernel.get_xen_hypervisor.return_value = False self.pxe.create()
class TestPxeBuilder(object): @patch('kiwi.builder.pxe.FileSystemBuilder') @patch('kiwi.builder.pxe.BootImage') def setup(self, mock_boot, mock_filesystem): self.setup = mock.Mock() kiwi.builder.pxe.SystemSetup = mock.Mock( return_value=self.setup ) self.boot_image_task = mock.MagicMock() self.boot_image_task.boot_root_directory = 'initrd_dir' self.boot_image_task.initrd_filename = 'initrd_file_name' mock_boot.return_value = self.boot_image_task self.filesystem = mock.MagicMock() self.filesystem.filename = 'myimage.fs' mock_filesystem.return_value = self.filesystem self.xml_state = mock.Mock() self.xml_state.get_image_version = mock.Mock( return_value='1.2.3' ) self.xml_state.xml_data.get_name = mock.Mock( return_value='some-image' ) kernel_type = namedtuple( 'kernel', ['filename', 'version'] ) xen_type = namedtuple( 'xen', ['filename', 'name'] ) self.kernel = mock.Mock() self.kernel.get_kernel = mock.Mock( return_value=kernel_type(filename='some-kernel', version='42') ) self.kernel.get_xen_hypervisor = mock.Mock( return_value=xen_type(filename='hypervisor', name='xen.gz') ) kiwi.builder.pxe.Kernel = mock.Mock( return_value=self.kernel ) self.pxe = PxeBuilder( self.xml_state, 'target_dir', 'root_dir', custom_args={'signing_keys': ['key_file_a', 'key_file_b']} ) self.pxe.image_name = 'myimage' @patch('kiwi.builder.pxe.Checksum') @patch('kiwi.builder.pxe.Compress') @patch('kiwi.logger.log.warning') @patch('kiwi.builder.pxe.Command.run') @patch('os.rename') def test_create( self, mock_rename, mock_command, mock_log_warn, mock_compress, mock_checksum ): compress = mock.Mock() mock_compress.return_value = compress compress.compressed_filename = 'compressed-file-name' checksum = mock.Mock() mock_checksum.return_value = checksum self.boot_image_task.required = mock.Mock( return_value=True ) self.pxe.create() self.filesystem.create.assert_called_once_with() mock_rename.assert_called_once_with( 'myimage.fs', 'myimage' ) compress.xz.assert_called_once_with(None) checksum.md5.assert_called_once_with( 'target_dir/some-image.x86_64-1.2.3.md5' ) self.boot_image_task.prepare.assert_called_once_with() self.setup.export_modprobe_setup.assert_called_once_with( 'initrd_dir' ) self.boot_image_task.create_initrd.assert_called_once_with() self.setup.export_package_list.assert_called_once_with( 'target_dir' ) self.setup.export_package_verification.assert_called_once_with( 'target_dir' ) mock_command.assert_called_once_with( [ 'bash', '-c', 'tar -C target_dir -c --to-stdout ' + 'myimage-42.kernel ' + 'initrd_file_name ' + 'compressed-file-name ' + 'some-image.x86_64-1.2.3.md5 ' + '| xz -f --threads=0 > ' + 'target_dir/some-image.x86_64-1.2.3.tar.xz' ] ) # warning for not implemented pxedeploy handling assert mock_log_warn.called @patch('kiwi.builder.pxe.Checksum') @patch('kiwi.builder.pxe.Compress') @patch('os.rename') @raises(KiwiPxeBootImageError) def test_create_no_kernel_found( self, mock_rename, mock_compress, mock_checksum ): compress = mock.Mock() mock_compress.return_value = compress compress.compressed_filename = 'compressed-file-name' self.kernel.get_kernel.return_value = False self.pxe.create() @patch('kiwi.builder.pxe.Checksum') @patch('kiwi.builder.pxe.Compress') @patch('os.rename') @raises(KiwiPxeBootImageError) def test_create_no_hypervisor_found( self, mock_rename, mock_compress, mock_checksum ): compress = mock.Mock() mock_compress.return_value = compress compress.compressed_filename = 'compressed-file-name' self.kernel.get_xen_hypervisor.return_value = False self.pxe.create()
class TestPxeBuilder(object): @patch('kiwi.builder.pxe.FileSystemBuilder') @patch('kiwi.builder.pxe.BootImage') def setup(self, mock_boot, mock_filesystem): self.setup = mock.Mock() kiwi.builder.pxe.SystemSetup = mock.Mock(return_value=self.setup) self.boot_image_task = mock.MagicMock() self.boot_image_task.boot_root_directory = 'initrd_dir' mock_boot.return_value = self.boot_image_task self.filesystem = mock.MagicMock() self.filesystem.filename = 'myimage' mock_filesystem.return_value = self.filesystem self.xml_state = mock.Mock() self.xml_state.get_image_version = mock.Mock(return_value='1.2.3') self.xml_state.xml_data.get_name = mock.Mock(return_value='some-image') kernel_type = namedtuple('kernel', ['filename', 'version']) xen_type = namedtuple('xen', ['filename', 'name']) self.kernel = mock.Mock() self.kernel.get_kernel = mock.Mock( return_value=kernel_type(filename='some-kernel', version='42')) self.kernel.get_xen_hypervisor = mock.Mock( return_value=xen_type(filename='hypervisor', name='xen.gz')) kiwi.builder.pxe.Kernel = mock.Mock(return_value=self.kernel) self.pxe = PxeBuilder(self.xml_state, 'target_dir', 'root_dir') self.machine = mock.Mock() self.machine.get_domain = mock.Mock(return_value='dom0') self.pxe.machine = self.machine self.pxe.image_name = 'myimage' @patch('kiwi.builder.pxe.Checksum') @patch('kiwi.builder.pxe.Compress') @patch('kiwi.logger.log.warning') def test_create(self, mock_log_warn, mock_compress, mock_checksum): compress = mock.Mock() mock_compress.return_value = compress checksum = mock.Mock() mock_checksum.return_value = checksum self.boot_image_task.required = mock.Mock(return_value=True) self.pxe.create() self.filesystem.create.assert_called_once_with() compress.xz.assert_called_once_with() checksum.md5.assert_called_once_with('myimage.md5') self.boot_image_task.prepare.assert_called_once_with() self.setup.export_modprobe_setup.assert_called_once_with('initrd_dir') self.boot_image_task.create_initrd.assert_called_once_with() self.setup.export_rpm_package_list.assert_called_once_with( 'target_dir') self.setup.export_rpm_package_verification.assert_called_once_with( 'target_dir') # warning for not implemented pxedeploy handling assert mock_log_warn.called @patch('kiwi.builder.pxe.Checksum') @patch('kiwi.builder.pxe.Compress') @raises(KiwiPxeBootImageError) def test_create_no_kernel_found(self, mock_compress, mock_checksum): self.kernel.get_kernel.return_value = False self.pxe.create() @patch('kiwi.builder.pxe.Checksum') @patch('kiwi.builder.pxe.Compress') @raises(KiwiPxeBootImageError) def test_create_no_hypervisor_found(self, mock_compress, mock_checksum): self.kernel.get_xen_hypervisor.return_value = False self.pxe.create()
class TestPxeBuilder(object): @patch('kiwi.builder.pxe.FileSystemBuilder') @patch('kiwi.builder.pxe.BootImage') def setup(self, mock_boot, mock_filesystem): self.setup = mock.Mock() kiwi.builder.pxe.SystemSetup = mock.Mock( return_value=self.setup ) self.boot_image_task = mock.MagicMock() self.boot_image_task.boot_root_directory = 'initrd_dir' mock_boot.return_value = self.boot_image_task self.filesystem = mock.MagicMock() self.filesystem.filename = 'myimage' mock_filesystem.return_value = self.filesystem self.xml_state = mock.Mock() self.xml_state.get_image_version = mock.Mock( return_value='1.2.3' ) self.xml_state.xml_data.get_name = mock.Mock( return_value='some-image' ) kernel_type = namedtuple( 'kernel', ['filename', 'version'] ) xen_type = namedtuple( 'xen', ['filename', 'name'] ) self.kernel = mock.Mock() self.kernel.get_kernel = mock.Mock( return_value=kernel_type(filename='some-kernel', version='42') ) self.kernel.get_xen_hypervisor = mock.Mock( return_value=xen_type(filename='hypervisor', name='xen.gz') ) kiwi.builder.pxe.Kernel = mock.Mock( return_value=self.kernel ) self.pxe = PxeBuilder( self.xml_state, 'target_dir', 'root_dir' ) self.machine = mock.Mock() self.machine.get_domain = mock.Mock( return_value='dom0' ) self.pxe.machine = self.machine self.pxe.image_name = 'myimage' @patch('kiwi.builder.pxe.Checksum') @patch('kiwi.builder.pxe.Compress') @patch('kiwi.logger.log.warning') def test_create(self, mock_log_warn, mock_compress, mock_checksum): compress = mock.Mock() mock_compress.return_value = compress checksum = mock.Mock() mock_checksum.return_value = checksum self.boot_image_task.required = mock.Mock( return_value=True ) self.pxe.create() self.filesystem.create.assert_called_once_with() compress.xz.assert_called_once_with() checksum.md5.assert_called_once_with('myimage.md5') self.boot_image_task.prepare.assert_called_once_with() self.setup.export_modprobe_setup.assert_called_once_with( 'initrd_dir' ) self.boot_image_task.create_initrd.assert_called_once_with() self.setup.export_rpm_package_list.assert_called_once_with( 'target_dir' ) self.setup.export_rpm_package_verification.assert_called_once_with( 'target_dir' ) # warning for not implemented pxedeploy handling assert mock_log_warn.called @patch('kiwi.builder.pxe.Checksum') @patch('kiwi.builder.pxe.Compress') @raises(KiwiPxeBootImageError) def test_create_no_kernel_found(self, mock_compress, mock_checksum): self.kernel.get_kernel.return_value = False self.pxe.create() @patch('kiwi.builder.pxe.Checksum') @patch('kiwi.builder.pxe.Compress') @raises(KiwiPxeBootImageError) def test_create_no_hypervisor_found(self, mock_compress, mock_checksum): self.kernel.get_xen_hypervisor.return_value = False self.pxe.create()