jeans/core
jeans/core
1
0
Fork 0
mirror of https://github.com/home-assistant/core.git synced 2025-07-19 19:27:45 +00:00
Code Issues Packages Projects Releases Wiki Activity

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

Browse Source
This commit is contained in:
Franck Nijhof 2024-12-18 23:46:18 +01:00 committed by GitHub
parent 03707e6308
commit 9f3c549f8d
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
2 changed files with 73 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
tests/components/peblar/conftest.py
View File

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

69
tests/components/peblar/test_init.py Normal file
View File

@ -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