mirror of
https://github.com/esphome/esphome.git
synced 2026-08-30 17:46:01 +00:00
Mustafa KURUandJonathan Swoboda
950cfc4da3
Co-authored-by: Jonathan Swoboda <154711427+swoboda1337@users.noreply.github.com>
83 lines
2.7 KiB
Python
83 lines
2.7 KiB
Python
"""Tests for the wake button's wakeup_pin requirement in ld6002b."""
|
|
|
|
from __future__ import annotations
|
|
|
|
import pytest
|
|
|
|
from esphome.components.ld6002b.button import CONFIG_SCHEMA, FINAL_VALIDATE_SCHEMA
|
|
from esphome.config import Config
|
|
import esphome.config_validation as cv
|
|
from esphome.const import CONF_ID, CONF_WAKEUP_PIN, PlatformFramework
|
|
from esphome.core import ID
|
|
from esphome.types import ConfigType
|
|
from tests.component_tests.types import SetCoreConfigCallable
|
|
|
|
HUB_ID = "ld6002b_hub"
|
|
|
|
|
|
def _full_config(hub: ConfigType) -> Config:
|
|
"""A full config carrying one ld6002b hub, as the ID pass leaves it.
|
|
|
|
final_validate resolves the hub through get_path_for_id, so the declaring
|
|
path has to be registered the way validate_config registers it: the path of
|
|
the id value itself, whose parent is the hub's own config.
|
|
"""
|
|
full = Config()
|
|
full["ld6002b"] = [hub]
|
|
full.declare_ids.append((hub[CONF_ID], ["ld6002b", 0, CONF_ID]))
|
|
return full
|
|
|
|
|
|
def _hub(*, wakeup_pin: bool) -> ConfigType:
|
|
hub: ConfigType = {CONF_ID: ID(HUB_ID, is_declaration=True, type="ld6002b")}
|
|
if wakeup_pin:
|
|
hub[CONF_WAKEUP_PIN] = {"number": 4}
|
|
return hub
|
|
|
|
|
|
def _buttons(**buttons: str) -> ConfigType:
|
|
"""A button platform config naming the given buttons on the shared hub."""
|
|
config: ConfigType = {
|
|
"ld6002b_id": ID(HUB_ID, is_declaration=False, type="ld6002b")
|
|
}
|
|
config.update({key: {"name": name} for key, name in buttons.items()})
|
|
return config
|
|
|
|
|
|
def _validated(config: ConfigType) -> ConfigType:
|
|
"""Run the button schema, then the final validation the hub is checked in."""
|
|
config = CONFIG_SCHEMA(config)
|
|
FINAL_VALIDATE_SCHEMA(config)
|
|
return config
|
|
|
|
|
|
def test_wake_without_wakeup_pin_is_rejected(
|
|
set_core_config: SetCoreConfigCallable,
|
|
) -> None:
|
|
"""The wake button drives the pin directly, so a hub without one cannot serve it."""
|
|
set_core_config(
|
|
PlatformFramework.ESP32_IDF, full_config=_full_config(_hub(wakeup_pin=False))
|
|
)
|
|
|
|
with pytest.raises(cv.Invalid, match="wake requires wakeup_pin"):
|
|
_validated(_buttons(wake="Wake"))
|
|
|
|
|
|
def test_wake_with_wakeup_pin_passes(set_core_config: SetCoreConfigCallable) -> None:
|
|
set_core_config(
|
|
PlatformFramework.ESP32_IDF, full_config=_full_config(_hub(wakeup_pin=True))
|
|
)
|
|
|
|
_validated(_buttons(wake="Wake"))
|
|
|
|
|
|
def test_other_buttons_do_not_need_the_pin(
|
|
set_core_config: SetCoreConfigCallable,
|
|
) -> None:
|
|
"""Only wake drives the pin; the query buttons stay usable without one."""
|
|
set_core_config(
|
|
PlatformFramework.ESP32_IDF, full_config=_full_config(_hub(wakeup_pin=False))
|
|
)
|
|
|
|
_validated(_buttons(get_delay="Get Delay"))
|