"""The tests for the Netatmo climate platform."""
from unittest.mock import Mock

import pytest

from homeassistant.components.climate import (
    DOMAIN as CLIMATE_DOMAIN,
    SERVICE_SET_HVAC_MODE,
    SERVICE_SET_PRESET_MODE,
    SERVICE_SET_TEMPERATURE,
    SERVICE_TURN_OFF,
    SERVICE_TURN_ON,
)
from homeassistant.components.climate.const import (
    ATTR_HVAC_MODE,
    ATTR_PRESET_MODE,
    HVAC_MODE_AUTO,
    HVAC_MODE_HEAT,
    HVAC_MODE_OFF,
    PRESET_AWAY,
    PRESET_BOOST,
)
from homeassistant.components.netatmo import climate
from homeassistant.components.netatmo.climate import (
    NA_THERM,
    NA_VALVE,
    PRESET_FROST_GUARD,
    PRESET_SCHEDULE,
)
from homeassistant.components.netatmo.const import (
    ATTR_SCHEDULE_NAME,
    SERVICE_SET_SCHEDULE,
)
from homeassistant.components.webhook import async_handle_webhook
from homeassistant.const import ATTR_ENTITY_ID, ATTR_TEMPERATURE, CONF_WEBHOOK_ID
from homeassistant.util.aiohttp import MockRequest


async def simulate_webhook(hass, webhook_id, response):
    """Simulate a webhook event."""
    request = MockRequest(content=response, mock_source="test")
    await async_handle_webhook(hass, webhook_id, request)
    await hass.async_block_till_done()


