388 lines
15 KiB
Python
388 lines
15 KiB
Python
"""Test Home Assistant yaml loader."""
|
|
import io
|
|
import os
|
|
import unittest
|
|
from unittest.mock import patch
|
|
|
|
from homeassistant.exceptions import HomeAssistantError
|
|
from homeassistant.util import yaml
|
|
from homeassistant.config import YAML_CONFIG_FILE, load_yaml_config_file
|
|
from tests.common import get_test_config_dir, patch_yaml_files
|
|
|
|
|
|
class TestYaml(unittest.TestCase):
|
|
"""Test util.yaml loader."""
|
|
|
|
# pylint: disable=no-self-use, invalid-name
|
|
|
|
def test_simple_list(self):
|
|
"""Test simple list."""
|
|
conf = "config:\n - simple\n - list"
|
|
with io.StringIO(conf) as file:
|
|
doc = yaml.yaml.safe_load(file)
|
|
assert doc['config'] == ["simple", "list"]
|
|
|
|
def test_simple_dict(self):
|
|
"""Test simple dict."""
|
|
conf = "key: value"
|
|
with io.StringIO(conf) as file:
|
|
doc = yaml.yaml.safe_load(file)
|
|
assert doc['key'] == 'value'
|
|
|
|
def test_duplicate_key(self):
|
|
"""Test duplicate dict keys."""
|
|
files = {YAML_CONFIG_FILE: 'key: thing1\nkey: thing2'}
|
|
with self.assertRaises(HomeAssistantError):
|
|
with patch_yaml_files(files):
|
|
load_yaml_config_file(YAML_CONFIG_FILE)
|
|
|
|
def test_unhashable_key(self):
|
|
"""Test an unhasable key."""
|
|
files = {YAML_CONFIG_FILE: 'message:\n {{ states.state }}'}
|
|
with self.assertRaises(HomeAssistantError), \
|
|
patch_yaml_files(files):
|
|
load_yaml_config_file(YAML_CONFIG_FILE)
|
|
|
|
def test_no_key(self):
|
|
"""Test item without an key."""
|
|
files = {YAML_CONFIG_FILE: 'a: a\nnokeyhere'}
|
|
with self.assertRaises(HomeAssistantError), \
|
|
patch_yaml_files(files):
|
|
yaml.load_yaml(YAML_CONFIG_FILE)
|
|
|
|
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 file:
|
|
doc = yaml.yaml.safe_load(file)
|
|
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"
|
|
with self.assertRaises(HomeAssistantError):
|
|
with io.StringIO(conf) as file:
|
|
yaml.yaml.safe_load(file)
|
|
|
|
def test_include_yaml(self):
|
|
"""Test include yaml."""
|
|
with patch_yaml_files({'test.yaml': 'value'}):
|
|
conf = 'key: !include test.yaml'
|
|
with io.StringIO(conf) as file:
|
|
doc = yaml.yaml.safe_load(file)
|
|
assert doc["key"] == "value"
|
|
|
|
with patch_yaml_files({'test.yaml': None}):
|
|
conf = 'key: !include test.yaml'
|
|
with io.StringIO(conf) as file:
|
|
doc = yaml.yaml.safe_load(file)
|
|
assert doc["key"] == {}
|
|
|
|
@patch('homeassistant.util.yaml.os.walk')
|
|
def test_include_dir_list(self, mock_walk):
|
|
"""Test include dir list yaml."""
|
|
mock_walk.return_value = [
|
|
['/tmp', [], ['one.yaml', 'two.yaml']],
|
|
]
|
|
|
|
with patch_yaml_files({
|
|
'/tmp/one.yaml': 'one',
|
|
'/tmp/two.yaml': 'two',
|
|
}):
|
|
conf = "key: !include_dir_list /tmp"
|
|
with io.StringIO(conf) as file:
|
|
doc = yaml.yaml.safe_load(file)
|
|
assert sorted(doc["key"]) == sorted(["one", "two"])
|
|
|
|
@patch('homeassistant.util.yaml.os.walk')
|
|
def test_include_dir_list_recursive(self, mock_walk):
|
|
"""Test include dir recursive list yaml."""
|
|
mock_walk.return_value = [
|
|
['/tmp', ['tmp2', '.ignore', 'ignore'], ['zero.yaml']],
|
|
['/tmp/tmp2', [], ['one.yaml', 'two.yaml']],
|
|
['/tmp/ignore', [], ['.ignore.yaml']]
|
|
]
|
|
|
|
with patch_yaml_files({
|
|
'/tmp/zero.yaml': 'zero',
|
|
'/tmp/tmp2/one.yaml': 'one',
|
|
'/tmp/tmp2/two.yaml': 'two'
|
|
}):
|
|
conf = "key: !include_dir_list /tmp"
|
|
with io.StringIO(conf) as file:
|
|
assert '.ignore' in mock_walk.return_value[0][1], \
|
|
"Expecting .ignore in here"
|
|
doc = yaml.yaml.safe_load(file)
|
|
assert 'tmp2' in mock_walk.return_value[0][1]
|
|
assert '.ignore' not in mock_walk.return_value[0][1]
|
|
assert sorted(doc["key"]) == sorted(["zero", "one", "two"])
|
|
|
|
@patch('homeassistant.util.yaml.os.walk')
|
|
def test_include_dir_named(self, mock_walk):
|
|
"""Test include dir named yaml."""
|
|
mock_walk.return_value = [
|
|
['/tmp', [], ['first.yaml', 'second.yaml']]
|
|
]
|
|
|
|
with patch_yaml_files({
|
|
'/tmp/first.yaml': 'one',
|
|
'/tmp/second.yaml': 'two'
|
|
}):
|
|
conf = "key: !include_dir_named /tmp"
|
|
correct = {'first': 'one', 'second': 'two'}
|
|
with io.StringIO(conf) as file:
|
|
doc = yaml.yaml.safe_load(file)
|
|
assert doc["key"] == correct
|
|
|
|
@patch('homeassistant.util.yaml.os.walk')
|
|
def test_include_dir_named_recursive(self, mock_walk):
|
|
"""Test include dir named yaml."""
|
|
mock_walk.return_value = [
|
|
['/tmp', ['tmp2', '.ignore', 'ignore'], ['first.yaml']],
|
|
['/tmp/tmp2', [], ['second.yaml', 'third.yaml']],
|
|
['/tmp/ignore', [], ['.ignore.yaml']]
|
|
]
|
|
|
|
with patch_yaml_files({
|
|
'/tmp/first.yaml': 'one',
|
|
'/tmp/tmp2/second.yaml': 'two',
|
|
'/tmp/tmp2/third.yaml': 'three'
|
|
}):
|
|
conf = "key: !include_dir_named /tmp"
|
|
correct = {'first': 'one', 'second': 'two', 'third': 'three'}
|
|
with io.StringIO(conf) as file:
|
|
assert '.ignore' in mock_walk.return_value[0][1], \
|
|
"Expecting .ignore in here"
|
|
doc = yaml.yaml.safe_load(file)
|
|
assert 'tmp2' in mock_walk.return_value[0][1]
|
|
assert '.ignore' not in mock_walk.return_value[0][1]
|
|
assert doc["key"] == correct
|
|
|
|
@patch('homeassistant.util.yaml.os.walk')
|
|
def test_include_dir_merge_list(self, mock_walk):
|
|
"""Test include dir merge list yaml."""
|
|
mock_walk.return_value = [['/tmp', [], ['first.yaml', 'second.yaml']]]
|
|
|
|
with patch_yaml_files({
|
|
'/tmp/first.yaml': '- one',
|
|
'/tmp/second.yaml': '- two\n- three'
|
|
}):
|
|
conf = "key: !include_dir_merge_list /tmp"
|
|
with io.StringIO(conf) as file:
|
|
doc = yaml.yaml.safe_load(file)
|
|
assert sorted(doc["key"]) == sorted(["one", "two", "three"])
|
|
|
|
@patch('homeassistant.util.yaml.os.walk')
|
|
def test_include_dir_merge_list_recursive(self, mock_walk):
|
|
"""Test include dir merge list yaml."""
|
|
mock_walk.return_value = [
|
|
['/tmp', ['tmp2', '.ignore', 'ignore'], ['first.yaml']],
|
|
['/tmp/tmp2', [], ['second.yaml', 'third.yaml']],
|
|
['/tmp/ignore', [], ['.ignore.yaml']]
|
|
]
|
|
|
|
with patch_yaml_files({
|
|
'/tmp/first.yaml': '- one',
|
|
'/tmp/tmp2/second.yaml': '- two',
|
|
'/tmp/tmp2/third.yaml': '- three\n- four'
|
|
}):
|
|
conf = "key: !include_dir_merge_list /tmp"
|
|
with io.StringIO(conf) as file:
|
|
assert '.ignore' in mock_walk.return_value[0][1], \
|
|
"Expecting .ignore in here"
|
|
doc = yaml.yaml.safe_load(file)
|
|
assert 'tmp2' in mock_walk.return_value[0][1]
|
|
assert '.ignore' not in mock_walk.return_value[0][1]
|
|
assert sorted(doc["key"]) == sorted(["one", "two",
|
|
"three", "four"])
|
|
|
|
@patch('homeassistant.util.yaml.os.walk')
|
|
def test_include_dir_merge_named(self, mock_walk):
|
|
"""Test include dir merge named yaml."""
|
|
mock_walk.return_value = [['/tmp', [], ['first.yaml', 'second.yaml']]]
|
|
|
|
files = {
|
|
'/tmp/first.yaml': 'key1: one',
|
|
'/tmp/second.yaml': 'key2: two\nkey3: three',
|
|
}
|
|
|
|
with patch_yaml_files(files):
|
|
conf = "key: !include_dir_merge_named /tmp"
|
|
with io.StringIO(conf) as file:
|
|
doc = yaml.yaml.safe_load(file)
|
|
assert doc["key"] == {
|
|
"key1": "one",
|
|
"key2": "two",
|
|
"key3": "three"
|
|
}
|
|
|
|
@patch('homeassistant.util.yaml.os.walk')
|
|
def test_include_dir_merge_named_recursive(self, mock_walk):
|
|
"""Test include dir merge named yaml."""
|
|
mock_walk.return_value = [
|
|
['/tmp', ['tmp2', '.ignore', 'ignore'], ['first.yaml']],
|
|
['/tmp/tmp2', [], ['second.yaml', 'third.yaml']],
|
|
['/tmp/ignore', [], ['.ignore.yaml']]
|
|
]
|
|
|
|
with patch_yaml_files({
|
|
'/tmp/first.yaml': 'key1: one',
|
|
'/tmp/tmp2/second.yaml': 'key2: two',
|
|
'/tmp/tmp2/third.yaml': 'key3: three\nkey4: four'
|
|
}):
|
|
conf = "key: !include_dir_merge_named /tmp"
|
|
with io.StringIO(conf) as file:
|
|
assert '.ignore' in mock_walk.return_value[0][1], \
|
|
"Expecting .ignore in here"
|
|
doc = yaml.yaml.safe_load(file)
|
|
assert 'tmp2' in mock_walk.return_value[0][1]
|
|
assert '.ignore' not in mock_walk.return_value[0][1]
|
|
assert doc["key"] == {
|
|
"key1": "one",
|
|
"key2": "two",
|
|
"key3": "three",
|
|
"key4": "four"
|
|
}
|
|
|
|
@patch('homeassistant.util.yaml.open', create=True)
|
|
def test_load_yaml_encoding_error(self, mock_open):
|
|
"""Test raising a UnicodeDecodeError."""
|
|
mock_open.side_effect = UnicodeDecodeError('', b'', 1, 0, '')
|
|
self.assertRaises(HomeAssistantError, yaml.load_yaml, 'test')
|
|
|
|
def test_dump(self):
|
|
"""The that the dump method returns empty None values."""
|
|
assert yaml.dump({'a': None, 'b': 'b'}) == 'a:\nb: b\n'
|
|
|
|
|
|
FILES = {}
|
|
|
|
|
|
def load_yaml(fname, string):
|
|
"""Write a string to file and return the parsed yaml."""
|
|
FILES[fname] = string
|
|
with patch_yaml_files(FILES):
|
|
return load_yaml_config_file(fname)
|
|
|
|
|
|
class FakeKeyring():
|
|
"""Fake a keyring class."""
|
|
|
|
def __init__(self, secrets_dict):
|
|
"""Store keyring dictionary."""
|
|
self._secrets = secrets_dict
|
|
|
|
# pylint: disable=protected-access
|
|
def get_password(self, domain, name):
|
|
"""Retrieve password."""
|
|
assert domain == yaml._SECRET_NAMESPACE
|
|
return self._secrets.get(name)
|
|
|
|
|
|
class TestSecrets(unittest.TestCase):
|
|
"""Test the secrets parameter in the yaml utility."""
|
|
|
|
# pylint: disable=protected-access,invalid-name
|
|
|
|
def setUp(self):
|
|
"""Create & load secrets file."""
|
|
config_dir = get_test_config_dir()
|
|
yaml.clear_secret_cache()
|
|
self._yaml_path = os.path.join(config_dir, YAML_CONFIG_FILE)
|
|
self._secret_path = os.path.join(config_dir, yaml._SECRET_YAML)
|
|
self._sub_folder_path = os.path.join(config_dir, 'subFolder')
|
|
self._unrelated_path = os.path.join(config_dir, 'unrelated')
|
|
|
|
load_yaml(self._secret_path,
|
|
'http_pw: pwhttp\n'
|
|
'comp1_un: un1\n'
|
|
'comp1_pw: pw1\n'
|
|
'stale_pw: not_used\n'
|
|
'logger: debug\n')
|
|
self._yaml = load_yaml(self._yaml_path,
|
|
'http:\n'
|
|
' api_password: !secret http_pw\n'
|
|
'component:\n'
|
|
' username: !secret comp1_un\n'
|
|
' password: !secret comp1_pw\n'
|
|
'')
|
|
|
|
def tearDown(self):
|
|
"""Clean up secrets."""
|
|
yaml.clear_secret_cache()
|
|
FILES.clear()
|
|
|
|
def test_secrets_from_yaml(self):
|
|
"""Did secrets load ok."""
|
|
expected = {'api_password': 'pwhttp'}
|
|
self.assertEqual(expected, self._yaml['http'])
|
|
|
|
expected = {
|
|
'username': 'un1',
|
|
'password': 'pw1'}
|
|
self.assertEqual(expected, self._yaml['component'])
|
|
|
|
def test_secrets_from_parent_folder(self):
|
|
"""Test loading secrets from parent foler."""
|
|
expected = {'api_password': 'pwhttp'}
|
|
self._yaml = load_yaml(os.path.join(self._sub_folder_path, 'sub.yaml'),
|
|
'http:\n'
|
|
' api_password: !secret http_pw\n'
|
|
'component:\n'
|
|
' username: !secret comp1_un\n'
|
|
' password: !secret comp1_pw\n'
|
|
'')
|
|
|
|
self.assertEqual(expected, self._yaml['http'])
|
|
|
|
def test_secret_overrides_parent(self):
|
|
"""Test loading current directory secret overrides the parent."""
|
|
expected = {'api_password': 'override'}
|
|
load_yaml(os.path.join(self._sub_folder_path, yaml._SECRET_YAML),
|
|
'http_pw: override')
|
|
self._yaml = load_yaml(os.path.join(self._sub_folder_path, 'sub.yaml'),
|
|
'http:\n'
|
|
' api_password: !secret http_pw\n'
|
|
'component:\n'
|
|
' username: !secret comp1_un\n'
|
|
' password: !secret comp1_pw\n'
|
|
'')
|
|
|
|
self.assertEqual(expected, self._yaml['http'])
|
|
|
|
def test_secrets_from_unrelated_fails(self):
|
|
"""Test loading secrets from unrelated folder fails."""
|
|
load_yaml(os.path.join(self._unrelated_path, yaml._SECRET_YAML),
|
|
'test: failure')
|
|
with self.assertRaises(HomeAssistantError):
|
|
load_yaml(os.path.join(self._sub_folder_path, 'sub.yaml'),
|
|
'http:\n'
|
|
' api_password: !secret test')
|
|
|
|
def test_secrets_keyring(self):
|
|
"""Test keyring fallback & get_password."""
|
|
yaml.keyring = None # Ensure its not there
|
|
yaml_str = 'http:\n api_password: !secret http_pw_keyring'
|
|
with self.assertRaises(yaml.HomeAssistantError):
|
|
load_yaml(self._yaml_path, yaml_str)
|
|
|
|
yaml.keyring = FakeKeyring({'http_pw_keyring': 'yeah'})
|
|
_yaml = load_yaml(self._yaml_path, yaml_str)
|
|
self.assertEqual({'http': {'api_password': 'yeah'}}, _yaml)
|
|
|
|
def test_secrets_logger_removed(self):
|
|
"""Ensure logger: debug was removed."""
|
|
with self.assertRaises(yaml.HomeAssistantError):
|
|
load_yaml(self._yaml_path, 'api_password: !secret logger')
|
|
|
|
@patch('homeassistant.util.yaml._LOGGER.error')
|
|
def test_bad_logger_value(self, mock_error):
|
|
"""Ensure logger: debug was removed."""
|
|
yaml.clear_secret_cache()
|
|
load_yaml(self._secret_path, 'logger: info\npw: abc')
|
|
load_yaml(self._yaml_path, 'api_password: !secret pw')
|
|
assert mock_error.call_count == 1, \
|
|
"Expected an error about logger: value"
|