Home-Assistant/core
1
0
Fork 0
mirror of https://github.com/home-assistant/core.git synced 2025-12-24 21:06:19 +00:00
Code Activity

Black

Browse Source
This commit is contained in:
Paulus Schoutsen
2019-07-31 12:25:30 -07:00
parent da05dfe708
commit 4de97abc3a
2676 changed files with 163166 additions and 140084 deletions
Download Patch File Download Diff File Expand all files Collapse all files

133
tests/components/melissa/test_climate.py
View File

@@ -3,8 +3,14 @@ import json
from unittest.mock import Mock, patch
from homeassistant.components.climate.const import (
HVAC_MODE_COOL, HVAC_MODE_DRY, HVAC_MODE_FAN_ONLY, HVAC_MODE_HEAT,
HVAC_MODE_OFF, SUPPORT_FAN_MODE, SUPPORT_TARGET_TEMPERATURE)
HVAC_MODE_COOL,
HVAC_MODE_DRY,
HVAC_MODE_FAN_ONLY,
HVAC_MODE_HEAT,
HVAC_MODE_OFF,
SUPPORT_FAN_MODE,
SUPPORT_TARGET_TEMPERATURE,
)
from homeassistant.components.fan import SPEED_HIGH, SPEED_LOW, SPEED_MEDIUM
from homeassistant.components.melissa import DATA_MELISSA, climate as melissa
from homeassistant.components.melissa.climate import MelissaClimate
@@ -19,11 +25,14 @@ def melissa_mock():
"""Use this to mock the melissa api."""
api = Mock()
api.async_fetch_devices = mock_coro_func(
return_value=json.loads(load_fixture('melissa_fetch_devices.json')))
api.async_status = mock_coro_func(return_value=json.loads(load_fixture(
'melissa_status.json')))
return_value=json.loads(load_fixture("melissa_fetch_devices.json"))
)
api.async_status = mock_coro_func(
return_value=json.loads(load_fixture("melissa_status.json"))
)
api.async_cur_settings = mock_coro_func(
return_value=json.loads(load_fixture('melissa_cur_settings.json')))
return_value=json.loads(load_fixture("melissa_cur_settings.json"))
)
api.async_send = mock_coro_func(return_value=True)
@@ -42,21 +51,21 @@ def melissa_mock():
api.FAN_MEDIUM = 2
api.FAN_HIGH = 3
api.STATE = 'state'
api.MODE = 'mode'
api.FAN = 'fan'
api.TEMP = 'temp'
api.STATE = "state"
api.MODE = "mode"
api.FAN = "fan"
api.TEMP = "temp"
return api
async def test_setup_platform(hass):
"""Test setup_platform."""
with patch("homeassistant.components.melissa.climate.MelissaClimate"
) as mocked_thermostat:
with patch(
"homeassistant.components.melissa.climate.MelissaClimate"
) as mocked_thermostat:
api = melissa_mock()
device = (await api.async_fetch_devices())[_SERIAL]
thermostat = mocked_thermostat(api, device['serial_number'],
device)
thermostat = mocked_thermostat(api, device["serial_number"], device)
thermostats = [thermostat]
hass.data[DATA_MELISSA] = api
@@ -65,14 +74,13 @@ async def test_setup_platform(hass):
add_entities = Mock()
discovery_info = {}
await melissa.async_setup_platform(
hass, config, add_entities, discovery_info)
await melissa.async_setup_platform(hass, config, add_entities, discovery_info)
add_entities.assert_called_once_with(thermostats)
async def test_get_name(hass):
"""Test name property."""
with patch('homeassistant.components.melissa'):
with patch("homeassistant.components.melissa"):
api = melissa_mock()
device = (await api.async_fetch_devices())[_SERIAL]
thermostat = MelissaClimate(api, _SERIAL, device)
@@ -81,7 +89,7 @@ async def test_get_name(hass):
async def test_current_fan_mode(hass):
"""Test current_fan_mode property."""
with patch('homeassistant.components.melissa'):
with patch("homeassistant.components.melissa"):
api = melissa_mock()
device = (await api.async_fetch_devices())[_SERIAL]
thermostat = MelissaClimate(api, _SERIAL, device)
@@ -94,7 +102,7 @@ async def test_current_fan_mode(hass):
async def test_current_temperature(hass):
"""Test current temperature."""
with patch('homeassistant.components.melissa'):
with patch("homeassistant.components.melissa"):
api = melissa_mock()
device = (await api.async_fetch_devices())[_SERIAL]
thermostat = MelissaClimate(api, _SERIAL, device)
@@ -103,7 +111,7 @@ async def test_current_temperature(hass):
async def test_current_temperature_no_data(hass):
"""Test current temperature without data."""
with patch('homeassistant.components.melissa'):
with patch("homeassistant.components.melissa"):
api = melissa_mock()
device = (await api.async_fetch_devices())[_SERIAL]
thermostat = MelissaClimate(api, _SERIAL, device)
@@ -113,7 +121,7 @@ async def test_current_temperature_no_data(hass):
async def test_target_temperature_step(hass):
"""Test current target_temperature_step."""
with patch('homeassistant.components.melissa'):
with patch("homeassistant.components.melissa"):
api = melissa_mock()
device = (await api.async_fetch_devices())[_SERIAL]
thermostat = MelissaClimate(api, _SERIAL, device)
@@ -122,7 +130,7 @@ async def test_target_temperature_step(hass):
async def test_current_operation(hass):
"""Test current operation."""
with patch('homeassistant.components.melissa'):
with patch("homeassistant.components.melissa"):
api = melissa_mock()
device = (await api.async_fetch_devices())[_SERIAL]
thermostat = MelissaClimate(api, _SERIAL, device)
@@ -135,27 +143,31 @@ async def test_current_operation(hass):
async def test_operation_list(hass):
"""Test the operation list."""
with patch('homeassistant.components.melissa'):
with patch("homeassistant.components.melissa"):
api = melissa_mock()
device = (await api.async_fetch_devices())[_SERIAL]
thermostat = MelissaClimate(api, _SERIAL, device)
assert [HVAC_MODE_HEAT, HVAC_MODE_COOL, HVAC_MODE_DRY,
HVAC_MODE_FAN_ONLY, HVAC_MODE_OFF] == thermostat.hvac_modes
assert [
HVAC_MODE_HEAT,
HVAC_MODE_COOL,
HVAC_MODE_DRY,
HVAC_MODE_FAN_ONLY,
HVAC_MODE_OFF,
] == thermostat.hvac_modes
async def test_fan_modes(hass):
"""Test the fan list."""
with patch('homeassistant.components.melissa'):
with patch("homeassistant.components.melissa"):
api = melissa_mock()
device = (await api.async_fetch_devices())[_SERIAL]
thermostat = MelissaClimate(api, _SERIAL, device)
assert ['auto', SPEED_HIGH, SPEED_MEDIUM, SPEED_LOW] == \
thermostat.fan_modes
assert ["auto", SPEED_HIGH, SPEED_MEDIUM, SPEED_LOW] == thermostat.fan_modes
async def test_target_temperature(hass):
"""Test target temperature."""
with patch('homeassistant.components.melissa'):
with patch("homeassistant.components.melissa"):
api = melissa_mock()
device = (await api.async_fetch_devices())[_SERIAL]
thermostat = MelissaClimate(api, _SERIAL, device)
@@ -168,7 +180,7 @@ async def test_target_temperature(hass):
async def test_state(hass):
"""Test state."""
with patch('homeassistant.components.melissa'):
with patch("homeassistant.components.melissa"):
api = melissa_mock()
device = (await api.async_fetch_devices())[_SERIAL]
thermostat = MelissaClimate(api, _SERIAL, device)
@@ -181,7 +193,7 @@ async def test_state(hass):
async def test_temperature_unit(hass):
"""Test temperature unit."""
with patch('homeassistant.components.melissa'):
with patch("homeassistant.components.melissa"):
api = melissa_mock()
device = (await api.async_fetch_devices())[_SERIAL]
thermostat = MelissaClimate(api, _SERIAL, device)
@@ -190,7 +202,7 @@ async def test_temperature_unit(hass):
async def test_min_temp(hass):
"""Test min temp."""
with patch('homeassistant.components.melissa'):
with patch("homeassistant.components.melissa"):
api = melissa_mock()
device = (await api.async_fetch_devices())[_SERIAL]
thermostat = MelissaClimate(api, _SERIAL, device)
@@ -199,7 +211,7 @@ async def test_min_temp(hass):
async def test_max_temp(hass):
"""Test max temp."""
with patch('homeassistant.components.melissa'):
with patch("homeassistant.components.melissa"):
api = melissa_mock()
device = (await api.async_fetch_devices())[_SERIAL]
thermostat = MelissaClimate(api, _SERIAL, device)
@@ -208,17 +220,17 @@ async def test_max_temp(hass):
async def test_supported_features(hass):
"""Test supported_features property."""
with patch('homeassistant.components.melissa'):
with patch("homeassistant.components.melissa"):
api = melissa_mock()
device = (await api.async_fetch_devices())[_SERIAL]
thermostat = MelissaClimate(api, _SERIAL, device)
features = (SUPPORT_TARGET_TEMPERATURE | SUPPORT_FAN_MODE)
features = SUPPORT_TARGET_TEMPERATURE | SUPPORT_FAN_MODE
assert features == thermostat.supported_features
async def test_set_temperature(hass):
"""Test set_temperature."""
with patch('homeassistant.components.melissa'):
with patch("homeassistant.components.melissa"):
api = melissa_mock()
device = (await api.async_fetch_devices())[_SERIAL]
thermostat = MelissaClimate(api, _SERIAL, device)
@@ -229,7 +241,7 @@ async def test_set_temperature(hass):
async def test_fan_mode(hass):
"""Test set_fan_mode."""
with patch('homeassistant.components.melissa'):
with patch("homeassistant.components.melissa"):
api = melissa_mock()
device = (await api.async_fetch_devices())[_SERIAL]
thermostat = MelissaClimate(api, _SERIAL, device)
@@ -242,7 +254,7 @@ async def test_fan_mode(hass):
async def test_set_operation_mode(hass):
"""Test set_operation_mode."""
with patch('homeassistant.components.melissa'):
with patch("homeassistant.components.melissa"):
api = melissa_mock()
device = (await api.async_fetch_devices())[_SERIAL]
thermostat = MelissaClimate(api, _SERIAL, device)
@@ -255,18 +267,18 @@ async def test_set_operation_mode(hass):
async def test_send(hass):
"""Test send."""
with patch('homeassistant.components.melissa'):
with patch("homeassistant.components.melissa"):
api = melissa_mock()
device = (await api.async_fetch_devices())[_SERIAL]
thermostat = MelissaClimate(api, _SERIAL, device)
await thermostat.async_update()
await hass.async_block_till_done()
await thermostat.async_send({'fan': api.FAN_MEDIUM})
await thermostat.async_send({"fan": api.FAN_MEDIUM})
await hass.async_block_till_done()
assert SPEED_MEDIUM == thermostat.fan_mode
api.async_send.return_value = mock_coro_func(return_value=False)
thermostat._cur_settings = None
await thermostat.async_send({'fan': api.FAN_LOW})
await thermostat.async_send({"fan": api.FAN_LOW})
await hass.async_block_till_done()
assert SPEED_LOW != thermostat.fan_mode
assert thermostat._cur_settings is None
@@ -274,24 +286,26 @@ async def test_send(hass):
async def test_update(hass):
"""Test update."""
with patch('homeassistant.components.melissa.climate._LOGGER.warning'
) as mocked_warning:
with patch('homeassistant.components.melissa'):
with patch(
"homeassistant.components.melissa.climate._LOGGER.warning"
) as mocked_warning:
with patch("homeassistant.components.melissa"):
api = melissa_mock()
device = (await api.async_fetch_devices())[_SERIAL]
thermostat = MelissaClimate(api, _SERIAL, device)
await thermostat.async_update()
assert SPEED_LOW == thermostat.fan_mode
assert HVAC_MODE_HEAT == thermostat.state
api.async_status = mock_coro_func(exception=KeyError('boom'))
api.async_status = mock_coro_func(exception=KeyError("boom"))
await thermostat.async_update()
mocked_warning.assert_called_once_with(
'Unable to update entity %s', thermostat.entity_id)
"Unable to update entity %s", thermostat.entity_id
)
async def test_melissa_op_to_hass(hass):
"""Test for translate melissa operations to hass."""
with patch('homeassistant.components.melissa'):
with patch("homeassistant.components.melissa"):
api = melissa_mock()
device = (await api.async_fetch_devices())[_SERIAL]
thermostat = MelissaClimate(api, _SERIAL, device)
@@ -304,11 +318,11 @@ async def test_melissa_op_to_hass(hass):
async def test_melissa_fan_to_hass(hass):
"""Test for translate melissa fan state to hass."""
with patch('homeassistant.components.melissa'):
with patch("homeassistant.components.melissa"):
api = melissa_mock()
device = (await api.async_fetch_devices())[_SERIAL]
thermostat = MelissaClimate(api, _SERIAL, device)
assert 'auto' == thermostat.melissa_fan_to_hass(0)
assert "auto" == thermostat.melissa_fan_to_hass(0)
assert SPEED_LOW == thermostat.melissa_fan_to_hass(1)
assert SPEED_MEDIUM == thermostat.melissa_fan_to_hass(2)
assert SPEED_HIGH == thermostat.melissa_fan_to_hass(3)
@@ -317,9 +331,10 @@ async def test_melissa_fan_to_hass(hass):
async def test_hass_mode_to_melissa(hass):
"""Test for hass operations to melssa."""
with patch('homeassistant.components.melissa.climate._LOGGER.warning'
) as mocked_warning:
with patch('homeassistant.components.melissa'):
with patch(
"homeassistant.components.melissa.climate._LOGGER.warning"
) as mocked_warning:
with patch("homeassistant.components.melissa"):
api = melissa_mock()
device = (await api.async_fetch_devices())[_SERIAL]
thermostat = MelissaClimate(api, _SERIAL, device)
@@ -329,22 +344,24 @@ async def test_hass_mode_to_melissa(hass):
assert 4 == thermostat.hass_mode_to_melissa(HVAC_MODE_DRY)
thermostat.hass_mode_to_melissa("test")
mocked_warning.assert_called_once_with(
"Melissa have no setting for %s mode", "test")
"Melissa have no setting for %s mode", "test"
)
async def test_hass_fan_to_melissa(hass):
"""Test for translate melissa states to hass."""
with patch(
'homeassistant.components.melissa.climate._LOGGER.warning'
) as mocked_warning:
with patch('homeassistant.components.melissa'):
"homeassistant.components.melissa.climate._LOGGER.warning"
) as mocked_warning:
with patch("homeassistant.components.melissa"):
api = melissa_mock()
device = (await api.async_fetch_devices())[_SERIAL]
thermostat = MelissaClimate(api, _SERIAL, device)
assert 0 == thermostat.hass_fan_to_melissa('auto')
assert 0 == thermostat.hass_fan_to_melissa("auto")
assert 1 == thermostat.hass_fan_to_melissa(SPEED_LOW)
assert 2 == thermostat.hass_fan_to_melissa(SPEED_MEDIUM)
assert 3 == thermostat.hass_fan_to_melissa(SPEED_HIGH)
thermostat.hass_fan_to_melissa("test")
mocked_warning.assert_called_once_with(
"Melissa have no setting for %s fan mode", "test")
"Melissa have no setting for %s fan mode", "test"
)