core/tests/components/test_no_ip.py

88 lines
2.4 KiB
Python

"""Test the NO-IP component."""
import asyncio
from datetime import timedelta
import pytest
from homeassistant.setup import async_setup_component
from homeassistant.components import no_ip
from homeassistant.util.dt import utcnow
from tests.common import async_fire_time_changed
DOMAIN = 'test.example.com'
PASSWORD = 'xyz789'
UPDATE_URL = no_ip.UPDATE_URL
USERNAME = 'abc@123.com'
@pytest.fixture
def setup_no_ip(hass, aioclient_mock):
"""Fixture that sets up NO-IP."""
aioclient_mock.get(
UPDATE_URL, params={'hostname': DOMAIN}, text='good 0.0.0.0')
hass.loop.run_until_complete(async_setup_component(hass, no_ip.DOMAIN, {
no_ip.DOMAIN: {
'domain': DOMAIN,
'username': USERNAME,
'password': PASSWORD,
}
}))
@asyncio.coroutine
def test_setup(hass, aioclient_mock):
"""Test setup works if update passes."""
aioclient_mock.get(
UPDATE_URL, params={'hostname': DOMAIN}, text='nochg 0.0.0.0')
result = yield from async_setup_component(hass, no_ip.DOMAIN, {
no_ip.DOMAIN: {
'domain': DOMAIN,
'username': USERNAME,
'password': PASSWORD,
}
})
assert result
assert aioclient_mock.call_count == 1
async_fire_time_changed(hass, utcnow() + timedelta(minutes=5))
yield from hass.async_block_till_done()
assert aioclient_mock.call_count == 2
@asyncio.coroutine
def test_setup_fails_if_update_fails(hass, aioclient_mock):
"""Test setup fails if first update fails."""
aioclient_mock.get(UPDATE_URL, params={'hostname': DOMAIN}, text='nohost')
result = yield from async_setup_component(hass, no_ip.DOMAIN, {
no_ip.DOMAIN: {
'domain': DOMAIN,
'username': USERNAME,
'password': PASSWORD,
}
})
assert not result
assert aioclient_mock.call_count == 1
@asyncio.coroutine
def test_setup_fails_if_wrong_auth(hass, aioclient_mock):
"""Test setup fails if first update fails through wrong authentication."""
aioclient_mock.get(UPDATE_URL, params={'hostname': DOMAIN}, text='badauth')
result = yield from async_setup_component(hass, no_ip.DOMAIN, {
no_ip.DOMAIN: {
'domain': DOMAIN,
'username': USERNAME,
'password': PASSWORD,
}
})
assert not result
assert aioclient_mock.call_count == 1