def __new__(self, name, root_dir, custom_args=None):
if name == 'docker':
return ContainerSetupDocker(root_dir, custom_args)
elif name == 'oci':
return ContainerSetupOCI(root_dir, custom_args)
else:
raise KiwiContainerSetupError(
'Support for %s container not implemented' % name)
def setup(self, mock_exists):
mock_exists.return_value = True
self.context_manager_mock = mock.MagicMock()
self.file_mock = mock.MagicMock()
self.enter_mock = mock.MagicMock()
self.exit_mock = mock.MagicMock()
self.enter_mock.return_value = self.file_mock
setattr(self.context_manager_mock, '__enter__', self.enter_mock)
setattr(self.context_manager_mock, '__exit__', self.exit_mock)
self.container = ContainerSetupDocker('root_dir',
{'container_name': 'system'})
self.container.create_fstab = mock.Mock()
self.container.deactivate_bootloader_setup = mock.Mock()
self.container.deactivate_root_filesystem_check = mock.Mock()
self.container.setup_static_device_nodes = mock.Mock()
self.container.setup_root_console = mock.Mock()
self.container.deactivate_systemd_service = mock.Mock()
def test_init(self, mock_exists):
mock_exists.return_value = True
container = ContainerSetupDocker('root_dir',
{'container_name': 'system'})
assert container.custom_args['container_name'] == 'system'
