import pytest from mock import Mock from esphome import cpp_helpers as ch from esphome import const from esphome.cpp_generator import MockObj @pytest.mark.asyncio async def test_gpio_pin_expression__conf_is_none(monkeypatch): actual = await ch.gpio_pin_expression(None) assert actual is None @pytest.mark.asyncio async def test_gpio_pin_expression__new_pin(monkeypatch): actual = await ch.gpio_pin_expression( {const.CONF_NUMBER: 42, const.CONF_MODE: "input", const.CONF_INVERTED: False} ) assert isinstance(actual, MockObj) @pytest.mark.asyncio async def test_register_component(monkeypatch): var = Mock(base="foo.bar") app_mock = Mock(register_component=Mock(return_value=var)) monkeypatch.setattr(ch, "App", app_mock) core_mock = Mock(component_ids=["foo.bar"]) monkeypatch.setattr(ch, "CORE", core_mock) add_mock = Mock() monkeypatch.setattr(ch, "add", add_mock) actual = await ch.register_component(var, {}) assert actual is var assert add_mock.call_count == 2 app_mock.register_component.assert_called_with(var) assert core_mock.component_ids == [] @pytest.mark.asyncio async def test_register_component__no_component_id(monkeypatch): var = Mock(base="foo.eek") core_mock = Mock(component_ids=["foo.bar"]) monkeypatch.setattr(ch, "CORE", core_mock) with pytest.raises(ValueError, match="Component ID foo.eek was not declared to"): await ch.register_component(var, {}) @pytest.mark.asyncio async def test_register_component__with_setup_priority(monkeypatch): var = Mock(base="foo.bar") app_mock = Mock(register_component=Mock(return_value=var)) monkeypatch.setattr(ch, "App", app_mock) core_mock = Mock(component_ids=["foo.bar"]) monkeypatch.setattr(ch, "CORE", core_mock) add_mock = Mock() monkeypatch.setattr(ch, "add", add_mock) actual = await ch.register_component( var, { const.CONF_SETUP_PRIORITY: "123", const.CONF_UPDATE_INTERVAL: "456", }, ) assert actual is var add_mock.assert_called() assert add_mock.call_count == 4 app_mock.register_component.assert_called_with(var) assert core_mock.component_ids == []