mirror of
https://github.com/esphome/esphome.git
synced 2025-09-01 19:02:18 +01:00
65 lines
2.1 KiB
Python
65 lines
2.1 KiB
Python
from esphome import yaml_util
|
|
from esphome.components import substitutions
|
|
from esphome.core import EsphomeError
|
|
|
|
|
|
def test_include_with_vars(fixture_path):
|
|
yaml_file = fixture_path / "yaml_util" / "includetest.yaml"
|
|
|
|
actual = yaml_util.load_yaml(yaml_file)
|
|
substitutions.do_substitution_pass(actual, None)
|
|
assert actual["esphome"]["name"] == "original"
|
|
assert actual["esphome"]["libraries"][0] == "Wire"
|
|
assert actual["esp8266"]["board"] == "nodemcu"
|
|
assert actual["wifi"]["ssid"] == "my_custom_ssid"
|
|
|
|
|
|
def test_loading_a_broken_yaml_file(fixture_path):
|
|
"""Ensure we fallback to pure python to give good errors."""
|
|
yaml_file = fixture_path / "yaml_util" / "broken_includetest.yaml"
|
|
|
|
try:
|
|
yaml_util.load_yaml(yaml_file)
|
|
except EsphomeError as err:
|
|
assert "broken_included.yaml" in str(err)
|
|
|
|
|
|
def test_loading_a_yaml_file_with_a_missing_component(fixture_path):
|
|
"""Ensure we show the filename for a yaml file with a missing component."""
|
|
yaml_file = fixture_path / "yaml_util" / "missing_comp.yaml"
|
|
|
|
try:
|
|
yaml_util.load_yaml(yaml_file)
|
|
except EsphomeError as err:
|
|
assert "missing_comp.yaml" in str(err)
|
|
|
|
|
|
def test_loading_a_missing_file(fixture_path):
|
|
"""We throw EsphomeError when loading a missing file."""
|
|
yaml_file = fixture_path / "yaml_util" / "missing.yaml"
|
|
|
|
try:
|
|
yaml_util.load_yaml(yaml_file)
|
|
except EsphomeError as err:
|
|
assert "missing.yaml" in str(err)
|
|
|
|
|
|
def test_parsing_with_custom_loader(fixture_path):
|
|
"""Test custom loader used for vscode connection
|
|
Default loader is tested in test_include_with_vars
|
|
"""
|
|
yaml_file = fixture_path / "yaml_util" / "includetest.yaml"
|
|
|
|
loader_calls = []
|
|
|
|
def custom_loader(fname):
|
|
loader_calls.append(fname)
|
|
|
|
with open(yaml_file, encoding="utf-8") as f_handle:
|
|
yaml_util.parse_yaml(yaml_file, f_handle, custom_loader)
|
|
|
|
assert len(loader_calls) == 3
|
|
assert loader_calls[0].endswith("includes/included.yaml")
|
|
assert loader_calls[1].endswith("includes/list.yaml")
|
|
assert loader_calls[2].endswith("includes/scalar.yaml")
|