Add integration setup tests to Peblar Rocksolid EV Chargers (#133532)

tests/components/peblar/conftest.py

@@ -38,9 +38,10 @@ def mock_setup_entry() -> Generator[None]:
 @pytest.fixture
 def mock_peblar() -> Generator[MagicMock]:
     """Return a mocked Peblar client."""
-    with patch(
-        "homeassistant.components.peblar.config_flow.Peblar", autospec=True
-    ) as peblar_mock:
+    with (
+        patch("homeassistant.components.peblar.Peblar", autospec=True) as peblar_mock,
+        patch("homeassistant.components.peblar.config_flow.Peblar", new=peblar_mock),
+    ):
         peblar = peblar_mock.return_value
         peblar.system_information.return_value = PeblarSystemInformation.from_json(
             load_fixture("system_information.json", DOMAIN)

tests/components/peblar/test_init.py

@@ -0,0 +1,69 @@
+"""Integration tests for the Peblar integration."""
+
+from unittest.mock import MagicMock
+
+from peblar import PeblarAuthenticationError, PeblarConnectionError, PeblarError
+import pytest
+
+from homeassistant.components.peblar.const import DOMAIN
+from homeassistant.config_entries import ConfigEntryState
+from homeassistant.core import HomeAssistant
+
+from tests.common import MockConfigEntry
+
+
+async def test_load_unload_config_entry(
+    hass: HomeAssistant,
+    mock_config_entry: MockConfigEntry,
+    mock_peblar: MagicMock,
+) -> None:
+    """Test the Peblar configuration entry loading/unloading."""
+    mock_config_entry.add_to_hass(hass)
+    await hass.config_entries.async_setup(mock_config_entry.entry_id)
+    await hass.async_block_till_done()
+
+    assert mock_config_entry.state is ConfigEntryState.LOADED
+    assert len(mock_peblar.login.mock_calls) == 1
+
+    await hass.config_entries.async_unload(mock_config_entry.entry_id)
+    await hass.async_block_till_done()
+
+    assert not hass.data.get(DOMAIN)
+    assert mock_config_entry.state is ConfigEntryState.NOT_LOADED
+
+
+@pytest.mark.parametrize(
+    "exception",
+    [PeblarConnectionError, PeblarError],
+)
+async def test_config_entry_not_ready(
+    hass: HomeAssistant,
+    mock_config_entry: MockConfigEntry,
+    mock_peblar: MagicMock,
+    exception: Exception,
+) -> None:
+    """Test the Peblar configuration entry not ready."""
+    mock_peblar.login.side_effect = exception
+
+    mock_config_entry.add_to_hass(hass)
+    await hass.config_entries.async_setup(mock_config_entry.entry_id)
+    await hass.async_block_till_done()
+
+    assert len(mock_peblar.login.mock_calls) == 1
+    assert mock_config_entry.state is ConfigEntryState.SETUP_RETRY
+
+
+async def test_config_entry_authentication_failed(
+    hass: HomeAssistant,
+    mock_config_entry: MockConfigEntry,
+    mock_peblar: MagicMock,
+) -> None:
+    """Test authentication error, aborts setup."""
+    mock_config_entry.add_to_hass(hass)
+
+    mock_peblar.login.side_effect = PeblarAuthenticationError
+
+    await hass.config_entries.async_setup(mock_config_entry.entry_id)
+    await hass.async_block_till_done()
+
+    assert mock_config_entry.state is ConfigEntryState.SETUP_ERROR