mirror of
https://github.com/esphome/esphome.git
synced 2025-09-28 16:12:24 +01:00
218 lines
7.2 KiB
Python
218 lines
7.2 KiB
Python
"""Unit tests for esphome.config module."""
|
|
|
|
from collections.abc import Generator
|
|
from pathlib import Path
|
|
from unittest.mock import MagicMock, Mock, patch
|
|
|
|
import pytest
|
|
|
|
from esphome import config, yaml_util
|
|
from esphome.core import CORE
|
|
|
|
|
|
@pytest.fixture
|
|
def mock_get_component() -> Generator[Mock, None, None]:
|
|
"""Fixture for mocking get_component."""
|
|
with patch("esphome.config.get_component") as mock_get_component:
|
|
yield mock_get_component
|
|
|
|
|
|
@pytest.fixture
|
|
def mock_get_platform() -> Generator[Mock, None, None]:
|
|
"""Fixture for mocking get_platform."""
|
|
with patch("esphome.config.get_platform") as mock_get_platform:
|
|
# Default mock platform
|
|
mock_get_platform.return_value = MagicMock()
|
|
yield mock_get_platform
|
|
|
|
|
|
@pytest.fixture
|
|
def fixtures_dir() -> Path:
|
|
"""Get the fixtures directory."""
|
|
return Path(__file__).parent / "fixtures"
|
|
|
|
|
|
def test_iter_components_handles_non_list_platform_component(
|
|
mock_get_component: Mock,
|
|
) -> None:
|
|
"""Test that iter_components handles platform components that have been normalized to empty list."""
|
|
test_config = {
|
|
"ota": [], # Normalized from None/dict to empty list by LoadValidationStep
|
|
"wifi": {"ssid": "test"},
|
|
}
|
|
|
|
components = {
|
|
"ota": MagicMock(is_platform_component=True),
|
|
"wifi": MagicMock(is_platform_component=False),
|
|
}
|
|
|
|
mock_get_component.side_effect = lambda domain: components.get(
|
|
domain, MagicMock(is_platform_component=False)
|
|
)
|
|
|
|
components_list = list(config.iter_components(test_config))
|
|
assert len(components_list) == 2 # ota and wifi (ota has no platforms)
|
|
|
|
|
|
def test_iter_component_configs_handles_non_list_platform_component(
|
|
mock_get_component: Mock,
|
|
mock_get_platform: Mock,
|
|
) -> None:
|
|
"""Test that iter_component_configs handles platform components that have been normalized."""
|
|
test_config = {
|
|
"ota": [], # Normalized from None/dict to empty list by LoadValidationStep
|
|
"one_wire": [{"platform": "gpio", "pin": 10}],
|
|
}
|
|
|
|
components: dict[str, Mock] = {
|
|
"ota": MagicMock(is_platform_component=True, multi_conf=False),
|
|
"one_wire": MagicMock(is_platform_component=True, multi_conf=False),
|
|
}
|
|
|
|
default_mock = MagicMock(is_platform_component=False, multi_conf=False)
|
|
mock_get_component.side_effect = lambda domain: components.get(domain, default_mock)
|
|
|
|
configs = list(config.iter_component_configs(test_config))
|
|
assert len(configs) == 3
|
|
|
|
domains = [c[0] for c in configs]
|
|
assert "ota" in domains
|
|
assert "one_wire" in domains
|
|
assert "one_wire.gpio" in domains
|
|
|
|
|
|
def test_iter_components_with_valid_platform_list(
|
|
mock_get_component: Mock,
|
|
mock_get_platform: Mock,
|
|
) -> None:
|
|
"""Test that iter_components works correctly with valid platform component list."""
|
|
mock_component = MagicMock()
|
|
mock_component.is_platform_component = True
|
|
|
|
test_config = {
|
|
"sensor": [
|
|
{"platform": "dht", "pin": 5},
|
|
{"platform": "bme280", "address": 0x76},
|
|
],
|
|
}
|
|
|
|
mock_get_component.return_value = mock_component
|
|
|
|
components = list(config.iter_components(test_config))
|
|
assert len(components) == 3
|
|
|
|
domains = [c[0] for c in components]
|
|
assert "sensor" in domains
|
|
assert "sensor.dht" in domains
|
|
assert "sensor.bme280" in domains
|
|
|
|
|
|
def test_ota_with_proper_platform_list(
|
|
mock_get_component: Mock,
|
|
mock_get_platform: Mock,
|
|
) -> None:
|
|
"""Test that OTA component works correctly when configured as a list with platforms."""
|
|
test_config = {
|
|
"ota": [
|
|
{"platform": "esphome", "password": "test123"},
|
|
],
|
|
}
|
|
|
|
mock_get_component.return_value = MagicMock(is_platform_component=True)
|
|
components = list(config.iter_components(test_config))
|
|
|
|
assert len(components) == 2
|
|
domains = [c[0] for c in components]
|
|
assert "ota" in domains
|
|
assert "ota.esphome" in domains
|
|
|
|
|
|
def test_ota_component_configs_with_proper_platform_list(
|
|
mock_get_component: Mock,
|
|
mock_get_platform: Mock,
|
|
) -> None:
|
|
"""Test that iter_component_configs handles OTA properly configured as a list."""
|
|
test_config = {
|
|
"ota": [
|
|
{"platform": "esphome", "password": "test123", "id": "my_ota"},
|
|
],
|
|
}
|
|
|
|
mock_get_component.return_value = MagicMock(
|
|
is_platform_component=True, multi_conf=False
|
|
)
|
|
|
|
configs = list(config.iter_component_configs(test_config))
|
|
assert len(configs) == 2
|
|
|
|
assert configs[0][0] == "ota"
|
|
assert configs[0][2] == test_config["ota"] # The list itself
|
|
|
|
assert configs[1][0] == "ota.esphome"
|
|
assert configs[1][2]["platform"] == "esphome"
|
|
assert configs[1][2]["password"] == "test123"
|
|
|
|
|
|
def test_iter_component_configs_with_multi_conf(mock_get_component: Mock) -> None:
|
|
"""Test that iter_component_configs handles multi_conf components correctly."""
|
|
test_config = {
|
|
"switch": [
|
|
{"name": "Switch 1"},
|
|
{"name": "Switch 2"},
|
|
],
|
|
}
|
|
|
|
mock_get_component.return_value = MagicMock(
|
|
is_platform_component=False, multi_conf=True
|
|
)
|
|
|
|
configs = list(config.iter_component_configs(test_config))
|
|
assert len(configs) == 2
|
|
|
|
for domain, component, conf in configs:
|
|
assert domain == "switch"
|
|
assert "name" in conf
|
|
|
|
|
|
def test_ota_no_platform_with_captive_portal(fixtures_dir: Path) -> None:
|
|
"""Test OTA with no platform (ota:) gets normalized when captive_portal auto-loads."""
|
|
CORE.config_path = fixtures_dir / "dummy.yaml"
|
|
|
|
config_file = fixtures_dir / "ota_no_platform.yaml"
|
|
raw_config = yaml_util.load_yaml(config_file)
|
|
result = config.validate_config(raw_config, {})
|
|
|
|
assert "ota" in result
|
|
assert isinstance(result["ota"], list), f"Expected list, got {type(result['ota'])}"
|
|
platforms = {p.get("platform") for p in result["ota"]}
|
|
assert "web_server" in platforms, f"Expected web_server platform in {platforms}"
|
|
|
|
|
|
def test_ota_empty_dict_with_captive_portal(fixtures_dir: Path) -> None:
|
|
"""Test OTA with empty dict ({}) gets normalized when captive_portal auto-loads."""
|
|
CORE.config_path = fixtures_dir / "dummy.yaml"
|
|
|
|
config_file = fixtures_dir / "ota_empty_dict.yaml"
|
|
raw_config = yaml_util.load_yaml(config_file)
|
|
result = config.validate_config(raw_config, {})
|
|
|
|
assert "ota" in result
|
|
assert isinstance(result["ota"], list), f"Expected list, got {type(result['ota'])}"
|
|
platforms = {p.get("platform") for p in result["ota"]}
|
|
assert "web_server" in platforms, f"Expected web_server platform in {platforms}"
|
|
|
|
|
|
def test_ota_with_platform_list_and_captive_portal(fixtures_dir: Path) -> None:
|
|
"""Test OTA with proper platform list remains valid when captive_portal auto-loads."""
|
|
CORE.config_path = fixtures_dir / "dummy.yaml"
|
|
|
|
config_file = fixtures_dir / "ota_with_platform_list.yaml"
|
|
raw_config = yaml_util.load_yaml(config_file)
|
|
result = config.validate_config(raw_config, {})
|
|
|
|
assert "ota" in result
|
|
assert isinstance(result["ota"], list), f"Expected list, got {type(result['ota'])}"
|
|
platforms = {p.get("platform") for p in result["ota"]}
|
|
assert "esphome" in platforms, f"Expected esphome platform in {platforms}"
|
|
assert "web_server" in platforms, f"Expected web_server platform in {platforms}"
|