mirror of
https://github.com/home-assistant/core.git
synced 2025-07-14 16:57:10 +00:00
Update homeassistant integration tests to async (#124117)
This commit is contained in:
Erik Montnemery
GitHub
parent
d72d4286db
commit
63d1cc10e2
@ -1,7 +1,5 @@
|
|||||||
"""The tests for Core components."""
|
"""The tests for Core components."""
|
||||||
|
|
||||||
import asyncio
|
|
||||||
import unittest
|
|
||||||
from unittest.mock import Mock, patch
|
from unittest.mock import Mock, patch
|
||||||
|
|
||||||
import pytest
|
import pytest
|
||||||
@ -44,206 +42,143 @@ from tests.common import (
|
|||||||
MockUser,
|
MockUser,
|
||||||
async_capture_events,
|
async_capture_events,
|
||||||
async_mock_service,
|
async_mock_service,
|
||||||
get_test_home_assistant,
|
|
||||||
mock_service,
|
|
||||||
patch_yaml_files,
|
patch_yaml_files,
|
||||||
)
|
)
|
||||||
|
|
||||||
|
|
||||||
def turn_on(hass, entity_id=None, **service_data):
|
async def test_is_on(hass: HomeAssistant) -> None:
|
||||||
"""Turn specified entity on if possible.
|
"""Test is_on method."""
|
||||||
|
with pytest.raises(
|
||||||
This is a legacy helper method. Do not use it for new tests.
|
RuntimeError,
|
||||||
"""
|
match="Detected code that uses homeassistant.components.is_on. This is deprecated and will stop working",
|
||||||
if entity_id is not None:
|
):
|
||||||
service_data[ATTR_ENTITY_ID] = entity_id
|
assert comps.is_on(hass, "light.Bowl")
|
||||||
|
|
||||||
hass.services.call(ha.DOMAIN, SERVICE_TURN_ON, service_data)
|
|
||||||
|
|
||||||
|
|
||||||
def turn_off(hass, entity_id=None, **service_data):
|
async def test_turn_on_without_entities(hass: HomeAssistant) -> None:
|
||||||
"""Turn specified entity off.
|
"""Test turn_on method without entities."""
|
||||||
|
await async_setup_component(hass, ha.DOMAIN, {})
|
||||||
This is a legacy helper method. Do not use it for new tests.
|
calls = async_mock_service(hass, "light", SERVICE_TURN_ON)
|
||||||
"""
|
await hass.services.async_call(ha.DOMAIN, SERVICE_TURN_ON, blocking=True)
|
||||||
if entity_id is not None:
|
assert len(calls) == 0
|
||||||
service_data[ATTR_ENTITY_ID] = entity_id
|
|
||||||
|
|
||||||
hass.services.call(ha.DOMAIN, SERVICE_TURN_OFF, service_data)
|
|
||||||
|
|
||||||
|
|
||||||
def toggle(hass, entity_id=None, **service_data):
|
async def test_turn_on(hass: HomeAssistant) -> None:
|
||||||
"""Toggle specified entity.
|
"""Test turn_on method."""
|
||||||
|
await async_setup_component(hass, ha.DOMAIN, {})
|
||||||
This is a legacy helper method. Do not use it for new tests.
|
calls = async_mock_service(hass, "light", SERVICE_TURN_ON)
|
||||||
"""
|
await hass.services.async_call(
|
||||||
if entity_id is not None:
|
ha.DOMAIN, SERVICE_TURN_ON, {ATTR_ENTITY_ID: "light.Ceiling"}, blocking=True
|
||||||
service_data[ATTR_ENTITY_ID] = entity_id
|
|
||||||
|
|
||||||
hass.services.call(ha.DOMAIN, SERVICE_TOGGLE, service_data)
|
|
||||||
|
|
||||||
|
|
||||||
def stop(hass):
|
|
||||||
"""Stop Home Assistant.
|
|
||||||
|
|
||||||
This is a legacy helper method. Do not use it for new tests.
|
|
||||||
"""
|
|
||||||
hass.services.call(ha.DOMAIN, SERVICE_HOMEASSISTANT_STOP)
|
|
||||||
|
|
||||||
|
|
||||||
def restart(hass):
|
|
||||||
"""Stop Home Assistant.
|
|
||||||
|
|
||||||
This is a legacy helper method. Do not use it for new tests.
|
|
||||||
"""
|
|
||||||
hass.services.call(ha.DOMAIN, SERVICE_HOMEASSISTANT_RESTART)
|
|
||||||
|
|
||||||
|
|
||||||
def check_config(hass):
|
|
||||||
"""Check the config files.
|
|
||||||
|
|
||||||
This is a legacy helper method. Do not use it for new tests.
|
|
||||||
"""
|
|
||||||
hass.services.call(ha.DOMAIN, SERVICE_CHECK_CONFIG)
|
|
||||||
|
|
||||||
|
|
||||||
def reload_core_config(hass):
|
|
||||||
"""Reload the core config.
|
|
||||||
|
|
||||||
This is a legacy helper method. Do not use it for new tests.
|
|
||||||
"""
|
|
||||||
hass.services.call(ha.DOMAIN, SERVICE_RELOAD_CORE_CONFIG)
|
|
||||||
|
|
||||||
|
|
||||||
class TestComponentsCore(unittest.TestCase):
|
|
||||||
"""Test homeassistant.components module."""
|
|
||||||
|
|
||||||
def setUp(self):
|
|
||||||
"""Set up things to be run when tests are started."""
|
|
||||||
self._manager = get_test_home_assistant()
|
|
||||||
self.hass = self._manager.__enter__()
|
|
||||||
assert asyncio.run_coroutine_threadsafe(
|
|
||||||
async_setup_component(self.hass, "homeassistant", {}), self.hass.loop
|
|
||||||
).result()
|
|
||||||
|
|
||||||
self.hass.states.set("light.Bowl", STATE_ON)
|
|
||||||
self.hass.states.set("light.Ceiling", STATE_OFF)
|
|
||||||
|
|
||||||
def tearDown(self) -> None:
|
|
||||||
"""Tear down hass object."""
|
|
||||||
self.hass.stop()
|
|
||||||
self._manager.__exit__(None, None, None)
|
|
||||||
|
|
||||||
def test_is_on(self):
|
|
||||||
"""Test is_on method."""
|
|
||||||
with pytest.raises(
|
|
||||||
RuntimeError,
|
|
||||||
match="Detected code that uses homeassistant.components.is_on. This is deprecated and will stop working",
|
|
||||||
):
|
|
||||||
assert comps.is_on(self.hass, "light.Bowl")
|
|
||||||
|
|
||||||
def test_turn_on_without_entities(self):
|
|
||||||
"""Test turn_on method without entities."""
|
|
||||||
calls = mock_service(self.hass, "light", SERVICE_TURN_ON)
|
|
||||||
turn_on(self.hass)
|
|
||||||
self.hass.block_till_done()
|
|
||||||
assert len(calls) == 0
|
|
||||||
|
|
||||||
def test_turn_on(self):
|
|
||||||
"""Test turn_on method."""
|
|
||||||
calls = mock_service(self.hass, "light", SERVICE_TURN_ON)
|
|
||||||
turn_on(self.hass, "light.Ceiling")
|
|
||||||
self.hass.block_till_done()
|
|
||||||
assert len(calls) == 1
|
|
||||||
|
|
||||||
def test_turn_off(self):
|
|
||||||
"""Test turn_off method."""
|
|
||||||
calls = mock_service(self.hass, "light", SERVICE_TURN_OFF)
|
|
||||||
turn_off(self.hass, "light.Bowl")
|
|
||||||
self.hass.block_till_done()
|
|
||||||
assert len(calls) == 1
|
|
||||||
|
|
||||||
def test_toggle(self):
|
|
||||||
"""Test toggle method."""
|
|
||||||
calls = mock_service(self.hass, "light", SERVICE_TOGGLE)
|
|
||||||
toggle(self.hass, "light.Bowl")
|
|
||||||
self.hass.block_till_done()
|
|
||||||
assert len(calls) == 1
|
|
||||||
|
|
||||||
@patch("homeassistant.config.os.path.isfile", Mock(return_value=True))
|
|
||||||
def test_reload_core_conf(self):
|
|
||||||
"""Test reload core conf service."""
|
|
||||||
ent = entity.Entity()
|
|
||||||
ent.entity_id = "test.entity"
|
|
||||||
ent.hass = self.hass
|
|
||||||
ent.schedule_update_ha_state()
|
|
||||||
self.hass.block_till_done()
|
|
||||||
|
|
||||||
state = self.hass.states.get("test.entity")
|
|
||||||
assert state is not None
|
|
||||||
assert state.state == "unknown"
|
|
||||||
assert state.attributes == {}
|
|
||||||
|
|
||||||
files = {
|
|
||||||
config.YAML_CONFIG_FILE: yaml.dump(
|
|
||||||
{
|
|
||||||
ha.DOMAIN: {
|
|
||||||
"country": "SE", # To avoid creating issue country_not_configured
|
|
||||||
"latitude": 10,
|
|
||||||
"longitude": 20,
|
|
||||||
"customize": {"test.Entity": {"hello": "world"}},
|
|
||||||
}
|
|
||||||
}
|
|
||||||
)
|
|
||||||
}
|
|
||||||
with patch_yaml_files(files, True):
|
|
||||||
reload_core_config(self.hass)
|
|
||||||
self.hass.block_till_done()
|
|
||||||
|
|
||||||
assert self.hass.config.latitude == 10
|
|
||||||
assert self.hass.config.longitude == 20
|
|
||||||
|
|
||||||
ent.schedule_update_ha_state()
|
|
||||||
self.hass.block_till_done()
|
|
||||||
|
|
||||||
state = self.hass.states.get("test.entity")
|
|
||||||
assert state is not None
|
|
||||||
assert state.state == "unknown"
|
|
||||||
assert state.attributes.get("hello") == "world"
|
|
||||||
|
|
||||||
@patch("homeassistant.config.os.path.isfile", Mock(return_value=True))
|
|
||||||
@patch("homeassistant.components.homeassistant._LOGGER.error")
|
|
||||||
@patch("homeassistant.config.async_process_ha_core_config")
|
|
||||||
def test_reload_core_with_wrong_conf(self, mock_process, mock_error):
|
|
||||||
"""Test reload core conf service."""
|
|
||||||
files = {config.YAML_CONFIG_FILE: yaml.dump(["invalid", "config"])}
|
|
||||||
with patch_yaml_files(files, True):
|
|
||||||
reload_core_config(self.hass)
|
|
||||||
self.hass.block_till_done()
|
|
||||||
|
|
||||||
assert mock_error.called
|
|
||||||
assert mock_process.called is False
|
|
||||||
|
|
||||||
@patch("homeassistant.core.HomeAssistant.async_stop", return_value=None)
|
|
||||||
@patch(
|
|
||||||
"homeassistant.config.async_check_ha_config_file",
|
|
||||||
side_effect=HomeAssistantError("Test error"),
|
|
||||||
)
|
)
|
||||||
def test_restart_homeassistant_wrong_conf(self, mock_check, mock_restart):
|
assert len(calls) == 1
|
||||||
"""Test stop service."""
|
|
||||||
restart(self.hass)
|
|
||||||
self.hass.block_till_done()
|
|
||||||
assert mock_check.called
|
|
||||||
assert not mock_restart.called
|
|
||||||
|
|
||||||
@patch("homeassistant.core.HomeAssistant.async_stop", return_value=None)
|
|
||||||
@patch("homeassistant.config.async_check_ha_config_file", return_value=None)
|
async def test_turn_off(hass: HomeAssistant) -> None:
|
||||||
def test_check_config(self, mock_check, mock_stop):
|
"""Test turn_off method."""
|
||||||
"""Test stop service."""
|
await async_setup_component(hass, ha.DOMAIN, {})
|
||||||
check_config(self.hass)
|
calls = async_mock_service(hass, "light", SERVICE_TURN_OFF)
|
||||||
self.hass.block_till_done()
|
await hass.services.async_call(
|
||||||
assert mock_check.called
|
ha.DOMAIN, SERVICE_TURN_OFF, {ATTR_ENTITY_ID: "light.Bowl"}, blocking=True
|
||||||
assert not mock_stop.called
|
)
|
||||||
|
assert len(calls) == 1
|
||||||
|
|
||||||
|
|
||||||
|
async def test_toggle(hass: HomeAssistant) -> None:
|
||||||
|
"""Test toggle method."""
|
||||||
|
await async_setup_component(hass, ha.DOMAIN, {})
|
||||||
|
calls = async_mock_service(hass, "light", SERVICE_TOGGLE)
|
||||||
|
await hass.services.async_call(
|
||||||
|
ha.DOMAIN, SERVICE_TOGGLE, {ATTR_ENTITY_ID: "light.Bowl"}, blocking=True
|
||||||
|
)
|
||||||
|
assert len(calls) == 1
|
||||||
|
|
||||||
|
|
||||||
|
@patch("homeassistant.config.os.path.isfile", Mock(return_value=True))
|
||||||
|
async def test_reload_core_conf(hass: HomeAssistant) -> None:
|
||||||
|
"""Test reload core conf service."""
|
||||||
|
await async_setup_component(hass, ha.DOMAIN, {})
|
||||||
|
ent = entity.Entity()
|
||||||
|
ent.entity_id = "test.entity"
|
||||||
|
ent.hass = hass
|
||||||
|
ent.async_write_ha_state()
|
||||||
|
|
||||||
|
state = hass.states.get("test.entity")
|
||||||
|
assert state is not None
|
||||||
|
assert state.state == "unknown"
|
||||||
|
assert state.attributes == {}
|
||||||
|
|
||||||
|
files = {
|
||||||
|
config.YAML_CONFIG_FILE: yaml.dump(
|
||||||
|
{
|
||||||
|
ha.DOMAIN: {
|
||||||
|
"country": "SE", # To avoid creating issue country_not_configured
|
||||||
|
"latitude": 10,
|
||||||
|
"longitude": 20,
|
||||||
|
"customize": {"test.Entity": {"hello": "world"}},
|
||||||
|
}
|
||||||
|
}
|
||||||
|
)
|
||||||
|
}
|
||||||
|
with patch_yaml_files(files, True):
|
||||||
|
await hass.services.async_call(
|
||||||
|
ha.DOMAIN, SERVICE_RELOAD_CORE_CONFIG, blocking=True
|
||||||
|
)
|
||||||
|
|
||||||
|
assert hass.config.latitude == 10
|
||||||
|
assert hass.config.longitude == 20
|
||||||
|
|
||||||
|
ent.async_write_ha_state()
|
||||||
|
|
||||||
|
state = hass.states.get("test.entity")
|
||||||
|
assert state is not None
|
||||||
|
assert state.state == "unknown"
|
||||||
|
assert state.attributes.get("hello") == "world"
|
||||||
|
|
||||||
|
|
||||||
|
@patch("homeassistant.config.os.path.isfile", Mock(return_value=True))
|
||||||
|
@patch("homeassistant.components.homeassistant._LOGGER.error")
|
||||||
|
@patch("homeassistant.config.async_process_ha_core_config")
|
||||||
|
async def test_reload_core_with_wrong_conf(
|
||||||
|
mock_process, mock_error, hass: HomeAssistant
|
||||||
|
) -> None:
|
||||||
|
"""Test reload core conf service."""
|
||||||
|
files = {config.YAML_CONFIG_FILE: yaml.dump(["invalid", "config"])}
|
||||||
|
await async_setup_component(hass, ha.DOMAIN, {})
|
||||||
|
with patch_yaml_files(files, True):
|
||||||
|
await hass.services.async_call(
|
||||||
|
ha.DOMAIN, SERVICE_RELOAD_CORE_CONFIG, blocking=True
|
||||||
|
)
|
||||||
|
|
||||||
|
assert mock_error.called
|
||||||
|
assert mock_process.called is False
|
||||||
|
|
||||||
|
|
||||||
|
@patch("homeassistant.core.HomeAssistant.async_stop", return_value=None)
|
||||||
|
@patch(
|
||||||
|
"homeassistant.config.async_check_ha_config_file",
|
||||||
|
side_effect=HomeAssistantError("Test error"),
|
||||||
|
)
|
||||||
|
async def test_restart_homeassistant_wrong_conf(
|
||||||
|
mock_check, mock_restart, hass: HomeAssistant
|
||||||
|
) -> None:
|
||||||
|
"""Test restart service with error."""
|
||||||
|
await async_setup_component(hass, ha.DOMAIN, {})
|
||||||
|
with pytest.raises(HomeAssistantError, match="Test error"):
|
||||||
|
await hass.services.async_call(
|
||||||
|
ha.DOMAIN, SERVICE_HOMEASSISTANT_RESTART, blocking=True
|
||||||
|
)
|
||||||
|
assert mock_check.called
|
||||||
|
assert not mock_restart.called
|
||||||
|
|
||||||
|
|
||||||
|
@patch("homeassistant.core.HomeAssistant.async_stop", return_value=None)
|
||||||
|
@patch("homeassistant.config.async_check_ha_config_file", return_value=None)
|
||||||
|
async def test_check_config(mock_check, mock_stop, hass: HomeAssistant) -> None:
|
||||||
|
"""Test stop service."""
|
||||||
|
await async_setup_component(hass, ha.DOMAIN, {})
|
||||||
|
await hass.services.async_call(ha.DOMAIN, SERVICE_CHECK_CONFIG, blocking=True)
|
||||||
|
assert mock_check.called
|
||||||
|
assert not mock_stop.called
|
||||||
|
|
||||||
|
|
||||||
async def test_turn_on_skips_domains_without_service(
|
async def test_turn_on_skips_domains_without_service(
|
||||||
|
|||||||
Loading…
x
Reference in New Issue
Block a user