add test for unsafe yaml

This commit is contained in:
Erik 2016-02-02 08:41:18 +01:00
parent 26efaa91a3
commit 00d1cab091

View File

@ -94,6 +94,15 @@ class TestConfig(unittest.TestCase):
with self.assertRaises(HomeAssistantError):
config_util.load_yaml_config_file(YAML_PATH)
def test_load_yaml_config_raises_error_if_unsafe_yaml(self):
""" Test error raised if unsafe YAML. """
with open(YAML_PATH, 'w') as f:
f.write('hello: !!python/object/apply:os.system')
with self.assertRaises(HomeAssistantError):
config_util.load_yaml_config_file(YAML_PATH)
def test_load_yaml_config_preserves_key_order(self):
with open(YAML_PATH, 'w') as f:
f.write('hello: 0\n')