core/tests/components/ifttt/test_init.py

"""Test the init file of IFTTT."""
from unittest.mock import patch

from homeassistant import data_entry_flow
from homeassistant.core import callback
from homeassistant.components import ifttt


async def test_config_flow_registers_webhook(hass, aiohttp_client):
    """Test setting up IFTTT and sending webhook."""
    with patch("homeassistant.util.get_local_ip", return_value="example.com"):
        result = await hass.config_entries.flow.async_init(
            "ifttt", context={"source": "user"}
        )
    assert result["type"] == data_entry_flow.RESULT_TYPE_FORM, result

    result = await hass.config_entries.flow.async_configure(result["flow_id"], {})
    assert result["type"] == data_entry_flow.RESULT_TYPE_CREATE_ENTRY
    webhook_id = result["result"].data["webhook_id"]

    ifttt_events = []

    @callback
    def handle_event(event):
        """Handle IFTTT event."""
        ifttt_events.append(event)

    hass.bus.async_listen(ifttt.EVENT_RECEIVED, handle_event)

    client = await aiohttp_client(hass.http.app)
    await client.post("/api/webhook/{}".format(webhook_id), json={"hello": "ifttt"})

    assert len(ifttt_events) == 1
    assert ifttt_events[0].data["webhook_id"] == webhook_id
    assert ifttt_events[0].data["hello"] == "ifttt"