Cleanup HVACAction and HVACMode in tests (#78813)

This commit is contained in:
epenet 2022-09-20 20:30:54 +02:00 committed by GitHub
parent 3cf26c4a5d
commit f453726b18
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
11 changed files with 497 additions and 515 deletions

View file

@ -3,7 +3,13 @@ import pytest
import voluptuous_serialize
import homeassistant.components.automation as automation
from homeassistant.components.climate import DOMAIN, const, device_trigger
from homeassistant.components.climate import (
DOMAIN,
HVACAction,
HVACMode,
const,
device_trigger,
)
from homeassistant.components.device_automation import DeviceAutomationType
from homeassistant.const import TEMP_CELSIUS
from homeassistant.helpers import config_validation as cv, device_registry
@ -52,9 +58,9 @@ async def test_get_triggers(hass, device_reg, entity_reg):
entity_id = f"{DOMAIN}.test_5678"
hass.states.async_set(
entity_id,
const.HVAC_MODE_COOL,
HVACMode.COOL,
{
const.ATTR_HVAC_ACTION: const.HVACAction.IDLE,
const.ATTR_HVAC_ACTION: HVACAction.IDLE,
const.ATTR_CURRENT_HUMIDITY: 23,
const.ATTR_CURRENT_TEMPERATURE: 18,
},
@ -114,9 +120,9 @@ async def test_get_triggers_hidden_auxiliary(
entity_id = f"{DOMAIN}.test_5678"
hass.states.async_set(
entity_id,
const.HVAC_MODE_COOL,
HVACMode.COOL,
{
const.ATTR_HVAC_ACTION: const.CURRENT_HVAC_IDLE,
const.ATTR_HVAC_ACTION: HVACAction.IDLE,
const.ATTR_CURRENT_HUMIDITY: 23,
const.ATTR_CURRENT_TEMPERATURE: 18,
},
@ -146,9 +152,9 @@ async def test_if_fires_on_state_change(hass, calls):
"""Test for turn_on and turn_off triggers firing."""
hass.states.async_set(
"climate.entity",
const.HVAC_MODE_COOL,
HVACMode.COOL,
{
const.ATTR_HVAC_ACTION: const.HVACAction.IDLE,
const.ATTR_HVAC_ACTION: HVACAction.IDLE,
const.ATTR_CURRENT_HUMIDITY: 23,
const.ATTR_CURRENT_TEMPERATURE: 18,
},
@ -166,7 +172,7 @@ async def test_if_fires_on_state_change(hass, calls):
"device_id": "",
"entity_id": "climate.entity",
"type": "hvac_mode_changed",
"to": const.HVAC_MODE_AUTO,
"to": HVACMode.AUTO,
},
"action": {
"service": "test.automation",
@ -208,9 +214,9 @@ async def test_if_fires_on_state_change(hass, calls):
# Fake that the HVAC mode is changing
hass.states.async_set(
"climate.entity",
const.HVAC_MODE_AUTO,
HVACMode.AUTO,
{
const.ATTR_HVAC_ACTION: const.HVACAction.COOLING,
const.ATTR_HVAC_ACTION: HVACAction.COOLING,
const.ATTR_CURRENT_HUMIDITY: 23,
const.ATTR_CURRENT_TEMPERATURE: 18,
},
@ -222,9 +228,9 @@ async def test_if_fires_on_state_change(hass, calls):
# Fake that the temperature is changing
hass.states.async_set(
"climate.entity",
const.HVAC_MODE_AUTO,
HVACMode.AUTO,
{
const.ATTR_HVAC_ACTION: const.HVACAction.COOLING,
const.ATTR_HVAC_ACTION: HVACAction.COOLING,
const.ATTR_CURRENT_HUMIDITY: 23,
const.ATTR_CURRENT_TEMPERATURE: 23,
},
@ -236,9 +242,9 @@ async def test_if_fires_on_state_change(hass, calls):
# Fake that the humidity is changing
hass.states.async_set(
"climate.entity",
const.HVAC_MODE_AUTO,
HVACMode.AUTO,
{
const.ATTR_HVAC_ACTION: const.HVACAction.COOLING,
const.ATTR_HVAC_ACTION: HVACAction.COOLING,
const.ATTR_CURRENT_HUMIDITY: 7,
const.ATTR_CURRENT_TEMPERATURE: 23,
},