f563068ce6
* Create config_flow.py * Update config_flow.py * work on config flow * Update test_config_flow.py * Update __init__.py * Update camera.py * Update test_config_flow.py * Update test_config_flow.py * Update config_flow.py * Update conftest.py * Update test_config_flow.py * Update test_init.py * Update test_init.py * Apply suggestions from code review * Update camera.py * Update test_init.py * Update camera.py * Update __init__.py * Update test_init.py * Update test_init.py * Update __init__.py * Update __init__.py * Apply suggestions from code review * Update __init__.py * Update test_init.py * Update __init__.py * Update __init__.py * Update config_flow.py * Update test_config_flow.py * Update config_flow.py * Update config_flow.py
58 lines
2 KiB
Python
58 lines
2 KiB
Python
"""Define fixtures available for all tests."""
|
|
from canary.api import Api
|
|
from pytest import fixture
|
|
|
|
from tests.async_mock import MagicMock, patch
|
|
|
|
|
|
def mock_canary_update(self, **kwargs):
|
|
"""Get the latest data from py-canary."""
|
|
self._update(**kwargs)
|
|
|
|
|
|
@fixture
|
|
def canary(hass):
|
|
"""Mock the CanaryApi for easier testing."""
|
|
with patch.object(Api, "login", return_value=True), patch(
|
|
"homeassistant.components.canary.CanaryData.update", mock_canary_update
|
|
), patch("homeassistant.components.canary.Api") as mock_canary:
|
|
instance = mock_canary.return_value = Api(
|
|
"test-username",
|
|
"test-password",
|
|
1,
|
|
)
|
|
|
|
instance.login = MagicMock(return_value=True)
|
|
instance.get_entries = MagicMock(return_value=[])
|
|
instance.get_locations = MagicMock(return_value=[])
|
|
instance.get_location = MagicMock(return_value=None)
|
|
instance.get_modes = MagicMock(return_value=[])
|
|
instance.get_readings = MagicMock(return_value=[])
|
|
instance.get_latest_readings = MagicMock(return_value=[])
|
|
instance.set_location_mode = MagicMock(return_value=None)
|
|
|
|
yield mock_canary
|
|
|
|
|
|
@fixture
|
|
def canary_config_flow(hass):
|
|
"""Mock the CanaryApi for easier config flow testing."""
|
|
with patch.object(Api, "login", return_value=True), patch(
|
|
"homeassistant.components.canary.config_flow.Api"
|
|
) as mock_canary:
|
|
instance = mock_canary.return_value = Api(
|
|
"test-username",
|
|
"test-password",
|
|
1,
|
|
)
|
|
|
|
instance.login = MagicMock(return_value=True)
|
|
instance.get_entries = MagicMock(return_value=[])
|
|
instance.get_locations = MagicMock(return_value=[])
|
|
instance.get_location = MagicMock(return_value=None)
|
|
instance.get_modes = MagicMock(return_value=[])
|
|
instance.get_readings = MagicMock(return_value=[])
|
|
instance.get_latest_readings = MagicMock(return_value=[])
|
|
instance.set_location_mode = MagicMock(return_value=None)
|
|
|
|
yield mock_canary
|