mirror of
https://github.com/esphome/esphome.git
synced 2026-10-08 03:54:03 +00:00
35 lines
1.2 KiB
Python
35 lines
1.2 KiB
Python
"""Tests for the homeassistant button platform."""
|
|
|
|
from collections.abc import Callable
|
|
from pathlib import Path
|
|
|
|
import pytest
|
|
|
|
from esphome.config import read_config
|
|
from esphome.core import CORE
|
|
|
|
DIR = Path(__file__).parent
|
|
|
|
|
|
def test_button_valid_domains(generate_main: Callable[[str | Path], str]) -> None:
|
|
"""Both button and input_button entities are accepted and wired up."""
|
|
main_cpp = generate_main(DIR / "button_valid.yaml")
|
|
|
|
assert "homeassistant::HomeassistantButton" in main_cpp
|
|
assert 'ha_button->set_entity_id("button.example");' in main_cpp
|
|
assert 'ha_input_button->set_entity_id("input_button.example");' in main_cpp
|
|
assert any(d.name == "USE_API_HOMEASSISTANT_SERVICES" for d in CORE.defines)
|
|
assert any(d.name == "USE_API_HOMEASSISTANT_STATES" for d in CORE.defines)
|
|
|
|
|
|
def test_button_invalid_domain(capsys: pytest.CaptureFixture[str]) -> None:
|
|
"""An entity from an unsupported domain is rejected."""
|
|
CORE.config_path = DIR / "button_invalid.yaml"
|
|
result = read_config({})
|
|
|
|
assert result is None
|
|
assert (
|
|
"Entity ID sensor.example is not supported by the button platform."
|
|
in capsys.readouterr().out
|
|
)
|