mirror of
https://github.com/home-assistant/core.git
synced 2025-12-24 21:06:19 +00:00
Add support for using a single endpoint for rest data (#46711)
This commit is contained in:
J. Nick Koston
GitHub
340
tests/components/rest/test_init.py
Normal file
340
tests/components/rest/test_init.py
Normal file
@@ -0,0 +1,340 @@
|
||||
"""Tests for rest component."""
|
||||
|
||||
import asyncio
|
||||
from datetime import timedelta
|
||||
from os import path
|
||||
from unittest.mock import patch
|
||||
|
||||
import respx
|
||||
|
||||
from homeassistant import config as hass_config
|
||||
from homeassistant.components.rest.const import DOMAIN
|
||||
from homeassistant.const import (
|
||||
ATTR_ENTITY_ID,
|
||||
DATA_MEGABYTES,
|
||||
SERVICE_RELOAD,
|
||||
STATE_UNAVAILABLE,
|
||||
)
|
||||
from homeassistant.setup import async_setup_component
|
||||
from homeassistant.util.dt import utcnow
|
||||
|
||||
from tests.common import async_fire_time_changed
|
||||
|
||||
|
||||
@respx.mock
|
||||
async def test_setup_with_endpoint_timeout_with_recovery(hass):
|
||||
"""Test setup with an endpoint that times out that recovers."""
|
||||
await async_setup_component(hass, "homeassistant", {})
|
||||
|
||||
respx.get("http://localhost").mock(side_effect=asyncio.TimeoutError())
|
||||
assert await async_setup_component(
|
||||
hass,
|
||||
DOMAIN,
|
||||
{
|
||||
DOMAIN: [
|
||||
{
|
||||
"resource": "http://localhost",
|
||||
"method": "GET",
|
||||
"verify_ssl": "false",
|
||||
"timeout": 30,
|
||||
"sensor": [
|
||||
{
|
||||
"unit_of_measurement": DATA_MEGABYTES,
|
||||
"name": "sensor1",
|
||||
"value_template": "{{ value_json.sensor1 }}",
|
||||
},
|
||||
{
|
||||
"unit_of_measurement": DATA_MEGABYTES,
|
||||
"name": "sensor2",
|
||||
"value_template": "{{ value_json.sensor2 }}",
|
||||
},
|
||||
],
|
||||
"binary_sensor": [
|
||||
{
|
||||
"name": "binary_sensor1",
|
||||
"value_template": "{{ value_json.binary_sensor1 }}",
|
||||
},
|
||||
{
|
||||
"name": "binary_sensor2",
|
||||
"value_template": "{{ value_json.binary_sensor2 }}",
|
||||
},
|
||||
],
|
||||
}
|
||||
]
|
||||
},
|
||||
)
|
||||
await hass.async_block_till_done()
|
||||
assert len(hass.states.async_all()) == 0
|
||||
|
||||
respx.get("http://localhost").respond(
|
||||
status_code=200,
|
||||
json={
|
||||
"sensor1": "1",
|
||||
"sensor2": "2",
|
||||
"binary_sensor1": "on",
|
||||
"binary_sensor2": "off",
|
||||
},
|
||||
)
|
||||
|
||||
# Refresh the coordinator
|
||||
async_fire_time_changed(hass, utcnow() + timedelta(seconds=31))
|
||||
await hass.async_block_till_done()
|
||||
|
||||
# Wait for platform setup retry
|
||||
async_fire_time_changed(hass, utcnow() + timedelta(seconds=61))
|
||||
await hass.async_block_till_done()
|
||||
|
||||
assert len(hass.states.async_all()) == 4
|
||||
|
||||
assert hass.states.get("sensor.sensor1").state == "1"
|
||||
assert hass.states.get("sensor.sensor2").state == "2"
|
||||
assert hass.states.get("binary_sensor.binary_sensor1").state == "on"
|
||||
assert hass.states.get("binary_sensor.binary_sensor2").state == "off"
|
||||
|
||||
# Now the end point flakes out again
|
||||
respx.get("http://localhost").mock(side_effect=asyncio.TimeoutError())
|
||||
|
||||
# Refresh the coordinator
|
||||
async_fire_time_changed(hass, utcnow() + timedelta(seconds=31))
|
||||
await hass.async_block_till_done()
|
||||
|
||||
assert hass.states.get("sensor.sensor1").state == STATE_UNAVAILABLE
|
||||
assert hass.states.get("sensor.sensor2").state == STATE_UNAVAILABLE
|
||||
assert hass.states.get("binary_sensor.binary_sensor1").state == STATE_UNAVAILABLE
|
||||
assert hass.states.get("binary_sensor.binary_sensor2").state == STATE_UNAVAILABLE
|
||||
|
||||
# We request a manual refresh when the
|
||||
# endpoint is working again
|
||||
|
||||
respx.get("http://localhost").respond(
|
||||
status_code=200,
|
||||
json={
|
||||
"sensor1": "1",
|
||||
"sensor2": "2",
|
||||
"binary_sensor1": "on",
|
||||
"binary_sensor2": "off",
|
||||
},
|
||||
)
|
||||
|
||||
await hass.services.async_call(
|
||||
"homeassistant",
|
||||
"update_entity",
|
||||
{ATTR_ENTITY_ID: ["sensor.sensor1"]},
|
||||
blocking=True,
|
||||
)
|
||||
assert hass.states.get("sensor.sensor1").state == "1"
|
||||
assert hass.states.get("sensor.sensor2").state == "2"
|
||||
assert hass.states.get("binary_sensor.binary_sensor1").state == "on"
|
||||
assert hass.states.get("binary_sensor.binary_sensor2").state == "off"
|
||||
|
||||
|
||||
@respx.mock
|
||||
async def test_setup_minimum_resource_template(hass):
|
||||
"""Test setup with minimum configuration (resource_template)."""
|
||||
|
||||
respx.get("http://localhost").respond(
|
||||
status_code=200,
|
||||
json={
|
||||
"sensor1": "1",
|
||||
"sensor2": "2",
|
||||
"binary_sensor1": "on",
|
||||
"binary_sensor2": "off",
|
||||
},
|
||||
)
|
||||
assert await async_setup_component(
|
||||
hass,
|
||||
DOMAIN,
|
||||
{
|
||||
DOMAIN: [
|
||||
{
|
||||
"resource_template": "{% set url = 'http://localhost' %}{{ url }}",
|
||||
"method": "GET",
|
||||
"verify_ssl": "false",
|
||||
"timeout": 30,
|
||||
"sensor": [
|
||||
{
|
||||
"unit_of_measurement": DATA_MEGABYTES,
|
||||
"name": "sensor1",
|
||||
"value_template": "{{ value_json.sensor1 }}",
|
||||
},
|
||||
{
|
||||
"unit_of_measurement": DATA_MEGABYTES,
|
||||
"name": "sensor2",
|
||||
"value_template": "{{ value_json.sensor2 }}",
|
||||
},
|
||||
],
|
||||
"binary_sensor": [
|
||||
{
|
||||
"name": "binary_sensor1",
|
||||
"value_template": "{{ value_json.binary_sensor1 }}",
|
||||
},
|
||||
{
|
||||
"name": "binary_sensor2",
|
||||
"value_template": "{{ value_json.binary_sensor2 }}",
|
||||
},
|
||||
],
|
||||
}
|
||||
]
|
||||
},
|
||||
)
|
||||
await hass.async_block_till_done()
|
||||
assert len(hass.states.async_all()) == 4
|
||||
|
||||
assert hass.states.get("sensor.sensor1").state == "1"
|
||||
assert hass.states.get("sensor.sensor2").state == "2"
|
||||
assert hass.states.get("binary_sensor.binary_sensor1").state == "on"
|
||||
assert hass.states.get("binary_sensor.binary_sensor2").state == "off"
|
||||
|
||||
|
||||
@respx.mock
|
||||
async def test_reload(hass):
|
||||
"""Verify we can reload."""
|
||||
|
||||
respx.get("http://localhost") % 200
|
||||
|
||||
assert await async_setup_component(
|
||||
hass,
|
||||
DOMAIN,
|
||||
{
|
||||
DOMAIN: [
|
||||
{
|
||||
"resource": "http://localhost",
|
||||
"method": "GET",
|
||||
"verify_ssl": "false",
|
||||
"timeout": 30,
|
||||
"sensor": [
|
||||
{
|
||||
"name": "mockrest",
|
||||
},
|
||||
],
|
||||
}
|
||||
]
|
||||
},
|
||||
)
|
||||
await hass.async_block_till_done()
|
||||
await hass.async_start()
|
||||
await hass.async_block_till_done()
|
||||
|
||||
assert len(hass.states.async_all()) == 1
|
||||
|
||||
assert hass.states.get("sensor.mockrest")
|
||||
|
||||
yaml_path = path.join(
|
||||
_get_fixtures_base_path(),
|
||||
"fixtures",
|
||||
"rest/configuration_top_level.yaml",
|
||||
)
|
||||
with patch.object(hass_config, "YAML_CONFIG_FILE", yaml_path):
|
||||
await hass.services.async_call(
|
||||
"rest",
|
||||
SERVICE_RELOAD,
|
||||
{},
|
||||
blocking=True,
|
||||
)
|
||||
await hass.async_block_till_done()
|
||||
|
||||
assert hass.states.get("sensor.mockreset") is None
|
||||
assert hass.states.get("sensor.rollout")
|
||||
assert hass.states.get("sensor.fallover")
|
||||
|
||||
|
||||
@respx.mock
|
||||
async def test_reload_and_remove_all(hass):
|
||||
"""Verify we can reload and remove all."""
|
||||
|
||||
respx.get("http://localhost") % 200
|
||||
|
||||
assert await async_setup_component(
|
||||
hass,
|
||||
DOMAIN,
|
||||
{
|
||||
DOMAIN: [
|
||||
{
|
||||
"resource": "http://localhost",
|
||||
"method": "GET",
|
||||
"verify_ssl": "false",
|
||||
"timeout": 30,
|
||||
"sensor": [
|
||||
{
|
||||
"name": "mockrest",
|
||||
},
|
||||
],
|
||||
}
|
||||
]
|
||||
},
|
||||
)
|
||||
await hass.async_block_till_done()
|
||||
await hass.async_start()
|
||||
await hass.async_block_till_done()
|
||||
|
||||
assert len(hass.states.async_all()) == 1
|
||||
|
||||
assert hass.states.get("sensor.mockrest")
|
||||
|
||||
yaml_path = path.join(
|
||||
_get_fixtures_base_path(),
|
||||
"fixtures",
|
||||
"rest/configuration_empty.yaml",
|
||||
)
|
||||
with patch.object(hass_config, "YAML_CONFIG_FILE", yaml_path):
|
||||
await hass.services.async_call(
|
||||
"rest",
|
||||
SERVICE_RELOAD,
|
||||
{},
|
||||
blocking=True,
|
||||
)
|
||||
await hass.async_block_till_done()
|
||||
|
||||
assert hass.states.get("sensor.mockreset") is None
|
||||
|
||||
|
||||
@respx.mock
|
||||
async def test_reload_fails_to_read_configuration(hass):
|
||||
"""Verify reload when configuration is missing or broken."""
|
||||
|
||||
respx.get("http://localhost") % 200
|
||||
|
||||
assert await async_setup_component(
|
||||
hass,
|
||||
DOMAIN,
|
||||
{
|
||||
DOMAIN: [
|
||||
{
|
||||
"resource": "http://localhost",
|
||||
"method": "GET",
|
||||
"verify_ssl": "false",
|
||||
"timeout": 30,
|
||||
"sensor": [
|
||||
{
|
||||
"name": "mockrest",
|
||||
},
|
||||
],
|
||||
}
|
||||
]
|
||||
},
|
||||
)
|
||||
await hass.async_block_till_done()
|
||||
await hass.async_start()
|
||||
await hass.async_block_till_done()
|
||||
|
||||
assert len(hass.states.async_all()) == 1
|
||||
|
||||
yaml_path = path.join(
|
||||
_get_fixtures_base_path(),
|
||||
"fixtures",
|
||||
"rest/configuration_invalid.notyaml",
|
||||
)
|
||||
with patch.object(hass_config, "YAML_CONFIG_FILE", yaml_path):
|
||||
await hass.services.async_call(
|
||||
"rest",
|
||||
SERVICE_RELOAD,
|
||||
{},
|
||||
blocking=True,
|
||||
)
|
||||
await hass.async_block_till_done()
|
||||
|
||||
assert len(hass.states.async_all()) == 1
|
||||
|
||||
|
||||
def _get_fixtures_base_path():
|
||||
return path.dirname(path.dirname(path.dirname(__file__)))
|
||||
Reference in New Issue
Block a user