Update m* tests to use entity & device registry fixtures (#103980)

This commit is contained in:
Jan-Philipp Benecke 2023-11-15 10:33:11 +01:00 committed by GitHub
parent 51c1ea85f3
commit 9326ea09a5
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
46 changed files with 361 additions and 298 deletions

View file

@ -78,13 +78,14 @@ async def setup_media_source(hass) -> None:
assert await async_setup_component(hass, "media_source", {})
async def test_async_browse_media_success(hass: HomeAssistant) -> None:
async def test_async_browse_media_success(
hass: HomeAssistant, device_registry: dr.DeviceRegistry
) -> None:
"""Test successful browse media."""
client = create_mock_motioneye_client()
config = await setup_mock_motioneye_config_entry(hass, client=client)
device_registry = dr.async_get(hass)
device = device_registry.async_get_or_create(
config_entry_id=config.entry_id,
identifiers={TEST_CAMERA_DEVICE_IDENTIFIER},
@ -295,13 +296,14 @@ async def test_async_browse_media_success(hass: HomeAssistant) -> None:
}
async def test_async_browse_media_images_success(hass: HomeAssistant) -> None:
async def test_async_browse_media_images_success(
hass: HomeAssistant, device_registry: dr.DeviceRegistry
) -> None:
"""Test successful browse media of images."""
client = create_mock_motioneye_client()
config = await setup_mock_motioneye_config_entry(hass, client=client)
device_registry = dr.async_get(hass)
device = device_registry.async_get_or_create(
config_entry_id=config.entry_id,
identifiers={TEST_CAMERA_DEVICE_IDENTIFIER},
@ -346,14 +348,15 @@ async def test_async_browse_media_images_success(hass: HomeAssistant) -> None:
}
async def test_async_resolve_media_success(hass: HomeAssistant) -> None:
async def test_async_resolve_media_success(
hass: HomeAssistant, device_registry: dr.DeviceRegistry
) -> None:
"""Test successful resolve media."""
client = create_mock_motioneye_client()
config = await setup_mock_motioneye_config_entry(hass, client=client)
device_registry = dr.async_get(hass)
device = device_registry.async_get_or_create(
config_entry_id=config.entry_id,
identifiers={TEST_CAMERA_DEVICE_IDENTIFIER},
@ -380,14 +383,15 @@ async def test_async_resolve_media_success(hass: HomeAssistant) -> None:
assert client.get_image_url.call_args == call(TEST_CAMERA_ID, "/foo.jpg")
async def test_async_resolve_media_failure(hass: HomeAssistant) -> None:
async def test_async_resolve_media_failure(
hass: HomeAssistant, device_registry: dr.DeviceRegistry
) -> None:
"""Test failed resolve media calls."""
client = create_mock_motioneye_client()
config = await setup_mock_motioneye_config_entry(hass, client=client)
device_registry = dr.async_get(hass)
device = device_registry.async_get_or_create(
config_entry_id=config.entry_id,
identifiers={TEST_CAMERA_DEVICE_IDENTIFIER},