supervisor/tests/conftest.py

"""Common test functions."""
from unittest.mock import patch, PropertyMock, MagicMock

import pytest

from hassio.bootstrap import initialize_coresys

# pylint: disable=redefined-outer-name


@pytest.fixture
def docker():
    """Mock Docker API."""
    with patch('hassio.coresys.DockerAPI') as mock:
        yield mock


@pytest.fixture
async def coresys(loop, docker):
    """Create a CoreSys Mock."""
    with patch('hassio.bootstrap.initialize_system_data'):
        coresys_obj = await initialize_coresys()

    yield coresys_obj


@pytest.fixture
def sys_machine():
    """Mock sys_machine."""
    with patch(
            'hassio.coresys.CoreSys.machine',
            new_callable=PropertyMock) as mock:
        yield mock


@pytest.fixture
def sys_supervisor():
    with patch(
            'hassio.coresys.CoreSys.supervisor',
            new_callable=PropertyMock) as mock:
        mock.return_value = MagicMock()
        yield MagicMock