async def test_webhook_event_handling_thermostats(hass, climate_entry):
    """Test service and webhook event handling with thermostats."""
    webhook_id = climate_entry.data[CONF_WEBHOOK_ID]
    climate_entity_livingroom = "climate.netatmo_livingroom"

    assert hass.states.get(climate_entity_livingroom).state == "auto"
    assert (
        hass.states.get(climate_entity_livingroom).attributes["preset_mode"]
        == "Schedule"
    )
    assert hass.states.get(climate_entity_livingroom).attributes["temperature"] == 12

    # Test service setting the temperature
    await hass.services.async_call(
        CLIMATE_DOMAIN,
        SERVICE_SET_TEMPERATURE,
        {ATTR_ENTITY_ID: climate_entity_livingroom, ATTR_TEMPERATURE: 21},
        blocking=True,
    )
    await hass.async_block_till_done()

    # Fake webhook thermostat manual set point
    response = (
        b'{"user_id": "91763b24c43d3e344f424e8b", "user": {"id": "91763b24c43d3e344f424e8b",'
        b'"email": "john@doe.com"}, "home_id": "91763b24c43d3e344f424e8b", "room_id": "2746182631",'
        b'"home": { "id": "91763b24c43d3e344f424e8b", "name": "MYHOME", "country": "DE",'
        b'"rooms": [{ "id": "2746182631", "name": "Livingroom", "type": "livingroom",'
        b'"therm_setpoint_mode": "manual", "therm_setpoint_temperature": 21,'
        b'"therm_setpoint_end_time": 1612734552}], "modules": [{"id": "12:34:56:00:01:ae",'
        b'"name": "Livingroom", "type": "NATherm1"}]}, "mode": "manual", "event_type": "set_point",'
        b'"temperature": 21, "push_type": "display_change"}'
    )
    await simulate_webhook(hass, webhook_id, response)

    assert hass.states.get(climate_entity_livingroom).state == "heat"
    assert (
        hass.states.get(climate_entity_livingroom).attributes["preset_mode"]
        == "Schedule"
    )
    assert hass.states.get(climate_entity_livingroom).attributes["temperature"] == 21

    # Test service setting the HVAC mode to "heat"
    await hass.services.async_call(
        CLIMATE_DOMAIN,
        SERVICE_SET_HVAC_MODE,
        {ATTR_ENTITY_ID: climate_entity_livingroom, ATTR_HVAC_MODE: HVAC_MODE_HEAT},
        blocking=True,
    )
    await hass.async_block_till_done()

    # Fake webhook thermostat mode change to "Max"
    response = (
        b'{"user_id": "91763b24c43d3e344f424e8b", "user": {"id": "91763b24c43d3e344f424e8b",'
        b'"email": "john@doe.com"}, "home_id": "91763b24c43d3e344f424e8b", "room_id": "2746182631",'
        b'"home": {"id": "91763b24c43d3e344f424e8b", "name": "MYHOME", "country": "DE",'
        b'"rooms": [{"id": "2746182631", "name": "Livingroom", "type": "livingroom",'
        b'"therm_setpoint_mode": "max", "therm_setpoint_end_time": 1612749189}],'
        b'"modules": [{"id": "12:34:56:00:01:ae", "name": "Livingroom", "type": "NATherm1"}]},'
        b'"mode": "max", "event_type": "set_point", "push_type": "display_change"}'
    )
    await simulate_webhook(hass, webhook_id, response)

    assert hass.states.get(climate_entity_livingroom).state == "heat"
    assert hass.states.get(climate_entity_livingroom).attributes["temperature"] == 30

    # Test service setting the HVAC mode to "off"
    await hass.services.async_call(
        CLIMATE_DOMAIN,
        SERVICE_SET_HVAC_MODE,
        {ATTR_ENTITY_ID: climate_entity_livingroom, ATTR_HVAC_MODE: HVAC_MODE_OFF},
        blocking=True,
    )
    await hass.async_block_till_done()

    # Fake webhook turn thermostat off
    response = (
        b'{"user_id": "91763b24c43d3e344f424e8b", "user": {"id": "91763b24c43d3e344f424e8b",'
        b'"email": "john@doe.com"}, "home_id": "91763b24c43d3e344f424e8b", "room_id": "2746182631",'
        b'"home": {"id": "91763b24c43d3e344f424e8b","name": "MYHOME","country": "DE",'
        b'"rooms": [{"id": "2746182631","name": "Livingroom","type": "livingroom",'
        b'"therm_setpoint_mode": "off"}],"modules": [{"id": "12:34:56:00:01:ae",'
        b'"name": "Livingroom", "type": "NATherm1"}]}, "mode": "off", "event_type": "set_point",'
        b'"push_type": "display_change"}'
    )
    await simulate_webhook(hass, webhook_id, response)

    assert hass.states.get(climate_entity_livingroom).state == "off"

    # Test service setting the HVAC mode to "auto"
    await hass.services.async_call(
        CLIMATE_DOMAIN,
        SERVICE_SET_HVAC_MODE,
        {ATTR_ENTITY_ID: climate_entity_livingroom, ATTR_HVAC_MODE: HVAC_MODE_AUTO},
        blocking=True,
    )
    await hass.async_block_till_done()

    # Fake webhook thermostat mode cancel set point
    response = (
        b'{"user_id": "91763b24c43d3e344f424e8b", "user": {"id": "91763b24c43d3e344f424e8b",'
        b'"email": "john@doe.com"}, "home_id": "91763b24c43d3e344f424e8b","room_id": "2746182631",'
        b'"home": {"id": "91763b24c43d3e344f424e8b","name": "MYHOME","country": "DE",'
        b'"rooms": [{"id": "2746182631","name": "Livingroom","type": "livingroom",'
        b'"therm_setpoint_mode": "home"}], "modules": [{"id": "12:34:56:00:01:ae",'
        b'"name": "Livingroom", "type": "NATherm1"}]}, "mode": "home",'
        b'"event_type": "cancel_set_point", "push_type": "display_change"}'
    )
    await simulate_webhook(hass, webhook_id, response)

    assert hass.states.get(climate_entity_livingroom).state == "auto"
    assert (
        hass.states.get(climate_entity_livingroom).attributes["preset_mode"]
        == "Schedule"
    )


