mirror of https://github.com/home-assistant/core
47 lines
1.3 KiB
Python
47 lines
1.3 KiB
Python
"""Tests for the Nettigo Air Monitor integration."""
|
|
|
|
from unittest.mock import AsyncMock, Mock, patch
|
|
|
|
from homeassistant.components.nam.const import DOMAIN
|
|
from homeassistant.core import HomeAssistant
|
|
|
|
from tests.common import MockConfigEntry, load_json_object_fixture
|
|
|
|
INCOMPLETE_NAM_DATA = {
|
|
"software_version": "NAMF-2020-36",
|
|
"sensordatavalues": [],
|
|
}
|
|
|
|
|
|
async def init_integration(
|
|
hass: HomeAssistant, co2_sensor: bool = True
|
|
) -> MockConfigEntry:
|
|
"""Set up the Nettigo Air Monitor integration in Home Assistant."""
|
|
entry = MockConfigEntry(
|
|
domain=DOMAIN,
|
|
title="10.10.2.3",
|
|
unique_id="aa:bb:cc:dd:ee:ff",
|
|
data={"host": "10.10.2.3"},
|
|
)
|
|
|
|
nam_data = load_json_object_fixture("nam/nam_data.json")
|
|
|
|
if not co2_sensor:
|
|
# Remove conc_co2_ppm value
|
|
nam_data["sensordatavalues"].pop(6)
|
|
|
|
update_response = Mock(json=AsyncMock(return_value=nam_data))
|
|
|
|
with (
|
|
patch("homeassistant.components.nam.NettigoAirMonitor.initialize"),
|
|
patch(
|
|
"homeassistant.components.nam.NettigoAirMonitor._async_http_request",
|
|
return_value=update_response,
|
|
),
|
|
):
|
|
entry.add_to_hass(hass)
|
|
await hass.config_entries.async_setup(entry.entry_id)
|
|
await hass.async_block_till_done()
|
|
|
|
return entry
|