def test_var_dir_should_be_path_to_stateroot_var_for_randomly_generated_stateroot( self): cfg = Config(Source.url('url'), 'ref') self.assertEqual( cfg.var_dir, os.path.join('/ostree', 'deploy', cfg.stateroot, 'var'))
def test_should_not_error_if_instance_is_created_without_deployment_name_set( self): cfg = Config(Source.url('url'), 'ref', stateroot='test') step = MountVar(cfg) self.assertIsNotNone(step)
def test_var_dir_should_include_sysroot(self): sysroot = os.path.join('/mnt', 'rootfs') cfg = Config(Source.url('url'), 'ref', sysroot=sysroot) self.assertEqual( cfg.var_dir, os.path.join(sysroot, 'ostree', 'deploy', cfg.stateroot, 'var'))
def test_should_return_HttpRemote_if_url_source(self): cfg = Config(Source.url('http://example.com/ostree'), 'ref') steps = get_steps(cfg) self.assertEqual(len(steps), 1) self.assertIsInstance(steps[0], HttpRemote)
def test_should_return_FileRemote_if_path_source(self): cfg = Config(Source.path('/ostree/repo'), 'ref') steps = get_steps(cfg) self.assertEqual(len(steps), 1) self.assertIsInstance(steps[0], FileRemote)
def test_should_raise_exception_if_nothing_was_added_to_deployments_dir( self): cfg = Config(Source.url('url'), 'ref') step = Deploy(cfg) with self.assertRaises(DeployError): step.run()
class TestBuiltinProvisioner(TestCase): cfg = Config(Source.url('url'), 'ref', stateroot='test') cfg.set_deployment_name('test-deploy.0') deploy_dir = os.path.join('/', 'ostree', 'deploy', 'test', 'deploy', 'test-deploy.0') def test_title_should_be_str_instance(self): self.assertIsInstance( BuiltinProvisioner(mock.Mock(), mock.Mock()).title, str) @mock.patch('deploy_ostree.steps.provisioners.builtin.run') def test_should_run_provisioner_script_with_no_arguments( self, run_mock: mock.Mock): provisioner = ProvisionerConfig('etc-fstab', {}) BuiltinProvisioner(self.cfg, provisioner).run() run_mock.assert_called_once_with( [os.path.join(PROVISIONERS_DIR, 'etc-fstab'), self.deploy_dir], check=True, env={}) @mock.patch('deploy_ostree.steps.provisioners.builtin.run') def test_should_run_provisioner_script_with_specified_sysroot( self, run_mock: mock.Mock): sysroot = os.path.join('/mnt', 'rootfs') cfg = Config(Source.url('url'), 'ref', stateroot='test', sysroot=sysroot) cfg.set_deployment_name('test-deploy.0') provisioner = ProvisionerConfig('etc-fstab', {}) BuiltinProvisioner(cfg, provisioner).run() run_mock.assert_called_once_with([ os.path.join(PROVISIONERS_DIR, 'etc-fstab'), os.path.join(sysroot, 'ostree', 'deploy', 'test', 'deploy', 'test-deploy.0') ], check=True, env={}) @mock.patch('deploy_ostree.steps.provisioners.builtin.run') def test_should_run_provisioner_script_with_arguments( self, run_mock: mock.Mock): provisioner = ProvisionerConfig('create-user', { 'username': '******', 'password': '******' }) BuiltinProvisioner(self.cfg, provisioner).run() run_mock.assert_called_once_with( [os.path.join(PROVISIONERS_DIR, 'create-user'), self.deploy_dir], check=True, env={ 'DEPLOY_OSTREE_username': '******', 'DEPLOY_OSTREE_password': '******' })
def test_deployment_path_should_return_path_after_setting_name(self): cfg = Config(Source.url('url'), 'ref', stateroot='test-stateroot') cfg.set_deployment_name('deployment-name') self.assertEqual( cfg.deployment_dir, os.path.join('/ostree', 'deploy', 'test-stateroot', 'deploy', 'deployment-name'))
def test_deployment_dir_should_include_sysroot(self): sysroot = os.path.join('/mnt', 'rootfs') cfg = Config(Source.url('url'), 'ref', sysroot=sysroot) cfg.set_deployment_name('0123deployment.0') self.assertEqual( cfg.deployment_dir, os.path.join(sysroot, 'ostree', 'deploy', cfg.stateroot, 'deploy', '0123deployment.0'))
def test_should_pull_ref(self, run_mock: mock.Mock): cfg = Config(Source.url('url'), 'fedora/28/x86_64/workstation', remote='ostree-remote') PullRef(cfg).run() run_mock.assert_called_once_with([ 'ostree', 'pull', '--repo=%s' % os.path.join('/ostree', 'repo'), 'ostree-remote', 'fedora/28/x86_64/workstation' ], check=True)
def test_should_unmount_on_cleanup(self, run_mock: mock.Mock): cfg = Config(Source.url('url'), 'ref', stateroot='test') cfg.set_deployment_name('test-deploy.0') MountVar(cfg).cleanup() run_mock.assert_called_once_with([ 'umount', '-l', os.path.join('/ostree', 'deploy', 'test', 'deploy', 'test-deploy.0', 'var'), ], check=True)
def test_should_add_ostree_remote_for_url_in_config( self, mock_run: mock.Mock): cfg = Config(Source.url('https://example.com/ostree'), 'debian/9/i386/desktop', remote='remote-name') HttpRemote(cfg).run() mock_run.assert_called_once_with([ 'ostree', 'remote', 'add', '--repo=%s' % os.path.join('/ostree', 'repo'), '--no-gpg-verify', 'remote-name', 'https://example.com/ostree' ], check=True)
def test_should_delete_ostree_remote(self, mock_run: mock.Mock): cfg = Config(Source.url('url'), 'ref', remote='remote-name') DeleteRemote(cfg).run() mock_run.assert_called_once_with([ 'ostree', 'remote', 'delete', '--repo=%s' % os.path.join('/ostree', 'repo'), '--if-exists', 'remote-name', ], check=True)
def test_should_return_BuiltinProvisioner_per_configured_provisioner(self): cfg = Config(Source.url('url'), 'ref', stateroot='test', default_provisioners=[ ProvisionerConfig('create-user', {}), ProvisionerConfig('etc-fstab', {'arg': 'value'}), ]) steps = get_steps(cfg) self.assertEqual(len(steps), 2) self.assertIsInstance(steps[0], BuiltinProvisioner) self.assertIsInstance(steps[1], BuiltinProvisioner)
def test_should_add_ostree_remote_for_path_in_config( self, mock_run: mock.Mock): cfg = Config(Source.path('/srv/ostree/debian/repo'), 'debian/9/i386/desktop', remote='remote-name') FileRemote(cfg).run() mock_run.assert_called_once_with([ 'ostree', 'remote', 'add', '--repo=%s' % os.path.join('/ostree', 'repo'), '--no-gpg-verify', 'remote-name', 'file:///srv/ostree/debian/repo' ], check=True)
def test_should_bind_mount_var_into_deployment(self, run_mock: mock.Mock): cfg = Config(Source.url('url'), 'ref', stateroot='test') cfg.set_deployment_name('test-deploy.0') MountVar(cfg).run() run_mock.assert_called_once_with([ 'mount', '-o', 'bind', os.path.join('/ostree', 'deploy', 'test', 'var'), os.path.join('/ostree', 'deploy', 'test', 'deploy', 'test-deploy.0', 'var'), ], check=True)
def should_not_create_stateroot_if_path_exists(): sysroot = Path('/', 'mnt', 'rootfs') cfg = Config(Source.url('url'), 'ref', stateroot='stateroot-name', sysroot=str(sysroot)) with ExitStack() as stack: exists = stack.enter_context( mock.patch('os.path.exists', mock.Mock(return_value=True))) run = stack.enter_context( mock.patch('deploy_ostree.steps.create_stateroot.run')) CreateStateroot(cfg).run() exists.assert_called_once_with('/mnt/rootfs/ostree/deploy/stateroot-name') run.assert_not_called()
def test_should_add_ostree_remote_with_absolute_path( self, mock_run: mock.Mock): cfg = Config(Source.path('repo'), 'debian/9/i386/desktop', remote='remote-name', base_dir='ostree') repo_abs_path = os.path.join(os.getcwd(), 'ostree', 'repo') FileRemote(cfg).run() mock_run.assert_called_once_with([ 'ostree', 'remote', 'add', '--repo=%s' % os.path.join('/ostree', 'repo'), '--no-gpg-verify', 'remote-name', 'file://%s' % repo_abs_path ], check=True)
def test_should_run_provisioner_script_with_specified_sysroot( self, run_mock: mock.Mock): sysroot = os.path.join('/mnt', 'rootfs') cfg = Config(Source.url('url'), 'ref', stateroot='test', sysroot=sysroot) cfg.set_deployment_name('test-deploy.0') provisioner = ProvisionerConfig('etc-fstab', {}) BuiltinProvisioner(cfg, provisioner).run() run_mock.assert_called_once_with([ os.path.join(PROVISIONERS_DIR, 'etc-fstab'), os.path.join(sysroot, 'ostree', 'deploy', 'test', 'deploy', 'test-deploy.0') ], check=True, env={})
def test_should_mount_with_specified_sysroot(self, run_mock: mock.Mock): sysroot = os.path.join('/mnt', 'rootfs') cfg = Config(Source.url('url'), 'ref', stateroot='test', sysroot=sysroot) cfg.set_deployment_name('test-deploy.0') MountVar(cfg).run() run_mock.assert_called_once_with([ 'mount', '-o', 'bind', os.path.join(sysroot, 'ostree', 'deploy', 'test', 'var'), os.path.join(sysroot, 'ostree', 'deploy', 'test', 'deploy', 'test-deploy.0', 'var'), ], check=True)
def test_should_deploy_commit_and_set_deployment_folder(self, run_mock): cfg = Config( Source.url('url'), 'fedora/28/x86_64/atomic-host', remote='fedora-atomic', stateroot='atomic-host', root_filesystem='/dev/mapper/atomic-root', ) Deploy(cfg).run() run_mock.assert_called_once_with([ 'ostree', 'admin', 'deploy', '--sysroot=/', '--os=atomic-host', 'fedora-atomic:fedora/28/x86_64/atomic-host', '--karg=root=/dev/mapper/atomic-root' ], check=True) self.assertEqual( cfg.deployment_dir, os.path.join('/ostree', 'deploy', 'atomic-host', 'deploy', 'abcdef.1'))
def test_should_deploy_into_specified_sysroot(self, listdir_mock, run_mock): sysroot = os.path.join('/mnt', 'rootfs') cfg = Config(Source.url('url'), 'ref', remote='remote', stateroot='test-stateroot', sysroot=sysroot, root_filesystem='/dev/mapper/atomic-root') listdir_mock.side_effect = [[], ['deploy', 'deploy.origin']] Deploy(cfg).run() listdir_mock.assert_called_with( os.path.join(sysroot, 'ostree', 'deploy', 'test-stateroot', 'deploy')) run_mock.assert_called_once_with([ 'ostree', 'admin', 'deploy', '--sysroot=%s' % sysroot, '--os=test-stateroot', 'remote:ref', '--karg=root=/dev/mapper/atomic-root' ], check=True)
def test_should_add_additional_kernel_args(self, run_mock): cfg = Config( Source.url('url'), 'ref', remote='remote', stateroot='os', kernel_args=['quiet', 'splash'], root_filesystem='/dev/rootfs', ) Deploy(cfg).run() run_mock.assert_called_once_with([ 'ostree', 'admin', 'deploy', '--sysroot=/', '--os=os', 'remote:ref', '--karg=root=/dev/rootfs', '--karg-append=quiet', '--karg-append=splash', ], check=True)
def test_repo_dir_should_include_sysroot(self): sysroot = os.path.join('/mnt', 'rootfs') cfg = Config(Source.url('url'), 'ref', sysroot=sysroot) self.assertEqual(cfg.ostree_repo, os.path.join(sysroot, 'ostree', 'repo'))
def test_deployment_path_should_raise_exception_if_name_is_not_set(self): cfg = Config(Source.url('url'), 'ref') with self.assertRaises(RuntimeError): cfg.deployment_dir
def test_var_dir_should_be_path_to_stateroot_var(self): cfg = Config(Source.url('url'), 'ref', stateroot='test-stateroot') self.assertEqual( cfg.var_dir, os.path.join('/ostree', 'deploy', 'test-stateroot', 'var'))
def test_repo_dir_should_default_to_system_ostree_repo(self): cfg = Config(Source.url('url'), 'ref') self.assertEqual(cfg.ostree_repo, os.path.join('/ostree', 'repo'))
def test_stateroot_should_be_randomly_generated_if_not_specified(self): cfg1 = Config(Source.url('url'), 'ref') cfg2 = Config(Source.url('url'), 'ref') self.assertNotEqual(cfg1.stateroot, cfg2.stateroot)
def test_fstab_should_default_to_system_fstab(): cfg = Config(Source.url('url'), 'ref') assert cfg.fstab == Path('/', 'etc', 'fstab')
def test_default_root_filesystem_should_be_determined_by_get_root_fs(self): cfg = Config(Source.url('url'), 'ref') self.assertEqual(cfg.root_filesystem, '/dev/sda1')