mirror of
https://github.com/home-assistant/supervisor.git
synced 2025-11-09 02:49:43 +00:00
f6faa18409
It seems that the codebase is not formatted with the latest ruff version. This PR reformats the codebase with ruff 0.5.7.
36 lines
1.2 KiB
Python
36 lines
1.2 KiB
Python
"""Test CGroup/Agent dbus interface."""
|
|
|
|
# pylint: disable=import-error
|
|
from dbus_fast.aio.message_bus import MessageBus
|
|
import pytest
|
|
|
|
from supervisor.dbus.agent import OSAgent
|
|
from supervisor.exceptions import DBusNotConnectedError
|
|
|
|
from tests.dbus_service_mocks.agent_cgroup import CGroup as CGroupService
|
|
from tests.dbus_service_mocks.base import DBusServiceMock
|
|
|
|
|
|
@pytest.fixture(name="cgroup_service", autouse=True)
|
|
async def fixture_cgroup_service(
|
|
os_agent_services: dict[str, DBusServiceMock],
|
|
) -> CGroupService:
|
|
"""Mock CGroup dbus service."""
|
|
yield os_agent_services["agent_cgroup"]
|
|
|
|
|
|
async def test_dbus_osagent_cgroup_add_devices(
|
|
cgroup_service: CGroupService, dbus_session_bus: MessageBus
|
|
):
|
|
"""Test wipe data partition on host."""
|
|
cgroup_service.AddDevicesAllowed.calls.clear()
|
|
os_agent = OSAgent()
|
|
|
|
with pytest.raises(DBusNotConnectedError):
|
|
await os_agent.cgroup.add_devices_allowed("9324kl23j4kl", "*:* rwm")
|
|
|
|
await os_agent.connect(dbus_session_bus)
|
|
|
|
assert await os_agent.cgroup.add_devices_allowed("9324kl23j4kl", "*:* rwm") is None
|
|
assert cgroup_service.AddDevicesAllowed.calls == [("9324kl23j4kl", "*:* rwm")]
|