1
0
mirror of https://github.com/home-assistant/supervisor.git synced 2025-12-23 03:48:57 +00:00
Files
supervisor/tests/utils/test_dbus.py
Pascal Vizeli 2809f23391 Make dbus-fast calls more robust (#4005)
* Make dbus-fast calls more robust

* Handle all exceptions and add test

* DBus minimal can't return commands list

Co-authored-by: Mike Degatano <michael.degatano@gmail.com>
2022-11-15 22:19:38 +01:00

43 lines
1.4 KiB
Python

"""Test dbus utility."""
from unittest.mock import AsyncMock, Mock
from dbus_fast.aio.message_bus import MessageBus
import pytest
from supervisor.dbus.const import DBUS_OBJECT_BASE
from supervisor.exceptions import DBusFatalError, DBusInterfaceError
from supervisor.utils.dbus import DBus
async def test_missing_properties_interface(dbus_bus: MessageBus, dbus: list[str]):
"""Test introspection missing properties interface."""
service = await DBus.connect(
dbus_bus, "test.no.properties.interface", DBUS_OBJECT_BASE
)
with pytest.raises(DBusInterfaceError):
await service.get_properties("test.no.properties.interface")
@pytest.mark.parametrize("err", [BrokenPipeError(), EOFError(), OSError()])
async def test_internal_dbus_errors(
dbus_minimal: MessageBus,
capture_exception: Mock,
err: Exception,
):
"""Test internal dbus library errors become dbus error."""
rauc = await DBus.connect(dbus_minimal, "de.pengutronix.rauc", DBUS_OBJECT_BASE)
setattr(
# pylint: disable=protected-access
rauc._proxies["de.pengutronix.rauc.Installer"],
# pylint: enable=protected-access
"call_mark",
proxy_mock := AsyncMock().call_mark,
)
proxy_mock.side_effect = err
with pytest.raises(DBusFatalError):
await rauc.Installer.call_mark("good", "booted")
capture_exception.assert_called_once_with(err)