mirror of https://github.com/home-assistant/core
78 lines
2.3 KiB
Python
78 lines
2.3 KiB
Python
"""Test the NamecheapDNS component."""
|
|
|
|
from datetime import timedelta
|
|
|
|
import pytest
|
|
|
|
from homeassistant.components import namecheapdns
|
|
from homeassistant.core import HomeAssistant
|
|
from homeassistant.setup import async_setup_component
|
|
from homeassistant.util.dt import utcnow
|
|
|
|
from tests.common import async_fire_time_changed
|
|
from tests.test_util.aiohttp import AiohttpClientMocker
|
|
|
|
HOST = "test"
|
|
DOMAIN = "bla"
|
|
PASSWORD = "abcdefgh"
|
|
|
|
|
|
@pytest.fixture
|
|
def setup_namecheapdns(
|
|
hass: HomeAssistant, aioclient_mock: AiohttpClientMocker
|
|
) -> None:
|
|
"""Fixture that sets up NamecheapDNS."""
|
|
aioclient_mock.get(
|
|
namecheapdns.UPDATE_URL,
|
|
params={"host": HOST, "domain": DOMAIN, "password": PASSWORD},
|
|
text="<interface-response><ErrCount>0</ErrCount></interface-response>",
|
|
)
|
|
|
|
hass.loop.run_until_complete(
|
|
async_setup_component(
|
|
hass,
|
|
namecheapdns.DOMAIN,
|
|
{"namecheapdns": {"host": HOST, "domain": DOMAIN, "password": PASSWORD}},
|
|
)
|
|
)
|
|
|
|
|
|
async def test_setup(hass: HomeAssistant, aioclient_mock: AiohttpClientMocker) -> None:
|
|
"""Test setup works if update passes."""
|
|
aioclient_mock.get(
|
|
namecheapdns.UPDATE_URL,
|
|
params={"host": HOST, "domain": DOMAIN, "password": PASSWORD},
|
|
text="<interface-response><ErrCount>0</ErrCount></interface-response>",
|
|
)
|
|
|
|
result = await async_setup_component(
|
|
hass,
|
|
namecheapdns.DOMAIN,
|
|
{"namecheapdns": {"host": HOST, "domain": DOMAIN, "password": PASSWORD}},
|
|
)
|
|
assert result
|
|
assert aioclient_mock.call_count == 1
|
|
|
|
async_fire_time_changed(hass, utcnow() + timedelta(minutes=5))
|
|
await hass.async_block_till_done()
|
|
assert aioclient_mock.call_count == 2
|
|
|
|
|
|
async def test_setup_fails_if_update_fails(
|
|
hass: HomeAssistant, aioclient_mock: AiohttpClientMocker
|
|
) -> None:
|
|
"""Test setup fails if first update fails."""
|
|
aioclient_mock.get(
|
|
namecheapdns.UPDATE_URL,
|
|
params={"host": HOST, "domain": DOMAIN, "password": PASSWORD},
|
|
text="<interface-response><ErrCount>1</ErrCount></interface-response>",
|
|
)
|
|
|
|
result = await async_setup_component(
|
|
hass,
|
|
namecheapdns.DOMAIN,
|
|
{"namecheapdns": {"host": HOST, "domain": DOMAIN, "password": PASSWORD}},
|
|
)
|
|
assert not result
|
|
assert aioclient_mock.call_count == 1
|