async def test_service_preset_mode_frost_guard_thermostat(hass, climate_entry):
    """Test service with frost guard preset for thermostats."""
    webhook_id = climate_entry.data[CONF_WEBHOOK_ID]
    climate_entity_livingroom = "climate.netatmo_livingroom"

    assert hass.states.get(climate_entity_livingroom).state == "auto"
    assert (
        hass.states.get(climate_entity_livingroom).attributes["preset_mode"]
        == "Schedule"
    )

    # Test service setting the preset mode to "frost guard"
    await hass.services.async_call(
        CLIMATE_DOMAIN,
        SERVICE_SET_PRESET_MODE,
        {
            ATTR_ENTITY_ID: climate_entity_livingroom,
            ATTR_PRESET_MODE: PRESET_FROST_GUARD,
        },
        blocking=True,
    )
    await hass.async_block_till_done()

    # Fake webhook thermostat mode change to "Frost Guard"
    response = (
        b'{"user_id": "91763b24c43d3e344f424e8b","user": {"id": "91763b24c43d3e344f424e8b",'
        b'"email": "john@doe.com"}, "home_id": "91763b24c43d3e344f424e8b",'
        b'"event_type": "therm_mode", "home": {"id": "91763b24c43d3e344f424e8b",'
        b'"therm_mode": "hg"}, "mode": "hg", "previous_mode": "schedule",'
        b'"push_type":"home_event_changed"}'
    )
    await simulate_webhook(hass, webhook_id, response)

    assert hass.states.get(climate_entity_livingroom).state == "auto"
    assert (
        hass.states.get(climate_entity_livingroom).attributes["preset_mode"]
        == "Frost Guard"
    )

    # Test service setting the preset mode to "frost guard"
    await hass.services.async_call(
        CLIMATE_DOMAIN,
        SERVICE_SET_PRESET_MODE,
        {
            ATTR_ENTITY_ID: climate_entity_livingroom,
            ATTR_PRESET_MODE: PRESET_SCHEDULE,
        },
        blocking=True,
    )
    await hass.async_block_till_done()

    # Test webhook thermostat mode change to "Schedule"
    response = (
        b'{"user_id": "91763b24c43d3e344f424e8b","user": {"id": "91763b24c43d3e344f424e8b",'
        b'"email": "john@doe.com"}, "home_id": "91763b24c43d3e344f424e8b",'
        b'"event_type": "therm_mode", "home": {"id": "91763b24c43d3e344f424e8b",'
        b'"therm_mode": "schedule"}, "mode": "schedule", "previous_mode": "hg",'
        b'"push_type": "home_event_changed"}'
    )
    await simulate_webhook(hass, webhook_id, response)

    assert hass.states.get(climate_entity_livingroom).state == "auto"
    assert (
        hass.states.get(climate_entity_livingroom).attributes["preset_mode"]
        == "Schedule"
    )


async def test_service_preset_modes_thermostat(hass, climate_entry):
    """Test service with preset modes for thermostats."""
    webhook_id = climate_entry.data[CONF_WEBHOOK_ID]
    climate_entity_livingroom = "climate.netatmo_livingroom"

    assert hass.states.get(climate_entity_livingroom).state == "auto"
    assert (
        hass.states.get(climate_entity_livingroom).attributes["preset_mode"]
        == "Schedule"
    )

    # Test service setting the preset mode to "away"
    await hass.services.async_call(
        CLIMATE_DOMAIN,
        SERVICE_SET_PRESET_MODE,
        {ATTR_ENTITY_ID: climate_entity_livingroom, ATTR_PRESET_MODE: PRESET_AWAY},
        blocking=True,
    )
    await hass.async_block_till_done()

    # Fake webhook thermostat mode change to "Away"
    response = (
        b'{"user_id": "91763b24c43d3e344f424e8b","user": {"id": "91763b24c43d3e344f424e8b",'
        b'"email": "john@doe.com"}, "home_id": "91763b24c43d3e344f424e8b", '
        b'"event_type": "therm_mode","home": {"id": "91763b24c43d3e344f424e8b",'
        b'"therm_mode": "away"},"mode": "away","previous_mode": "schedule",'
        b'"push_type": "home_event_changed"}'
    )
    await simulate_webhook(hass, webhook_id, response)

    assert hass.states.get(climate_entity_livingroom).state == "auto"
    assert (
        hass.states.get(climate_entity_livingroom).attributes["preset_mode"] == "away"
    )

    # Test service setting the preset mode to "boost"
    await hass.services.async_call(
        CLIMATE_DOMAIN,
        SERVICE_SET_PRESET_MODE,
        {ATTR_ENTITY_ID: climate_entity_livingroom, ATTR_PRESET_MODE: PRESET_BOOST},
        blocking=True,
    )
    await hass.async_block_till_done()

    # TFakeest webhook thermostat mode change to "Max"
    response = (
        b'{"user_id": "91763b24c43d3e344f424e8b", "user": {"id": "91763b24c43d3e344f424e8b",'
        b'"email":"john@doe.com"}, "home_id": "91763b24c43d3e344f424e8b", "room_id": "2746182631",'
        b'"home": {"id": "91763b24c43d3e344f424e8b", "name": "MYHOME", "country": "DE",'
        b'"rooms": [{"id": "2746182631", "name": "Livingroom", "type": "livingroom",'
        b'"therm_setpoint_mode": "max", "therm_setpoint_end_time": 1612749189}],'
        b'"modules": [{"id": "12:34:56:00:01:ae", "name": "Livingroom", "type": "NATherm1"}]},'
        b'"mode": "max", "event_type": "set_point", "push_type": "display_change"}'
    )
    await simulate_webhook(hass, webhook_id, response)

    assert hass.states.get(climate_entity_livingroom).state == "heat"
    assert hass.states.get(climate_entity_livingroom).attributes["temperature"] == 30


