mirror of
https://github.com/home-assistant/supervisor.git
synced 2025-04-23 04:37:15 +00:00
1d83c0c77a
* Add known first party to isort configuration * Run isort on supervisor and tests folder
49 lines
1.1 KiB
Python
49 lines
1.1 KiB
Python
"""Common test functions."""
|
|
from unittest.mock import MagicMock, PropertyMock, patch
|
|
|
|
import pytest
|
|
|
|
from supervisor.bootstrap import initialize_coresys
|
|
|
|
from tests.common import mock_coro
|
|
|
|
# pylint: disable=redefined-outer-name
|
|
|
|
|
|
@pytest.fixture
|
|
def docker():
|
|
"""Mock Docker API."""
|
|
with patch("supervisor.coresys.DockerAPI") as mock:
|
|
yield mock
|
|
|
|
|
|
@pytest.fixture
|
|
async def coresys(loop, docker):
|
|
"""Create a CoreSys Mock."""
|
|
with patch("supervisor.bootstrap.initialize_system_data"), patch(
|
|
"supervisor.bootstrap.fetch_timezone",
|
|
return_value=mock_coro(return_value="Europe/Zurich"),
|
|
):
|
|
coresys_obj = await initialize_coresys()
|
|
|
|
coresys_obj.ingress.save_data = MagicMock()
|
|
|
|
yield coresys_obj
|
|
|
|
|
|
@pytest.fixture
|
|
def sys_machine():
|
|
"""Mock sys_machine."""
|
|
with patch("supervisor.coresys.CoreSys.machine", new_callable=PropertyMock) as mock:
|
|
yield mock
|
|
|
|
|
|
@pytest.fixture
|
|
def sys_supervisor():
|
|
"""Mock sys_supervisor."""
|
|
with patch(
|
|
"supervisor.coresys.CoreSys.supervisor", new_callable=PropertyMock
|
|
) as mock:
|
|
mock.return_value = MagicMock()
|
|
yield MagicMock
|