86 lines
2.8 KiB
Python
86 lines
2.8 KiB
Python
|
"""Tests for the Anova integration."""
|
||
|
|
from __future__ import annotations
|
||
|
|
|
||
|
|
from unittest.mock import patch
|
||
|
|
|
||
|
|
from anova_wifi import (
|
||
|
|
AnovaPrecisionCooker,
|
||
|
|
AnovaPrecisionCookerBinarySensor,
|
||
|
|
AnovaPrecisionCookerSensor,
|
||
|
|
)
|
||
|
|
|
||
|
|
from homeassistant.components.anova.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
|
||
|
|
|
||
|
|
DEVICE_UNIQUE_ID = "abc123def"
|
||
|
|
|
||
|
|
CONF_INPUT = {CONF_USERNAME: "sample@gmail.com", CONF_PASSWORD: "sample"}
|
||
|
|
|
||
|
|
ONLINE_UPDATE = {
|
||
|
|
"sensors": {
|
||
|
|
AnovaPrecisionCookerSensor.COOK_TIME: 0,
|
||
|
|
AnovaPrecisionCookerSensor.MODE: "Low water",
|
||
|
|
AnovaPrecisionCookerSensor.STATE: "No state",
|
||
|
|
AnovaPrecisionCookerSensor.TARGET_TEMPERATURE: 23.33,
|
||
|
|
AnovaPrecisionCookerSensor.COOK_TIME_REMAINING: 0,
|
||
|
|
AnovaPrecisionCookerSensor.FIRMWARE_VERSION: "2.2.0",
|
||
|
|
AnovaPrecisionCookerSensor.HEATER_TEMPERATURE: 20.87,
|
||
|
|
AnovaPrecisionCookerSensor.TRIAC_TEMPERATURE: 21.79,
|
||
|
|
AnovaPrecisionCookerSensor.WATER_TEMPERATURE: 21.33,
|
||
|
|
},
|
||
|
|
"binary_sensors": {
|
||
|
|
AnovaPrecisionCookerBinarySensor.COOKING: False,
|
||
|
|
AnovaPrecisionCookerBinarySensor.DEVICE_SAFE: True,
|
||
|
|
AnovaPrecisionCookerBinarySensor.WATER_LEAK: False,
|
||
|
|
AnovaPrecisionCookerBinarySensor.WATER_LEVEL_CRITICAL: True,
|
||
|
|
AnovaPrecisionCookerBinarySensor.WATER_TEMP_TOO_HIGH: False,
|
||
|
|
},
|
||
|
|
}
|
||
|
|
|
||
|
|
|
||
|
|
def create_entry(hass: HomeAssistant, device_id: str = DEVICE_UNIQUE_ID) -> ConfigEntry:
|
||
|
|
"""Add config entry in Home Assistant."""
|
||
|
|
entry = MockConfigEntry(
|
||
|
|
domain=DOMAIN,
|
||
|
|
title="Anova",
|
||
|
|
data={
|
||
|
|
CONF_USERNAME: "sample@gmail.com",
|
||
|
|
CONF_PASSWORD: "sample",
|
||
|
|
"devices": [(device_id, "type_sample")],
|
||
|
|
},
|
||
|
|
unique_id="sample@gmail.com",
|
||
|
|
)
|
||
|
|
entry.add_to_hass(hass)
|
||
|
|
return entry
|
||
|
|
|
||
|
|
|
||
|
|
async def async_init_integration(
|
||
|
|
hass: HomeAssistant,
|
||
|
|
skip_setup: bool = False,
|
||
|
|
error: str | None = None,
|
||
|
|
) -> ConfigEntry:
|
||
|
|
"""Set up the Anova integration in Home Assistant."""
|
||
|
|
with patch(
|
||
|
|
"homeassistant.components.anova.coordinator.AnovaPrecisionCooker.update"
|
||
|
|
) as update_patch, patch(
|
||
|
|
"homeassistant.components.anova.AnovaApi.authenticate"
|
||
|
|
), patch(
|
||
|
|
"homeassistant.components.anova.AnovaApi.get_devices"
|
||
|
|
) as device_patch:
|
||
|
|
update_patch.return_value = ONLINE_UPDATE
|
||
|
|
device_patch.return_value = [
|
||
|
|
AnovaPrecisionCooker(None, DEVICE_UNIQUE_ID, "type_sample", None)
|
||
|
|
]
|
||
|
|
|
||
|
|
entry = create_entry(hass)
|
||
|
|
|
||
|
|
if not skip_setup:
|
||
|
|
await hass.config_entries.async_setup(entry.entry_id)
|
||
|
|
await hass.async_block_till_done()
|
||
|
|
|
||
|
|
return entry
|