async def test_webhook_event_handling_no_data(hass, climate_entry):
    """Test service and webhook event handling with erroneous data."""
    # Test webhook without home entry
    webhook_id = climate_entry.data[CONF_WEBHOOK_ID]

    response = (
        b'{"user_id": "91763b24c43d3e344f424e8b", "user": {"id": "91763b24c43d3e344f424e8b",'
        b'"email": "john@doe.com"}, "home_id": "91763b24c43d3e344f424e8b",'
        b'"push_type": "home_event_changed"}'
    )
    await simulate_webhook(hass, webhook_id, response)

    # Test webhook with different home id
    response = (
        b'{"user_id": "91763b24c43d3e344f424e8b", "user": {"id": "91763b24c43d3e344f424e8b",'
        b'"email": "john@doe.com"}, "home_id": "3d3e344f491763b24c424e8b",'
        b'"room_id": "2746182631", "home": {"id": "3d3e344f491763b24c424e8b",'
        b'"name": "MYHOME","country": "DE", "rooms": [], "modules": []}, "mode": "home",'
        b'"event_type": "cancel_set_point", "push_type": "display_change"}'
    )
    await simulate_webhook(hass, webhook_id, response)

    # Test webhook without room entries
    response = (
        b'{"user_id": "91763b24c43d3e344f424e8b", "user": {"id": "91763b24c43d3e344f424e8b",'
        b'"email": "john@doe.com"}, "home_id": "91763b24c43d3e344f424e8b",'
        b'"room_id": "2746182631", "home": {"id": "91763b24c43d3e344f424e8b",'
        b'"name": "MYHOME",  "country": "DE",  "rooms": [], "modules": []}, "mode": "home",'
        b'"event_type": "cancel_set_point","push_type": "display_change"}'
    )
    await simulate_webhook(hass, webhook_id, response)


async def test_service_schedule_thermostats(hass, climate_entry, caplog):
    """Test service for selecting Netatmo schedule with thermostats."""
    climate_entity_livingroom = "climate.netatmo_livingroom"

    # Test setting a valid schedule
    await hass.services.async_call(
        "netatmo",
        SERVICE_SET_SCHEDULE,
        {ATTR_ENTITY_ID: climate_entity_livingroom, ATTR_SCHEDULE_NAME: "Winter"},
        blocking=True,
    )
    await hass.async_block_till_done()

    assert (
        "Setting 91763b24c43d3e344f424e8b schedule to Winter (b1b54a2f45795764f59d50d8)"
        in caplog.text
    )

    # Test setting an invalid schedule
    await hass.services.async_call(
        "netatmo",
        SERVICE_SET_SCHEDULE,
        {ATTR_ENTITY_ID: climate_entity_livingroom, ATTR_SCHEDULE_NAME: "summer"},
        blocking=True,
    )
    await hass.async_block_till_done()

    assert "summer is not a invalid schedule" in caplog.text


