github/ha-core
1
Fork 0
mirror of https://github.com/home-assistant/core synced 2024-07-12 07:21:24 +02:00
Code Releases Activity

Merge pull request #2068 from home-assistant/yaml_env

Browse Source 
Add test for yaml enviroment
This commit is contained in:
Daniel Høyer Iversen 2016-05-14 20:33:46 +02:00
commit 0626a80186
1 changed files with 21 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

21
tests/util/test_yaml.py
View File

@ -1,6 +1,7 @@
"""Test Home Assistant yaml loader."""
import io
import unittest
import os
from homeassistant.util import yaml
@ -32,3 +33,23 @@ class TestYaml(unittest.TestCase):
pass
else:
assert 0
def test_enviroment_variable(self):
"""Test config file with enviroment variable."""
os.environ["PASSWORD"] = "secret_password"
conf = "password: !env_var PASSWORD"
with io.StringIO(conf) as f:
doc = yaml.yaml.safe_load(f)
assert doc['password'] == "secret_password"
del os.environ["PASSWORD"]
def test_invalid_enviroment_variable(self):
"""Test config file with no enviroment variable sat."""
conf = "password: !env_var PASSWORD"
try:
with io.StringIO(conf) as f:
yaml.yaml.safe_load(f)
except Exception:
pass
else:
assert 0