Merge pull request #1097 from molobrakos/add-test

add test for unsafe yaml
pull/1117/head
Paulus Schoutsen 2016-02-02 00:22:19 -08:00
commit 35613d7fbf
1 changed files with 9 additions and 0 deletions

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')