Zeversolar integration (#84887)

Co-authored-by: Franck Nijhof <frenck@frenck.nl>

tests/components/zeversolar/__init__.py

@@ -0,0 +1 @@
+"""Tests for the Zeversolar integration."""

tests/components/zeversolar/test_config_flow.py

@@ -0,0 +1,135 @@
+"""Test the Zeversolar config flow."""
+from unittest.mock import MagicMock, patch
+
+import pytest
+from zeversolar.exceptions import (
+    ZeverSolarHTTPError,
+    ZeverSolarHTTPNotFound,
+    ZeverSolarTimeout,
+)
+
+from homeassistant import config_entries
+from homeassistant.components.zeversolar.const import DOMAIN
+from homeassistant.const import CONF_HOST
+from homeassistant.core import HomeAssistant
+from homeassistant.data_entry_flow import FlowResultType
+
+from tests.common import MockConfigEntry
+
+
+async def test_form(hass: HomeAssistant) -> None:
+    """Test we get the form."""
+    result = await hass.config_entries.flow.async_init(
+        DOMAIN, context={"source": config_entries.SOURCE_USER}
+    )
+    assert result["type"] == FlowResultType.FORM
+    assert result["errors"] is None
+
+    await _set_up_zeversolar(hass=hass, flow_id=result["flow_id"])
+
+
+@pytest.mark.parametrize(
+    "side_effect,errors",
+    (
+        (
+            ZeverSolarHTTPNotFound,
+            {"base": "invalid_host"},
+        ),
+        (
+            ZeverSolarHTTPError,
+            {"base": "cannot_connect"},
+        ),
+        (
+            ZeverSolarTimeout,
+            {"base": "timeout_connect"},
+        ),
+        (
+            RuntimeError,
+            {"base": "unknown"},
+        ),
+    ),
+)
+async def test_form_errors(
+    hass: HomeAssistant,
+    side_effect: Exception,
+    errors: dict,
+) -> None:
+    """Test we handle errors."""
+    result = await hass.config_entries.flow.async_init(
+        DOMAIN, context={"source": config_entries.SOURCE_USER}
+    )
+
+    with patch(
+        "zeversolar.ZeverSolarClient.get_data",
+        side_effect=side_effect,
+    ):
+        result2 = await hass.config_entries.flow.async_configure(
+            flow_id=result["flow_id"],
+            user_input={
+                CONF_HOST: "test_ip",
+            },
+        )
+
+    assert result2["type"] == FlowResultType.FORM
+    assert result2["errors"] == errors
+
+    await _set_up_zeversolar(hass=hass, flow_id=result["flow_id"])
+
+
+async def test_abort_already_configured(hass: HomeAssistant) -> None:
+    """Test we abort when the device is already configured."""
+    entry = MockConfigEntry(
+        domain=DOMAIN,
+        title="Zeversolar",
+        data={CONF_HOST: "test_ip"},
+        unique_id="test_serial",
+    )
+    entry.add_to_hass(hass)
+
+    result = await hass.config_entries.flow.async_init(
+        DOMAIN, context={"source": config_entries.SOURCE_USER}
+    )
+    assert result.get("type") == FlowResultType.FORM
+    assert result.get("errors") is None
+    assert "flow_id" in result
+
+    mock_data = MagicMock()
+    mock_data.serial_number = "test_serial"
+    with patch("zeversolar.ZeverSolarClient.get_data", return_value=mock_data), patch(
+        "homeassistant.components.zeversolar.async_setup_entry",
+    ) as mock_setup_entry:
+        result2 = await hass.config_entries.flow.async_configure(
+            flow_id=result["flow_id"],
+            user_input={
+                CONF_HOST: "test_ip",
+            },
+        )
+        await hass.async_block_till_done()
+
+    assert result2.get("type") == FlowResultType.ABORT
+    assert result2.get("reason") == "already_configured"
+    assert len(mock_setup_entry.mock_calls) == 0
+
+
+async def _set_up_zeversolar(hass: HomeAssistant, flow_id: str) -> None:
+    """Reusable successful setup of Zeversolar sensor."""
+    mock_data = MagicMock()
+    mock_data.serial_number = "test_serial"
+    with patch("zeversolar.ZeverSolarClient.get_data", return_value=mock_data), patch(
+        "homeassistant.components.zeversolar.async_setup_entry",
+        return_value=True,
+    ) as mock_setup_entry:
+        result2 = await hass.config_entries.flow.async_configure(
+            flow_id=flow_id,
+            user_input={
+                CONF_HOST: "test_ip",
+            },
+        )
+        await hass.async_block_till_done()
+
+    assert result2["type"] == FlowResultType.CREATE_ENTRY
+    assert result2["title"] == "Zeversolar"
+    assert result2["data"] == {
+        CONF_HOST: "test_ip",
+    }
+    assert len(mock_setup_entry.mock_calls) == 1