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

Add select entities to AirGradient (#117136)

Browse source
This commit is contained in:
Joost Lekkerkerker 2024-05-29 20:12:51 +02:00 • committed by GitHub
parent 6382cb9134
commit c80718628e
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
12 changed files with 549 additions and 33 deletions
Download patch file Download diff file Expand all files Collapse all files

10
tests/components/airgradient/test_sensor.py
View file

@ -1,7 +1,7 @@
"""Tests for the AirGradient sensor platform."""
from datetime import timedelta
from unittest.mock import AsyncMock
from unittest.mock import AsyncMock, patch
from airgradient import AirGradientError, Measures
from freezegun.api import FrozenDateTimeFactory
@ -9,7 +9,7 @@ import pytest
from syrupy import SnapshotAssertion
from homeassistant.components.airgradient import DOMAIN
from homeassistant.const import STATE_UNAVAILABLE
from homeassistant.const import STATE_UNAVAILABLE, Platform
from homeassistant.core import HomeAssistant
from homeassistant.helpers import entity_registry as er
@ -32,7 +32,8 @@ async def test_all_entities(
entity_registry: er.EntityRegistry,
) -> None:
"""Test all entities."""
await setup_integration(hass, mock_config_entry)
with patch("homeassistant.components.airgradient.PLATFORMS", [Platform.SENSOR]):
await setup_integration(hass, mock_config_entry)
await snapshot_platform(hass, entity_registry, snapshot, mock_config_entry.entry_id)
@ -47,7 +48,8 @@ async def test_create_entities(
mock_airgradient_client.get_current_measures.return_value = Measures.from_json(
load_fixture("measures_after_boot.json", DOMAIN)
)
await setup_integration(hass, mock_config_entry)
with patch("homeassistant.components.airgradient.PLATFORMS", [Platform.SENSOR]):
await setup_integration(hass, mock_config_entry)
assert len(hass.states.async_all()) == 0
mock_airgradient_client.get_current_measures.return_value = Measures.from_json(