async def test_service_preset_mode_already_boost_valves(hass, climate_entry):
    """Test service with boost preset for valves when already in boost mode."""
    webhook_id = climate_entry.data[CONF_WEBHOOK_ID]
    climate_entity_entrada = "climate.netatmo_entrada"

    assert hass.states.get(climate_entity_entrada).state == "auto"
    assert (
        hass.states.get(climate_entity_entrada).attributes["preset_mode"]
        == "Frost Guard"
    )
    assert hass.states.get(climate_entity_entrada).attributes["temperature"] == 7

    # Test webhook valve mode change to "Max"
    response = (
        b'{"user_id": "91763b24c43d3e344f424e8b", "user": {"id": "91763b24c43d3e344f424e8b",'
        b'"email": "john@doe.com"}, "home_id": "91763b24c43d3e344f424e8b",'
        b'"room_id": "2833524037", "home": {"id": "91763b24c43d3e344f424e8b", "name": "MYHOME",'
        b'"country": "DE","rooms": [{"id": "2833524037", "name": "Entrada", "type": "lobby",'
        b'"therm_setpoint_mode": "max", "therm_setpoint_end_time": 1612749189}],'
        b'"modules": [{"id": "12:34:56:00:01:ae", "name": "Entrada", "type": "NRV"}]},'
        b'"mode": "max","event_type": "set_point","push_type": "display_change"}'
    )
    await simulate_webhook(hass, webhook_id, response)

    # Test service setting the preset mode to "boost"
    await hass.services.async_call(
        CLIMATE_DOMAIN,
        SERVICE_SET_PRESET_MODE,
        {ATTR_ENTITY_ID: climate_entity_entrada, ATTR_PRESET_MODE: PRESET_BOOST},
        blocking=True,
    )
    await hass.async_block_till_done()

    # Test webhook valve mode change to "Max"
    response = (
        b'{"user_id": "91763b24c43d3e344f424e8b", "user": {"id": "91763b24c43d3e344f424e8b",'
        b'"email": "john@doe.com"}, "home_id": "91763b24c43d3e344f424e8b",'
        b'"room_id": "2833524037", "home": {"id": "91763b24c43d3e344f424e8b",'
        b'"name": "MYHOME","country": "DE","rooms": [{"id": "2833524037", "name": "Entrada",'
        b'"type": "lobby", "therm_setpoint_mode": "max", "therm_setpoint_end_time": 1612749189}],'
        b'"modules": [{"id": "12:34:56:00:01:ae", "name": "Entrada", "type": "NRV"}]},'
        b'"mode": "max", "event_type": "set_point", "push_type": "display_change"}'
    )
    await simulate_webhook(hass, webhook_id, response)

    assert hass.states.get(climate_entity_entrada).state == "heat"
    assert hass.states.get(climate_entity_entrada).attributes["temperature"] == 30


async def test_service_preset_mode_boost_valves(hass, climate_entry):
    """Test service with boost preset for valves."""
    webhook_id = climate_entry.data[CONF_WEBHOOK_ID]
    climate_entity_entrada = "climate.netatmo_entrada"

    # Test service setting the preset mode to "boost"
    assert hass.states.get(climate_entity_entrada).state == "auto"
    assert hass.states.get(climate_entity_entrada).attributes["temperature"] == 7

    await hass.services.async_call(
        CLIMATE_DOMAIN,
        SERVICE_SET_PRESET_MODE,
        {ATTR_ENTITY_ID: climate_entity_entrada, ATTR_PRESET_MODE: PRESET_BOOST},
        blocking=True,
    )
    await hass.async_block_till_done()

    # Fake backend response
    response = (
        b'{"user_id": "91763b24c43d3e344f424e8b", "user": {"id": "91763b24c43d3e344f424e8b",'
        b'"email": "john@doe.com"}, "home_id": "91763b24c43d3e344f424e8b",'
        b'"room_id": "2833524037", "home": {"id": "91763b24c43d3e344f424e8b", "name": "MYHOME",'
        b'"country": "DE", "rooms": [{"id": "2833524037","name": "Entrada","type": "lobby",'
        b'"therm_setpoint_mode": "max","therm_setpoint_end_time": 1612749189}],'
        b'"modules": [{"id": "12:34:56:00:01:ae", "name": "Entrada", "type": "NRV"}]},'
        b'"mode": "max", "event_type": "set_point", "push_type": "display_change"}'
    )
    await simulate_webhook(hass, webhook_id, response)

    assert hass.states.get(climate_entity_entrada).state == "heat"
    assert hass.states.get(climate_entity_entrada).attributes["temperature"] == 30


async def test_service_preset_mode_invalid(hass, climate_entry, caplog):
    """Test service with invalid preset."""
    await hass.services.async_call(
        CLIMATE_DOMAIN,
        SERVICE_SET_PRESET_MODE,
        {ATTR_ENTITY_ID: "climate.netatmo_cocina", ATTR_PRESET_MODE: "invalid"},
        blocking=True,
    )
    await hass.async_block_till_done()

    assert "Preset mode 'invalid' not available" in caplog.text


