7bbd28d385
* Migrate pyLoad integration to externa API library * Add const to .coveragerc * raise update failed when cookie expired * fix exceptions * Add tests * bump to PyLoadAPI 1.1.0 * remove unreachable code * fix tests * Improve logging and exception handling - Modify manifest.json to update logger configuration. - Improve error messages for authentication failures in sensor.py. - Simplify and rename pytest fixtures in conftest.py. - Update test cases in test_sensor.py to check for log entries and remove unnecessary code. * remove exception translations
85 lines
2.4 KiB
Python
85 lines
2.4 KiB
Python
"""Tests for the pyLoad Sensors."""
|
|
|
|
from unittest.mock import AsyncMock
|
|
|
|
from pyloadapi.exceptions import CannotConnect, InvalidAuth, ParserError
|
|
import pytest
|
|
from syrupy.assertion import SnapshotAssertion
|
|
|
|
from homeassistant.components.sensor import DOMAIN
|
|
from homeassistant.core import HomeAssistant
|
|
from homeassistant.helpers.typing import ConfigType
|
|
from homeassistant.setup import async_setup_component
|
|
|
|
|
|
@pytest.mark.usefixtures("mock_pyloadapi")
|
|
async def test_setup(
|
|
hass: HomeAssistant,
|
|
pyload_config: ConfigType,
|
|
snapshot: SnapshotAssertion,
|
|
) -> None:
|
|
"""Test setup of the pyload sensor platform."""
|
|
|
|
assert await async_setup_component(hass, DOMAIN, pyload_config)
|
|
await hass.async_block_till_done()
|
|
|
|
result = hass.states.get("sensor.pyload_speed")
|
|
assert result == snapshot
|
|
|
|
|
|
@pytest.mark.parametrize(
|
|
("exception", "expected_exception"),
|
|
[
|
|
(CannotConnect, "Unable to connect and retrieve data from pyLoad API"),
|
|
(ParserError, "Unable to parse data from pyLoad API"),
|
|
(
|
|
InvalidAuth,
|
|
"Authentication failed for username, check your login credentials",
|
|
),
|
|
],
|
|
)
|
|
async def test_setup_exceptions(
|
|
hass: HomeAssistant,
|
|
pyload_config: ConfigType,
|
|
mock_pyloadapi: AsyncMock,
|
|
exception: Exception,
|
|
expected_exception: str,
|
|
caplog: pytest.LogCaptureFixture,
|
|
) -> None:
|
|
"""Test exceptions during setup up pyLoad platform."""
|
|
|
|
mock_pyloadapi.login.side_effect = exception
|
|
|
|
assert await async_setup_component(hass, DOMAIN, pyload_config)
|
|
await hass.async_block_till_done()
|
|
|
|
assert len(hass.states.async_all(DOMAIN)) == 0
|
|
assert expected_exception in caplog.text
|
|
|
|
|
|
@pytest.mark.parametrize(
|
|
("exception", "expected_exception"),
|
|
[
|
|
(CannotConnect, "UpdateFailed"),
|
|
(ParserError, "UpdateFailed"),
|
|
(InvalidAuth, "UpdateFailed"),
|
|
],
|
|
)
|
|
async def test_sensor_update_exceptions(
|
|
hass: HomeAssistant,
|
|
pyload_config: ConfigType,
|
|
mock_pyloadapi: AsyncMock,
|
|
exception: Exception,
|
|
expected_exception: str,
|
|
caplog: pytest.LogCaptureFixture,
|
|
) -> None:
|
|
"""Test exceptions during update of pyLoad sensor."""
|
|
|
|
mock_pyloadapi.get_status.side_effect = exception
|
|
|
|
assert await async_setup_component(hass, DOMAIN, pyload_config)
|
|
await hass.async_block_till_done()
|
|
|
|
assert len(hass.states.async_all(DOMAIN)) == 0
|
|
assert expected_exception in caplog.text
|