evlli/hass-core
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Add AirQuality SensorState support for Google Assistant (#80579)

Browse source 
* Add AirQuality SensorState support for Google Assistant

* Code formatting

* Apply suggestions from code review

Co-authored-by: Joakim Plate <elupus@ecce.se>

* Update trait.py

* Update trait.py

* Fix google_assistant tests

* Update trait.py

* Simplify sensor state payload and tests

* Add more tests to fix coverage

* fix tests

* Truncate once

---------

Co-authored-by: Joakim Plate <elupus@ecce.se>
Co-authored-by: Erik <erik@montnemery.com>
This commit is contained in:
Don Holly 2023-06-27 23:55:00 -07:00 • committed by GitHub
parent 72806bfaf2
commit 04e277ac95
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
3 changed files with 142 additions and 20 deletions
Download patch file Download diff file Expand all files Collapse all files

96
tests/components/google_assistant/test_trait.py
View file

@ -3291,6 +3291,50 @@ async def test_channel(hass: HomeAssistant) -> None:
assert len(media_player_calls) == 1
async def test_air_quality_description_for_aqi(hass: HomeAssistant) -> None:
"""Test air quality description for a given AQI value."""
trt = trait.SensorStateTrait(
hass,
State(
"sensor.test",
100.0,
{
"device_class": sensor.SensorDeviceClass.AQI,
},
),
BASIC_CONFIG,
)
assert trt._air_quality_description_for_aqi("0") == "healthy"
assert trt._air_quality_description_for_aqi("75") == "moderate"
assert (
trt._air_quality_description_for_aqi("125") == "unhealthy for sensitive groups"
)
assert trt._air_quality_description_for_aqi("175") == "unhealthy"
assert trt._air_quality_description_for_aqi("250") == "very unhealthy"
assert trt._air_quality_description_for_aqi("350") == "hazardous"
assert trt._air_quality_description_for_aqi("-1") == "unknown"
assert trt._air_quality_description_for_aqi("non-numeric") == "unknown"
async def test_null_device_class(hass: HomeAssistant) -> None:
"""Test handling a null device_class in sync_attributes and query_attributes."""
trt = trait.SensorStateTrait(
hass,
State(
"sensor.test",
100.0,
{
"device_class": None,
},
),
BASIC_CONFIG,
)
assert trt.sync_attributes() == {}
assert trt.query_attributes() == {}
async def test_sensorstate(hass: HomeAssistant) -> None:
"""Test SensorState trait support for sensor domain."""
sensor_types = {
@ -3324,16 +3368,52 @@ async def test_sensorstate(hass: HomeAssistant) -> None:
name = sensor_types[sensor_type][0]
unit = sensor_types[sensor_type][1]
assert trt.sync_attributes() == {
"sensorStatesSupported": {
"name": name,
"numericCapabilities": {"rawValueUnit": unit},
if sensor_type == sensor.SensorDeviceClass.AQI:
assert trt.sync_attributes() == {
"sensorStatesSupported": [
{
"name": name,
"numericCapabilities": {"rawValueUnit": unit},
"descriptiveCapabilities": {
"availableStates": [
"healthy",
"moderate",
"unhealthy for sensitive groups",
"unhealthy",
"very unhealthy",
"hazardous",
"unknown",
],
},
}
]
}
else:
assert trt.sync_attributes() == {
"sensorStatesSupported": [
{
"name": name,
"numericCapabilities": {"rawValueUnit": unit},
}
]
}
}
assert trt.query_attributes() == {
"currentSensorStateData": [{"name": name, "rawValue": "100.0"}]
}
if sensor_type == sensor.SensorDeviceClass.AQI:
assert trt.query_attributes() == {
"currentSensorStateData": [
{
"name": name,
"currentSensorState": trt._air_quality_description_for_aqi(
trt.state.state
),
"rawValue": trt.state.state,
},
]
}
else:
assert trt.query_attributes() == {
"currentSensorStateData": [{"name": name, "rawValue": trt.state.state}]
}
assert helpers.get_google_type(sensor.DOMAIN, None) is not None
assert (