40 lines
1.3 KiB
Python
40 lines
1.3 KiB
Python
|
"""Test the init file of Mailgun."""
|
||
|
|
from unittest.mock import patch
|
||
|
|
|
||
|
|
from homeassistant import data_entry_flow
|
||
|
|
from homeassistant.components import mailgun
|
||
|
|
|
||
|
|
from homeassistant.core import callback
|
||
|
|
|
||
|
|
|
||
|
|
async def test_config_flow_registers_webhook(hass, aiohttp_client):
|
||
|
|
"""Test setting up Mailgun and sending webhook."""
|
||
|
|
with patch('homeassistant.util.get_local_ip', return_value='example.com'):
|
||
|
|
result = await hass.config_entries.flow.async_init('mailgun', 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']
|
||
|
|
|
||
|
|
mailgun_events = []
|
||
|
|
|
||
|
|
@callback
|
||
|
|
def handle_event(event):
|
||
|
|
"""Handle Mailgun event."""
|
||
|
|
mailgun_events.append(event)
|
||
|
|
|
||
|
|
hass.bus.async_listen(mailgun.MESSAGE_RECEIVED, handle_event)
|
||
|
|
|
||
|
|
client = await aiohttp_client(hass.http.app)
|
||
|
|
await client.post('/api/webhook/{}'.format(webhook_id), data={
|
||
|
|
'hello': 'mailgun'
|
||
|
|
})
|
||
|
|
|
||
|
|
assert len(mailgun_events) == 1
|
||
|
|
assert mailgun_events[0].data['webhook_id'] == webhook_id
|
||
|
|
assert mailgun_events[0].data['hello'] == 'mailgun'
|