def test_unload_policy_file_with_bad_schema(self, os_unlink, _1, _2, _3,
_4, _5):
host = Host(DEFAULT_EVENT_QUEUE_ID, DEFAULT_EVENT_QUEUE_NAME,
DEFAULT_POLICY_STORAGE)
# Schedule is invalid
policy_string = """
policie:
- name: foo
mode:
type: container-periodic
schedule: '* * * *'
resource: azure.resourcegroup
"""
# Create a bad yaml file
file_path = tempfile.mktemp(suffix=".yaml")
with open(file_path, 'w') as f:
f.write(policy_string)
host.load_policy(file_path, host.policies)
host.unload_policy_file(file_path, host.policies)
os_unlink.assert_called()
# Clean up the file
os.remove(file_path)
def test_unload_policy_file_with_yaml_error(self, os_unlink,
yaml_safe_load, _1, _2, _3, _4,
_5):
host = Host(DEFAULT_EVENT_QUEUE_ID, DEFAULT_EVENT_QUEUE_NAME,
DEFAULT_POLICY_STORAGE)
# Create a bad yaml file
file_path = tempfile.mktemp(suffix=".yaml")
with open(file_path, 'w') as f:
f.write("bad yaml file")
host.load_policy(file_path, host.policies)
host.unload_policy_file(file_path, host.policies)
os_unlink.assert_called()
# Clean up the file
os.remove(file_path)
def test_unload_policy_file_without_name(self, os_unlink, yaml_safe_load,
_1, _2, _3, _4):
host = Host(DEFAULT_EVENT_QUEUE_ID, DEFAULT_EVENT_QUEUE_NAME,
DEFAULT_POLICY_STORAGE)
# Create a bad yaml file (no name field)
file_path = tempfile.mktemp(suffix=".yaml")
with open(file_path, 'w') as f:
f.write("""
policies:
- mode:
type: container-periodic
schedule: '* * * * *'
resource: azure.resourcegroup
""")
host.load_policy(file_path, host.policies)
host.unload_policy_file(file_path, host.policies)
os_unlink.assert_called()
# Clean up the file
os.remove(file_path)