def test_upgrade_policy_json_file_log_warning(self, mock_log):
test_mgr = stevedore.named.NamedExtensionManager.make_test_instance(
extensions=self.extensions, namespace='test_upgrade')
with mock.patch('stevedore.named.NamedExtensionManager',
return_value=test_mgr):
testargs = [
'olsopolicy-policy-upgrade', '--policy',
self.get_config_file_fullname('policy.json'), '--namespace',
'test_upgrade', '--output-file',
self.get_config_file_fullname('new_policy.json'), '--format',
'json'
]
with mock.patch('sys.argv', testargs):
generator.upgrade_policy(conf=self.local_conf)
mock_log.warning.assert_any_call(policy.WARN_JSON)
def test_upgrade_policy_yaml_stdout(self):
test_mgr = stevedore.named.NamedExtensionManager.make_test_instance(
extensions=self.extensions, namespace='test_upgrade')
stdout = self._capture_stdout()
with mock.patch('stevedore.named.NamedExtensionManager',
return_value=test_mgr):
testargs = [
'olsopolicy-policy-upgrade', '--policy',
self.get_config_file_fullname('policy.json'), '--namespace',
'test_upgrade', '--format', 'yaml'
]
with mock.patch('sys.argv', testargs):
generator.upgrade_policy(conf=self.local_conf)
expected = '''new_policy_name: rule:admin
'''
self.assertEqual(expected, stdout.getvalue())
def test_upgrade_policy_yaml_file(self):
test_mgr = stevedore.named.NamedExtensionManager.make_test_instance(
extensions=self.extensions, namespace='test_upgrade')
with mock.patch('stevedore.named.NamedExtensionManager',
return_value=test_mgr):
testargs = [
'olsopolicy-policy-upgrade', '--policy',
self.get_config_file_fullname('policy.json'), '--namespace',
'test_upgrade', '--output-file',
self.get_config_file_fullname('new_policy.yaml'), '--format',
'yaml'
]
with mock.patch('sys.argv', testargs):
generator.upgrade_policy()
new_file = self.get_config_file_fullname('new_policy.yaml')
new_policy = yaml.safe_load(open(new_file, 'r'))
self.assertIsNotNone(new_policy.get('new_policy_name'))
self.assertIsNone(new_policy.get('deprecated_name'))
