irm-kmi-ha/tests/test_repairs.py

import json
import logging
from unittest.mock import MagicMock, AsyncMock

from homeassistant.core import HomeAssistant
from homeassistant.data_entry_flow import FlowResultType
from homeassistant.helpers import issue_registry
from pytest_homeassistant_custom_component.common import MockConfigEntry, load_fixture

from custom_components.irm_kmi import DOMAIN, IrmKmiCoordinator
from custom_components.irm_kmi.const import (REPAIR_OPT_DELETE,
                                             REPAIR_OPT_MOVE, REPAIR_SOLUTION)
from custom_components.irm_kmi.repairs import (OutOfBeneluxRepairFlow,
                                               async_create_fix_flow)

_LOGGER = logging.getLogger(__name__)


async def get_repair_flow(
        hass: HomeAssistant,
        mock_config_entry: MockConfigEntry
) -> OutOfBeneluxRepairFlow:
    hass.states.async_set(
        "zone.home",
        0,
        {"latitude": 50.738681639, "longitude": 4.054077148},
    )
    mock_config_entry.add_to_hass(hass)
    await hass.config_entries.async_setup(mock_config_entry.entry_id)
    await hass.async_block_till_done()
    coordinator = IrmKmiCoordinator(hass, mock_config_entry)

    fixture: str = "forecast_out_of_benelux.json"
    forecast = json.loads(load_fixture(fixture))
    coordinator._api.get_forecasts_coord = AsyncMock(return_value=forecast)

    await coordinator._async_update_data()
    ir = issue_registry.async_get(hass)
    issue = ir.async_get_issue(DOMAIN, "zone_moved")
    repair_flow = await async_create_fix_flow(hass, issue.issue_id, issue.data)
    repair_flow.hass = hass
    return repair_flow


async def test_repair_triggers_when_out_of_benelux(
        hass: HomeAssistant,
        mock_config_entry: MockConfigEntry
) -> None:
    hass.states.async_set(
        "zone.home",
        0,
        {"latitude": 50.738681639, "longitude": 4.054077148},
    )

    mock_config_entry.add_to_hass(hass)

    coordinator = IrmKmiCoordinator(hass, mock_config_entry)
    coordinator._api.get_forecasts_coord = AsyncMock(return_value=json.loads(load_fixture("forecast_out_of_benelux.json")))

    await coordinator._async_update_data()

    ir = issue_registry.async_get(hass)

    issue = ir.async_get_issue(DOMAIN, "zone_moved")

    assert issue is not None
    assert issue.data == {'config_entry_id': mock_config_entry.entry_id, 'zone': "zone.home"}
    assert issue.translation_key == "zone_moved"
    assert issue.is_fixable
    assert issue.translation_placeholders == {'zone': "zone.home"}


async def test_repair_flow(
        hass: HomeAssistant,
        mock_irm_kmi_api_repair_in_benelux: MagicMock,
        mock_config_entry: MockConfigEntry
) -> None:
    repair_flow = await get_repair_flow(hass, mock_config_entry)
    result = await repair_flow.async_step_init()

    assert result['type'] == FlowResultType.FORM
    assert result['errors'] == {}
    assert result['description_placeholders'] == {"zone": "zone.home"}

    user_input = {REPAIR_SOLUTION: REPAIR_OPT_MOVE}

    result = await repair_flow.async_step_confirm(user_input)

    assert result['type'] == FlowResultType.CREATE_ENTRY
    assert result['title'] == ""
    assert result['data'] == {}


async def test_repair_flow_invalid_choice(
        hass: HomeAssistant,
        mock_irm_kmi_api_repair_in_benelux: MagicMock,
        mock_config_entry: MockConfigEntry
) -> None:
    repair_flow = await get_repair_flow(hass, mock_config_entry)
    result = await repair_flow.async_step_init()

    assert result['type'] == FlowResultType.FORM
    user_input = {REPAIR_SOLUTION: "whut?"}

    result = await repair_flow.async_step_confirm(user_input)

    assert result['type'] == FlowResultType.FORM
    assert REPAIR_SOLUTION in result['errors']
    assert result['errors'][REPAIR_SOLUTION] == 'invalid_choice'


async def test_repair_flow_api_error(
        hass: HomeAssistant,
        mock_get_forecast_api_error_repair: MagicMock,
        mock_config_entry: MockConfigEntry
) -> None:
    repair_flow = await get_repair_flow(hass, mock_config_entry)
    result = await repair_flow.async_step_init()

    assert result['type'] == FlowResultType.FORM
    user_input = {REPAIR_SOLUTION: REPAIR_OPT_MOVE}

    result = await repair_flow.async_step_confirm(user_input)

    assert result['type'] == FlowResultType.FORM
    assert REPAIR_SOLUTION in result['errors']
    assert result['errors'][REPAIR_SOLUTION] == 'api_error'


async def test_repair_flow_out_of_benelux(
        hass: HomeAssistant,
        mock_irm_kmi_api_repair_out_of_benelux: MagicMock,
        mock_config_entry: MockConfigEntry
) -> None:
    repair_flow = await get_repair_flow(hass, mock_config_entry)
    result = await repair_flow.async_step_init()

    assert result['type'] == FlowResultType.FORM
    user_input = {REPAIR_SOLUTION: REPAIR_OPT_MOVE}

    result = await repair_flow.async_step_confirm(user_input)

    assert result['type'] == FlowResultType.FORM
    assert REPAIR_SOLUTION in result['errors']
    assert result['errors'][REPAIR_SOLUTION] == 'out_of_benelux'


async def test_repair_flow_delete_entry(
        hass: HomeAssistant,
        mock_config_entry: MockConfigEntry
) -> None:
    repair_flow = await get_repair_flow(hass, mock_config_entry)
    result = await repair_flow.async_step_init()

    assert result['type'] == FlowResultType.FORM
    assert len(hass.config_entries.async_entries(DOMAIN)) == 1
    assert hass.config_entries.async_entries(DOMAIN)[0].entry_id == mock_config_entry.entry_id

    user_input = {REPAIR_SOLUTION: REPAIR_OPT_DELETE}
    result = await repair_flow.async_step_confirm(user_input)

    assert result['type'] == FlowResultType.CREATE_ENTRY
    assert result['title'] == ""
    assert result['data'] == {}
    assert len(hass.config_entries.async_entries(DOMAIN)) == 0