def setUp(self): super(TestFixPuppetManifestHook, self).setUp() conf = self.fake_config conf.from_version = '5.0' self.hook = FixPuppetManifests(self.upgraders, conf)
class TestFixPuppetManifestHook(TestPreUpgradeHooksBase): iterfiles_returns = [ '/tmp/upgrade_path/config/5.0/modules/package/lib/puppet' '/provider/package/yum.rb', '/tmp/upgrade_path/config/5.0/manifests/centos-versions.yaml' ] def setUp(self): super(TestFixPuppetManifestHook, self).setUp() conf = self.fake_config conf.from_version = '5.0' self.hook = FixPuppetManifests(self.upgraders, conf) def test_is_required_returns_true(self): self.hook.config.from_version = '5.0' self.assertTrue(self.hook.check_if_required()) self.hook.config.from_version = '5.0.1' self.assertTrue(self.hook.check_if_required()) def test_is_required_returns_false(self): self.hook.config.from_version = '5.1' self.assertFalse(self.hook.check_if_required()) @mock.patch( 'fuel_upgrade.pre_upgrade_hooks.from_5_0_to_any_fix_puppet_manifests.' 'iterfiles', return_value=iterfiles_returns) @mock.patch( 'fuel_upgrade.pre_upgrade_hooks.from_5_0_to_any_fix_puppet_manifests.' 'copy') def test_run(self, copy, _): self.hook.run() copy.assert_has_calls([ mock.call( '/tmp/upgrade_path/config/5.0/modules/package/lib' '/puppet/provider/package/yum.rb', '/etc/puppet/modules/package/lib/puppet/provider/package' '/yum.rb'), mock.call( '/tmp/upgrade_path/config/5.0/manifests' '/centos-versions.yaml', '/etc/puppet/manifests/centos-versions.yaml') ])
class TestFixPuppetManifestHook(TestPreUpgradeHooksBase): iterfiles_returns = [ '/tmp/upgrade_path/config/5.0/modules/package/lib/puppet' '/provider/package/yum.rb', '/tmp/upgrade_path/config/5.0/manifests/centos-versions.yaml'] def setUp(self): super(TestFixPuppetManifestHook, self).setUp() conf = self.fake_config conf.from_version = '5.0' self.hook = FixPuppetManifests(self.upgraders, conf) def test_is_required_returns_true(self): self.hook.config.from_version = '5.0' self.assertTrue(self.hook.check_if_required()) self.hook.config.from_version = '5.0.1' self.assertTrue(self.hook.check_if_required()) def test_is_required_returns_false(self): self.hook.config.from_version = '5.1' self.assertFalse(self.hook.check_if_required()) @mock.patch( 'fuel_upgrade.pre_upgrade_hooks.from_5_0_to_any_fix_puppet_manifests.' 'iterfiles', return_value=iterfiles_returns) @mock.patch( 'fuel_upgrade.pre_upgrade_hooks.from_5_0_to_any_fix_puppet_manifests.' 'copy') def test_run(self, copy, _): self.hook.run() copy.assert_has_calls([ mock.call( '/tmp/upgrade_path/config/5.0/modules/package/lib' '/puppet/provider/package/yum.rb', '/etc/puppet/modules/package/lib/puppet/provider/package' '/yum.rb'), mock.call( '/tmp/upgrade_path/config/5.0/manifests' '/centos-versions.yaml', '/etc/puppet/manifests/centos-versions.yaml')])