evlli/hass-core
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Small tweaks of switch_as_x tests (#68195)

Browse source
This commit is contained in:
Erik Montnemery 2022-03-15 18:34:46 +01:00 • committed by GitHub
parent 0c4efae31f
commit f026245cb4
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
2 changed files with 62 additions and 88 deletions
Download patch file Download diff file Expand all files Collapse all files

35
tests/components/switch_as_x/test_config_flow.py
View file

@ -12,8 +12,18 @@ from homeassistant.core import HomeAssistant
from homeassistant.data_entry_flow import RESULT_TYPE_CREATE_ENTRY, RESULT_TYPE_FORM from homeassistant.data_entry_flow import RESULT_TYPE_CREATE_ENTRY, RESULT_TYPE_FORM
from homeassistant.helpers import entity_registry as er from homeassistant.helpers import entity_registry as er
from tests.common import MockConfigEntry
@pytest.mark.parametrize("target_domain", (Platform.LIGHT,)) PLATFORMS_TO_TEST = (
Platform.COVER,
Platform.FAN,
Platform.LIGHT,
Platform.LOCK,
Platform.SIREN,
)
@pytest.mark.parametrize("target_domain", PLATFORMS_TO_TEST)
async def test_config_flow( async def test_config_flow(
hass: HomeAssistant, hass: HomeAssistant,
target_domain: Platform, target_domain: Platform,
@ -59,7 +69,7 @@ async def test_config_flow(
(None, er.RegistryEntryHider.INTEGRATION.value), (None, er.RegistryEntryHider.INTEGRATION.value),
), ),
) )
@pytest.mark.parametrize("target_domain", (Platform.LIGHT,)) @pytest.mark.parametrize("target_domain", PLATFORMS_TO_TEST)
async def test_config_flow_registered_entity( async def test_config_flow_registered_entity(
hass: HomeAssistant, hass: HomeAssistant,
target_domain: Platform, target_domain: Platform,
@ -110,29 +120,26 @@ async def test_config_flow_registered_entity(
assert switch_entity_entry.hidden_by == hidden_by_after assert switch_entity_entry.hidden_by == hidden_by_after
@pytest.mark.parametrize("target_domain", (Platform.LIGHT,)) @pytest.mark.parametrize("target_domain", PLATFORMS_TO_TEST)
async def test_options( async def test_options(
hass: HomeAssistant, hass: HomeAssistant,
target_domain: Platform, target_domain: Platform,
mock_setup_entry: AsyncMock, mock_setup_entry: AsyncMock,
) -> None: ) -> None:
"""Test reconfiguring.""" """Test reconfiguring."""
result = await hass.config_entries.flow.async_init( switch_as_x_config_entry = MockConfigEntry(
DOMAIN, context={"source": config_entries.SOURCE_USER} data={},
) domain=DOMAIN,
assert result["type"] == RESULT_TYPE_FORM options={
assert result["errors"] is None
result = await hass.config_entries.flow.async_configure(
result["flow_id"],
{
CONF_ENTITY_ID: "switch.ceiling", CONF_ENTITY_ID: "switch.ceiling",
CONF_TARGET_DOMAIN: target_domain, CONF_TARGET_DOMAIN: target_domain,
}, },
title="ABC",
) )
await hass.async_block_till_done() switch_as_x_config_entry.add_to_hass(hass)
assert result["type"] == RESULT_TYPE_CREATE_ENTRY assert await hass.config_entries.async_setup(switch_as_x_config_entry.entry_id)
await hass.async_block_till_done()
config_entry = hass.config_entries.async_entries(DOMAIN)[0] config_entry = hass.config_entries.async_entries(DOMAIN)[0]
assert config_entry assert config_entry

105
tests/components/switch_as_x/test_init.py
View file

@ -6,23 +6,31 @@ from unittest.mock import patch
import pytest import pytest
from homeassistant.components.switch_as_x.const import CONF_TARGET_DOMAIN, DOMAIN from homeassistant.components.switch_as_x.const import CONF_TARGET_DOMAIN, DOMAIN
from homeassistant.const import CONF_ENTITY_ID, STATE_OFF, STATE_ON, Platform from homeassistant.const import (
CONF_ENTITY_ID,
STATE_CLOSED,
STATE_LOCKED,
STATE_OFF,
STATE_ON,
STATE_OPEN,
STATE_UNLOCKED,
Platform,
)
from homeassistant.core import HomeAssistant from homeassistant.core import HomeAssistant
from homeassistant.helpers import device_registry as dr, entity_registry as er from homeassistant.helpers import device_registry as dr, entity_registry as er
from tests.common import MockConfigEntry from tests.common import MockConfigEntry
PLATFORMS_TO_TEST = (
@pytest.mark.parametrize(
"target_domain",
(
Platform.COVER, Platform.COVER,
Platform.FAN, Platform.FAN,
Platform.LIGHT, Platform.LIGHT,
Platform.LOCK, Platform.LOCK,
Platform.SIREN, Platform.SIREN,
),
) )
@pytest.mark.parametrize("target_domain", PLATFORMS_TO_TEST)
async def test_config_entry_unregistered_uuid( async def test_config_entry_unregistered_uuid(
hass: HomeAssistant, target_domain: str hass: HomeAssistant, target_domain: str
) -> None: ) -> None:
@ -48,19 +56,23 @@ async def test_config_entry_unregistered_uuid(
@pytest.mark.parametrize( @pytest.mark.parametrize(
"target_domain", "target_domain,state_on,state_off",
( (
Platform.FAN, (Platform.COVER, STATE_OPEN, STATE_CLOSED),
Platform.LIGHT, (Platform.FAN, STATE_ON, STATE_OFF),
Platform.SIREN, (Platform.LIGHT, STATE_ON, STATE_OFF),
(Platform.LOCK, STATE_UNLOCKED, STATE_LOCKED),
(Platform.SIREN, STATE_ON, STATE_OFF),
), ),
) )
async def test_entity_registry_events(hass: HomeAssistant, target_domain: str) -> None: async def test_entity_registry_events(
hass: HomeAssistant, target_domain: str, state_on: str, state_off: str
) -> None:
"""Test entity registry events are tracked.""" """Test entity registry events are tracked."""
registry = er.async_get(hass) registry = er.async_get(hass)
registry_entry = registry.async_get_or_create("switch", "test", "unique") registry_entry = registry.async_get_or_create("switch", "test", "unique")
switch_entity_id = registry_entry.entity_id switch_entity_id = registry_entry.entity_id
hass.states.async_set(switch_entity_id, "on") hass.states.async_set(switch_entity_id, STATE_ON)
config_entry = MockConfigEntry( config_entry = MockConfigEntry(
data={}, data={},
@ -77,7 +89,7 @@ async def test_entity_registry_events(hass: HomeAssistant, target_domain: str) -
assert await hass.config_entries.async_setup(config_entry.entry_id) assert await hass.config_entries.async_setup(config_entry.entry_id)
await hass.async_block_till_done() await hass.async_block_till_done()
assert hass.states.get(f"{target_domain}.abc").state == STATE_ON assert hass.states.get(f"{target_domain}.abc").state == state_on
# Change entity_id # Change entity_id
new_switch_entity_id = f"{switch_entity_id}_new" new_switch_entity_id = f"{switch_entity_id}_new"
@ -87,12 +99,12 @@ async def test_entity_registry_events(hass: HomeAssistant, target_domain: str) -
# Check tracking the new entity_id # Check tracking the new entity_id
await hass.async_block_till_done() await hass.async_block_till_done()
assert hass.states.get(f"{target_domain}.abc").state == STATE_OFF assert hass.states.get(f"{target_domain}.abc").state == state_off
# The old entity_id should no longer be tracked # The old entity_id should no longer be tracked
hass.states.async_set(switch_entity_id, STATE_ON) hass.states.async_set(switch_entity_id, STATE_ON)
await hass.async_block_till_done() await hass.async_block_till_done()
assert hass.states.get(f"{target_domain}.abc").state == STATE_OFF assert hass.states.get(f"{target_domain}.abc").state == state_off
# Check changing name does not reload the config entry # Check changing name does not reload the config entry
with patch( with patch(
@ -111,16 +123,7 @@ async def test_entity_registry_events(hass: HomeAssistant, target_domain: str) -
assert len(hass.config_entries.async_entries("switch_as_x")) == 0 assert len(hass.config_entries.async_entries("switch_as_x")) == 0
@pytest.mark.parametrize( @pytest.mark.parametrize("target_domain", PLATFORMS_TO_TEST)
"target_domain",
(
Platform.COVER,
Platform.FAN,
Platform.LIGHT,
Platform.LOCK,
Platform.SIREN,
),
)
async def test_device_registry_config_entry_1( async def test_device_registry_config_entry_1(
hass: HomeAssistant, target_domain: str hass: HomeAssistant, target_domain: str
) -> None: ) -> None:
@ -178,16 +181,7 @@ async def test_device_registry_config_entry_1(
assert switch_as_x_config_entry.entry_id not in device_entry.config_entries assert switch_as_x_config_entry.entry_id not in device_entry.config_entries
@pytest.mark.parametrize( @pytest.mark.parametrize("target_domain", PLATFORMS_TO_TEST)
"target_domain",
(
Platform.COVER,
Platform.FAN,
Platform.LIGHT,
Platform.LOCK,
Platform.SIREN,
),
)
async def test_device_registry_config_entry_2( async def test_device_registry_config_entry_2(
hass: HomeAssistant, target_domain: str hass: HomeAssistant, target_domain: str
) -> None: ) -> None:
@ -238,16 +232,7 @@ async def test_device_registry_config_entry_2(
assert switch_as_x_config_entry.entry_id not in device_entry.config_entries assert switch_as_x_config_entry.entry_id not in device_entry.config_entries
@pytest.mark.parametrize( @pytest.mark.parametrize("target_domain", PLATFORMS_TO_TEST)
"target_domain",
(
Platform.COVER,
Platform.FAN,
Platform.LIGHT,
Platform.LOCK,
Platform.SIREN,
),
)
async def test_config_entry_entity_id( async def test_config_entry_entity_id(
hass: HomeAssistant, target_domain: Platform hass: HomeAssistant, target_domain: Platform
) -> None: ) -> None:
@ -282,16 +267,7 @@ async def test_config_entry_entity_id(
assert entity_entry.unique_id == config_entry.entry_id assert entity_entry.unique_id == config_entry.entry_id
@pytest.mark.parametrize( @pytest.mark.parametrize("target_domain", PLATFORMS_TO_TEST)
"target_domain",
(
Platform.COVER,
Platform.FAN,
Platform.LIGHT,
Platform.LOCK,
Platform.SIREN,
),
)
async def test_config_entry_uuid(hass: HomeAssistant, target_domain: Platform) -> None: async def test_config_entry_uuid(hass: HomeAssistant, target_domain: Platform) -> None:
"""Test light switch setup from config entry with entity registry id.""" """Test light switch setup from config entry with entity registry id."""
registry = er.async_get(hass) registry = er.async_get(hass)
@ -315,16 +291,7 @@ async def test_config_entry_uuid(hass: HomeAssistant, target_domain: Platform) -
assert hass.states.get(f"{target_domain}.abc") assert hass.states.get(f"{target_domain}.abc")
@pytest.mark.parametrize( @pytest.mark.parametrize("target_domain", PLATFORMS_TO_TEST)
"target_domain",
(
Platform.COVER,
Platform.FAN,
Platform.LIGHT,
Platform.LOCK,
Platform.SIREN,
),
)
async def test_device(hass: HomeAssistant, target_domain: Platform) -> None: async def test_device(hass: HomeAssistant, target_domain: Platform) -> None:
"""Test the entity is added to the wrapped entity's device.""" """Test the entity is added to the wrapped entity's device."""
device_registry = dr.async_get(hass) device_registry = dr.async_get(hass)
@ -360,7 +327,7 @@ async def test_device(hass: HomeAssistant, target_domain: Platform) -> None:
assert entity_entry.device_id == switch_entity_entry.device_id assert entity_entry.device_id == switch_entity_entry.device_id
@pytest.mark.parametrize("target_domain", (Platform.LIGHT,)) @pytest.mark.parametrize("target_domain", PLATFORMS_TO_TEST)
async def test_setup_and_remove_config_entry( async def test_setup_and_remove_config_entry(
hass: HomeAssistant, hass: HomeAssistant,
target_domain: Platform, target_domain: Platform,
@ -391,8 +358,8 @@ async def test_setup_and_remove_config_entry(
await hass.async_block_till_done() await hass.async_block_till_done()
# Check the state and entity registry entry are removed # Check the state and entity registry entry are removed
assert hass.states.get(f"{target_domain}.my_min_max") is None assert hass.states.get(f"{target_domain}.abc") is None
assert registry.async_get(f"{target_domain}.my_min_max") is None assert registry.async_get(f"{target_domain}.abc") is None
@pytest.mark.parametrize( @pytest.mark.parametrize(
@ -402,7 +369,7 @@ async def test_setup_and_remove_config_entry(
(er.RegistryEntryHider.INTEGRATION.value, None), (er.RegistryEntryHider.INTEGRATION.value, None),
), ),
) )
@pytest.mark.parametrize("target_domain", (Platform.LIGHT,)) @pytest.mark.parametrize("target_domain", PLATFORMS_TO_TEST)
async def test_reset_hidden_by( async def test_reset_hidden_by(
hass: HomeAssistant, hass: HomeAssistant,
target_domain: Platform, target_domain: Platform,