async def test_valves_service_turn_off(hass, climate_entry):
    """Test service turn off for valves."""
    webhook_id = climate_entry.data[CONF_WEBHOOK_ID]
    climate_entity_entrada = "climate.netatmo_entrada"

    # Test turning valve off
    await hass.services.async_call(
        CLIMATE_DOMAIN,
        SERVICE_TURN_OFF,
        {ATTR_ENTITY_ID: climate_entity_entrada},
        blocking=True,
    )
    await hass.async_block_till_done()

    # Fake backend response for valve being turned off
    response = (
        b'{"user_id": "91763b24c43d3e344f424e8b", "user": {"id": "91763b24c43d3e344f424e8b",'
        b'"email": "john@doe.com"}, "home_id": "91763b24c43d3e344f424e8b", "room_id": "2833524037",'
        b'"home": {"id": "91763b24c43d3e344f424e8b","name": "MYHOME","country": "DE",'
        b'"rooms": [{"id": "2833524037","name": "Entrada","type": "lobby",'
        b'"therm_setpoint_mode": "off"}], "modules": [{"id": "12:34:56:00:01:ae","name": "Entrada",'
        b'"type": "NRV"}]}, "mode": "off", "event_type": "set_point", "push_type":"display_change"}'
    )
    await simulate_webhook(hass, webhook_id, response)

    assert hass.states.get(climate_entity_entrada).state == "off"


async def test_valves_service_turn_on(hass, climate_entry):
    """Test service turn on for valves."""
    webhook_id = climate_entry.data[CONF_WEBHOOK_ID]
    climate_entity_entrada = "climate.netatmo_entrada"

    # Test turning valve on
    await hass.services.async_call(
        CLIMATE_DOMAIN,
        SERVICE_TURN_ON,
        {ATTR_ENTITY_ID: climate_entity_entrada},
        blocking=True,
    )
    await hass.async_block_till_done()

    # Fake backend response for valve being turned on
    response = (
        b'{"user_id": "91763b24c43d3e344f424e8b", "user": {"id": "91763b24c43d3e344f424e8b",'
        b'"email": "john@doe.com"}, "home_id": "91763b24c43d3e344f424e8b", "room_id": "2833524037",'
        b'"home": {"id": "91763b24c43d3e344f424e8b","name": "MYHOME","country": "DE",'
        b'"rooms": [{"id": "2833524037","name": "Entrada","type": "lobby",'
        b'"therm_setpoint_mode": "home"}], "modules": [{"id": "12:34:56:00:01:ae",'
        b'"name": "Entrada", "type": "NRV"}]}, "mode": "home", "event_type": "cancel_set_point",'
        b'"push_type": "display_change"}'
    )
    await simulate_webhook(hass, webhook_id, response)

    assert hass.states.get(climate_entity_entrada).state == "auto"


@pytest.mark.parametrize(
    "batterylevel, module_type, expected",
    [
        (4101, NA_THERM, 100),
        (3601, NA_THERM, 80),
        (3450, NA_THERM, 65),
        (3301, NA_THERM, 50),
        (3001, NA_THERM, 20),
        (2799, NA_THERM, 0),
        (3201, NA_VALVE, 100),
        (2701, NA_VALVE, 80),
        (2550, NA_VALVE, 65),
        (2401, NA_VALVE, 50),
        (2201, NA_VALVE, 20),
        (2001, NA_VALVE, 0),
    ],
)
async def test_interpolate(batterylevel, module_type, expected):
    """Test interpolation of battery levels depending on device type."""
    assert climate.interpolate(batterylevel, module_type) == expected


async def test_get_all_home_ids():
    """Test extracting all home ids returned by NetAtmo API."""
    # Test with backend returning no data
    assert climate.get_all_home_ids(None) == []

    # Test with fake data
    home_data = Mock()
    home_data.homes = {
        "123": {"id": "123", "name": "Home 1", "modules": [], "therm_schedules": []},
        "987": {"id": "987", "name": "Home 2", "modules": [], "therm_schedules": []},
    }
    expected = ["123", "987"]
    assert climate.get_all_home_ids(home_data) == expected