"""Tests for the homeassistant number platform."""

from collections.abc import Callable
from pathlib import Path

import pytest

from esphome.config import read_config
from esphome.core import CORE

VALID_CONFIG = "tests/component_tests/homeassistant/test_number.yaml"
INVALID_CONFIG = "tests/component_tests/homeassistant/test_number_invalid_domain.yaml"


def test_number_and_input_number_entities(
    generate_main: Callable[[str | Path], str],
) -> None:
    """Both number and input_number entity IDs are accepted and passed to the C++ class."""
    main_cpp = generate_main(VALID_CONFIG)

    assert 'ha_number->set_entity_id("number.some_number");' in main_cpp
    assert 'ha_input_number->set_entity_id("input_number.some_input");' in main_cpp


def test_number_rejects_unsupported_domain(
    capsys: pytest.CaptureFixture[str],
) -> None:
    """An entity ID from another domain fails config validation."""
    CORE.config_path = Path(INVALID_CONFIG)

    assert read_config({}) is None
    assert "is not supported by the number platform" in capsys.readouterr().out