mirror of
https://github.com/home-assistant/core.git
synced 2025-04-24 09:17:53 +00:00
1bd8ff884e
* improve snoo testing * change to asyncMock method of testing * Apply suggestions from code review Co-authored-by: Joost Lekkerkerker <joostlek@outlook.com> * address comments * Apply suggestions from code review Co-authored-by: Joost Lekkerkerker <joostlek@outlook.com> * adress comments --------- Co-authored-by: Joost Lekkerkerker <joostlek@outlook.com>
55 lines
1.5 KiB
Python
55 lines
1.5 KiB
Python
"""Tests for the Happiest Baby Snoo integration."""
|
|
|
|
from collections.abc import Awaitable, Callable
|
|
from unittest.mock import AsyncMock
|
|
|
|
import pytest
|
|
from python_snoo.containers import SnooData
|
|
|
|
from homeassistant.components.snoo.const import DOMAIN
|
|
from homeassistant.config_entries import ConfigEntry
|
|
from homeassistant.const import CONF_PASSWORD, CONF_USERNAME
|
|
from homeassistant.core import HomeAssistant
|
|
|
|
from tests.common import MockConfigEntry
|
|
|
|
|
|
def create_entry(
|
|
hass: HomeAssistant,
|
|
) -> ConfigEntry:
|
|
"""Add config entry in Home Assistant."""
|
|
entry = MockConfigEntry(
|
|
domain=DOMAIN,
|
|
title="test-username",
|
|
data={
|
|
CONF_USERNAME: "test-username",
|
|
CONF_PASSWORD: "sample",
|
|
},
|
|
# This is also gotten from the fake jwt
|
|
unique_id="123e4567-e89b-12d3-a456-426614174000",
|
|
version=1,
|
|
)
|
|
entry.add_to_hass(hass)
|
|
return entry
|
|
|
|
|
|
async def async_init_integration(hass: HomeAssistant) -> ConfigEntry:
|
|
"""Set up the Snoo integration in Home Assistant."""
|
|
|
|
entry = create_entry(hass)
|
|
|
|
await hass.config_entries.async_setup(entry.entry_id)
|
|
await hass.async_block_till_done()
|
|
|
|
return entry
|
|
|
|
|
|
def find_update_callback(
|
|
mock: AsyncMock, serial_number: str
|
|
) -> Callable[[SnooData], Awaitable[None]]:
|
|
"""Find the update callback for a specific identifier."""
|
|
for call in mock.subscribe.call_args_list:
|
|
if call[0][0].serialNumber == serial_number:
|
|
return call[0][1]
|
|
pytest.fail(f"Callback for identifier {serial_number} not found")
|