117 lines
4.0 KiB
Python
117 lines
4.0 KiB
Python
"""Tests for Almond set up."""
|
|
from time import time
|
|
|
|
import pytest
|
|
|
|
from homeassistant import config_entries, core
|
|
from homeassistant.components.almond import const
|
|
from homeassistant.config import async_process_ha_core_config
|
|
from homeassistant.const import EVENT_HOMEASSISTANT_START
|
|
from homeassistant.setup import async_setup_component
|
|
from homeassistant.util.dt import utcnow
|
|
|
|
from tests.async_mock import patch
|
|
from tests.common import MockConfigEntry, async_fire_time_changed
|
|
|
|
|
|
@pytest.fixture(autouse=True)
|
|
def patch_hass_state(hass):
|
|
"""Mock the hass.state to be not_running."""
|
|
hass.state = core.CoreState.not_running
|
|
|
|
|
|
async def test_set_up_oauth_remote_url(hass, aioclient_mock):
|
|
"""Test we set up Almond to connect to HA if we have external url."""
|
|
entry = MockConfigEntry(
|
|
domain="almond",
|
|
data={
|
|
"type": const.TYPE_OAUTH2,
|
|
"auth_implementation": "local",
|
|
"host": "http://localhost:9999",
|
|
"token": {"expires_at": time() + 1000, "access_token": "abcd"},
|
|
},
|
|
)
|
|
entry.add_to_hass(hass)
|
|
|
|
with patch(
|
|
"homeassistant.helpers.config_entry_oauth2_flow.async_get_config_entry_implementation",
|
|
):
|
|
assert await async_setup_component(hass, "almond", {})
|
|
|
|
assert entry.state == config_entries.ENTRY_STATE_LOADED
|
|
|
|
hass.config.components.add("cloud")
|
|
with patch("homeassistant.components.almond.ALMOND_SETUP_DELAY", 0), patch(
|
|
"homeassistant.helpers.network.get_url",
|
|
return_value="https://example.nabu.casa",
|
|
), patch("pyalmond.WebAlmondAPI.async_create_device") as mock_create_device:
|
|
hass.bus.async_fire(EVENT_HOMEASSISTANT_START)
|
|
await hass.async_block_till_done()
|
|
async_fire_time_changed(hass, utcnow())
|
|
await hass.async_block_till_done()
|
|
|
|
assert len(mock_create_device.mock_calls) == 1
|
|
|
|
|
|
async def test_set_up_oauth_no_external_url(hass, aioclient_mock):
|
|
"""Test we do not set up Almond to connect to HA if we have no external url."""
|
|
entry = MockConfigEntry(
|
|
domain="almond",
|
|
data={
|
|
"type": const.TYPE_OAUTH2,
|
|
"auth_implementation": "local",
|
|
"host": "http://localhost:9999",
|
|
"token": {"expires_at": time() + 1000, "access_token": "abcd"},
|
|
},
|
|
)
|
|
entry.add_to_hass(hass)
|
|
|
|
with patch(
|
|
"homeassistant.helpers.config_entry_oauth2_flow.async_get_config_entry_implementation",
|
|
), patch("pyalmond.WebAlmondAPI.async_create_device") as mock_create_device:
|
|
assert await async_setup_component(hass, "almond", {})
|
|
|
|
assert entry.state == config_entries.ENTRY_STATE_LOADED
|
|
assert len(mock_create_device.mock_calls) == 0
|
|
|
|
|
|
async def test_set_up_hassio(hass, aioclient_mock):
|
|
"""Test we do not set up Almond to connect to HA if we use Hass.io."""
|
|
entry = MockConfigEntry(
|
|
domain="almond",
|
|
data={
|
|
"is_hassio": True,
|
|
"type": const.TYPE_LOCAL,
|
|
"host": "http://localhost:9999",
|
|
},
|
|
)
|
|
entry.add_to_hass(hass)
|
|
|
|
with patch("pyalmond.WebAlmondAPI.async_create_device") as mock_create_device:
|
|
assert await async_setup_component(hass, "almond", {})
|
|
|
|
assert entry.state == config_entries.ENTRY_STATE_LOADED
|
|
assert len(mock_create_device.mock_calls) == 0
|
|
|
|
|
|
async def test_set_up_local(hass, aioclient_mock):
|
|
"""Test we do not set up Almond to connect to HA if we use local."""
|
|
|
|
# Set up an internal URL, as Almond won't be set up if there is no URL available
|
|
await async_process_ha_core_config(
|
|
hass,
|
|
{"internal_url": "https://192.168.0.1"},
|
|
)
|
|
|
|
entry = MockConfigEntry(
|
|
domain="almond",
|
|
data={"type": const.TYPE_LOCAL, "host": "http://localhost:9999"},
|
|
)
|
|
entry.add_to_hass(hass)
|
|
|
|
with patch("pyalmond.WebAlmondAPI.async_create_device") as mock_create_device:
|
|
assert await async_setup_component(hass, "almond", {})
|
|
|
|
assert entry.state == config_entries.ENTRY_STATE_LOADED
|
|
assert len(mock_create_device.mock_calls) == 1
|