b65d4a9efd
* bump whirlpool_sixth_sense to 18.0 add callback method initiated in 18.0 * add disconnect to mock
100 lines
3.4 KiB
Python
100 lines
3.4 KiB
Python
"""Fixtures for the Whirlpool Sixth Sense integration tests."""
|
|
from unittest import mock
|
|
from unittest.mock import AsyncMock
|
|
|
|
import pytest
|
|
import whirlpool
|
|
import whirlpool.aircon
|
|
from whirlpool.backendselector import Brand, Region
|
|
|
|
MOCK_SAID1 = "said1"
|
|
MOCK_SAID2 = "said2"
|
|
|
|
|
|
@pytest.fixture(
|
|
name="region",
|
|
params=[("EU", Region.EU, Brand.Whirlpool), ("US", Region.US, Brand.Maytag)],
|
|
)
|
|
def fixture_region(request):
|
|
"""Return a region for input."""
|
|
return request.param
|
|
|
|
|
|
@pytest.fixture(name="mock_auth_api")
|
|
def fixture_mock_auth_api():
|
|
"""Set up Auth fixture."""
|
|
with mock.patch("homeassistant.components.whirlpool.Auth") as mock_auth:
|
|
mock_auth.return_value.do_auth = AsyncMock()
|
|
mock_auth.return_value.is_access_token_valid.return_value = True
|
|
yield mock_auth
|
|
|
|
|
|
@pytest.fixture(name="mock_appliances_manager_api")
|
|
def fixture_mock_appliances_manager_api():
|
|
"""Set up AppliancesManager fixture."""
|
|
with mock.patch(
|
|
"homeassistant.components.whirlpool.AppliancesManager"
|
|
) as mock_appliances_manager:
|
|
mock_appliances_manager.return_value.fetch_appliances = AsyncMock()
|
|
mock_appliances_manager.return_value.aircons = [
|
|
{"SAID": MOCK_SAID1, "NAME": "TestZone"},
|
|
{"SAID": MOCK_SAID2, "NAME": "TestZone"},
|
|
]
|
|
yield mock_appliances_manager
|
|
|
|
|
|
@pytest.fixture(name="mock_backend_selector_api")
|
|
def fixture_mock_backend_selector_api():
|
|
"""Set up BackendSelector fixture."""
|
|
with mock.patch(
|
|
"homeassistant.components.whirlpool.BackendSelector"
|
|
) as mock_backend_selector:
|
|
yield mock_backend_selector
|
|
|
|
|
|
def get_aircon_mock(said):
|
|
"""Get a mock of an air conditioner."""
|
|
mock_aircon = mock.Mock(said=said)
|
|
mock_aircon.connect = AsyncMock()
|
|
mock_aircon.disconnect = AsyncMock()
|
|
mock_aircon.get_online.return_value = True
|
|
mock_aircon.get_power_on.return_value = True
|
|
mock_aircon.get_mode.return_value = whirlpool.aircon.Mode.Cool
|
|
mock_aircon.get_fanspeed.return_value = whirlpool.aircon.FanSpeed.Auto
|
|
mock_aircon.get_current_temp.return_value = 15
|
|
mock_aircon.get_temp.return_value = 20
|
|
mock_aircon.get_current_humidity.return_value = 80
|
|
mock_aircon.get_humidity.return_value = 50
|
|
mock_aircon.get_h_louver_swing.return_value = True
|
|
|
|
mock_aircon.set_power_on = AsyncMock()
|
|
mock_aircon.set_mode = AsyncMock()
|
|
mock_aircon.set_temp = AsyncMock()
|
|
mock_aircon.set_humidity = AsyncMock()
|
|
mock_aircon.set_mode = AsyncMock()
|
|
mock_aircon.set_fanspeed = AsyncMock()
|
|
mock_aircon.set_h_louver_swing = AsyncMock()
|
|
|
|
return mock_aircon
|
|
|
|
|
|
@pytest.fixture(name="mock_aircon1_api", autouse=True)
|
|
def fixture_mock_aircon1_api(mock_auth_api, mock_appliances_manager_api):
|
|
"""Set up air conditioner API fixture."""
|
|
yield get_aircon_mock(MOCK_SAID1)
|
|
|
|
|
|
@pytest.fixture(name="mock_aircon2_api", autouse=True)
|
|
def fixture_mock_aircon2_api(mock_auth_api, mock_appliances_manager_api):
|
|
"""Set up air conditioner API fixture."""
|
|
yield get_aircon_mock(MOCK_SAID2)
|
|
|
|
|
|
@pytest.fixture(name="mock_aircon_api_instances", autouse=True)
|
|
def fixture_mock_aircon_api_instances(mock_aircon1_api, mock_aircon2_api):
|
|
"""Set up air conditioner API fixture."""
|
|
with mock.patch(
|
|
"homeassistant.components.whirlpool.climate.Aircon"
|
|
) as mock_aircon_api:
|
|
mock_aircon_api.side_effect = [mock_aircon1_api, mock_aircon2_api]
|
|
yield mock_aircon_api
|