Home-Assistant/core
1
0
Fork 0
mirror of https://github.com/home-assistant/core.git synced 2025-12-24 21:06:19 +00:00
Code Activity

Clean up unnecessary registry mocks from integration tests (#87763)

Browse Source
This commit is contained in:
Franck Nijhof
2023-02-09 10:43:45 +01:00
committed by GitHub
parent 59890cca68
commit 330e1c6cbb
45 changed files with 544 additions and 628 deletions
Download Patch File Download Diff File Expand all files Collapse all files

34
tests/components/broadlink/test_remote.py
View File

@@ -11,12 +11,10 @@ from homeassistant.components.remote import (
)
from homeassistant.const import ATTR_FRIENDLY_NAME, STATE_OFF, STATE_ON, Platform
from homeassistant.core import HomeAssistant
from homeassistant.helpers.entity_registry import async_entries_for_device
from homeassistant.helpers import device_registry as dr, entity_registry as er
from . import get_device
from tests.common import mock_device_registry, mock_registry
REMOTE_DEVICES = ["Entrance", "Living Room", "Office", "Garage"]
IR_PACKET = (
@@ -25,17 +23,19 @@ IR_PACKET = (
)
async def test_remote_setup_works(hass: HomeAssistant) -> None:
async def test_remote_setup_works(
hass: HomeAssistant,
device_registry: dr.DeviceRegistry,
entity_registry: er.EntityRegistry,
) -> None:
"""Test a successful setup with all remotes."""
for device in map(get_device, REMOTE_DEVICES):
device_registry = mock_device_registry(hass)
entity_registry = mock_registry(hass)
mock_setup = await device.setup_entry(hass)
device_entry = device_registry.async_get_device(
{(DOMAIN, mock_setup.entry.unique_id)}
)
entries = async_entries_for_device(entity_registry, device_entry.id)
entries = er.async_entries_for_device(entity_registry, device_entry.id)
remotes = [entry for entry in entries if entry.domain == Platform.REMOTE]
assert len(remotes) == 1
@@ -48,17 +48,19 @@ async def test_remote_setup_works(hass: HomeAssistant) -> None:
assert mock_setup.api.auth.call_count == 1
async def test_remote_send_command(hass: HomeAssistant) -> None:
async def test_remote_send_command(
hass: HomeAssistant,
device_registry: dr.DeviceRegistry,
entity_registry: er.EntityRegistry,
) -> None:
"""Test sending a command with all remotes."""
for device in map(get_device, REMOTE_DEVICES):
device_registry = mock_device_registry(hass)
entity_registry = mock_registry(hass)
mock_setup = await device.setup_entry(hass)
device_entry = device_registry.async_get_device(
{(DOMAIN, mock_setup.entry.unique_id)}
)
entries = async_entries_for_device(entity_registry, device_entry.id)
entries = er.async_entries_for_device(entity_registry, device_entry.id)
remotes = [entry for entry in entries if entry.domain == Platform.REMOTE]
assert len(remotes) == 1
@@ -75,17 +77,19 @@ async def test_remote_send_command(hass: HomeAssistant) -> None:
assert mock_setup.api.auth.call_count == 1
async def test_remote_turn_off_turn_on(hass: HomeAssistant) -> None:
async def test_remote_turn_off_turn_on(
hass: HomeAssistant,
device_registry: dr.DeviceRegistry,
entity_registry: er.EntityRegistry,
) -> None:
"""Test we do not send commands if the remotes are off."""
for device in map(get_device, REMOTE_DEVICES):
device_registry = mock_device_registry(hass)
entity_registry = mock_registry(hass)
mock_setup = await device.setup_entry(hass)
device_entry = device_registry.async_get_device(
{(DOMAIN, mock_setup.entry.unique_id)}
)
entries = async_entries_for_device(entity_registry, device_entry.id)
entries = er.async_entries_for_device(entity_registry, device_entry.id)
remotes = [entry for entry in entries if entry.domain == Platform.REMOTE]
assert len(remotes) == 1