jeans/core
jeans/core
1
0
Fork 0
mirror of https://github.com/home-assistant/core.git synced 2025-07-16 01:37:08 +00:00
Code Issues Packages Projects Releases Wiki Activity

Improve buienradar tests (#50101)

Browse Source
This commit is contained in:
Rob Bierbooms 2021-05-05 09:05:46 +02:00 committed by GitHub
parent 44383f25ce
commit 2b461073ff
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
3 changed files with 24 additions and 16 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
tests/components/buienradar/test_init.py
View File

@ -1,7 +1,7 @@
"""Tests for the buienradar component.""" """Tests for the buienradar component."""
from unittest.mock import patch from unittest.mock import patch
from homeassistant.components.buienradar import async_setup from homeassistant import setup
from homeassistant.components.buienradar.const import DOMAIN from homeassistant.components.buienradar.const import DOMAIN
from homeassistant.const import CONF_LATITUDE, CONF_LONGITUDE from homeassistant.const import CONF_LATITUDE, CONF_LONGITUDE
from homeassistant.helpers.entity_registry import async_get_registry from homeassistant.helpers.entity_registry import async_get_registry
@ -30,7 +30,7 @@ async def test_import_all(hass):
with patch( with patch(
"homeassistant.components.buienradar.async_setup_entry", return_value=True "homeassistant.components.buienradar.async_setup_entry", return_value=True
): ):
await async_setup(hass, config) await setup.async_setup_component(hass, DOMAIN, config)
await hass.async_block_till_done() await hass.async_block_till_done()
conf_entries = hass.config_entries.async_entries(DOMAIN) conf_entries = hass.config_entries.async_entries(DOMAIN)
@ -68,7 +68,7 @@ async def test_import_camera(hass):
with patch( with patch(
"homeassistant.components.buienradar.async_setup_entry", return_value=True "homeassistant.components.buienradar.async_setup_entry", return_value=True
): ):
await async_setup(hass, config) await setup.async_setup_component(hass, DOMAIN, config)
await hass.async_block_till_done() await hass.async_block_till_done()
conf_entries = hass.config_entries.async_entries(DOMAIN) conf_entries = hass.config_entries.async_entries(DOMAIN)
@ -97,7 +97,7 @@ async def test_import_camera(hass):
assert entity.original_name == "test_name" assert entity.original_name == "test_name"
async def test_load_unload(hass): async def test_load_unload(aioclient_mock, hass):
"""Test options flow.""" """Test options flow."""
entry = MockConfigEntry( entry = MockConfigEntry(
domain=DOMAIN, domain=DOMAIN,

30
tests/components/buienradar/test_sensor.py
View File

@ -1,29 +1,37 @@
"""The tests for the Buienradar sensor platform.""" """The tests for the Buienradar sensor platform."""
from unittest.mock import patch
from homeassistant.components.buienradar.const import DOMAIN from homeassistant.components.buienradar.const import DOMAIN
from homeassistant.const import CONF_LATITUDE, CONF_LONGITUDE from homeassistant.const import CONF_LATITUDE, CONF_LONGITUDE
from homeassistant.helpers.entity_registry import async_get
from tests.common import MockConfigEntry from tests.common import MockConfigEntry
TEST_LONGITUDE = 51.5288504
TEST_LATITUDE = 5.4002156
CONDITIONS = ["stationname", "temperature"] CONDITIONS = ["stationname", "temperature"]
TEST_CFG_DATA = {CONF_LATITUDE: 51.5288504, CONF_LONGITUDE: 5.4002156} TEST_CFG_DATA = {CONF_LATITUDE: TEST_LATITUDE, CONF_LONGITUDE: TEST_LONGITUDE}
async def test_smoke_test_setup_component(hass): async def test_smoke_test_setup_component(aioclient_mock, hass):
"""Smoke test for successfully set-up with default config.""" """Smoke test for successfully set-up with default config."""
mock_entry = MockConfigEntry(domain=DOMAIN, unique_id="TEST_ID", data=TEST_CFG_DATA) mock_entry = MockConfigEntry(domain=DOMAIN, unique_id="TEST_ID", data=TEST_CFG_DATA)
mock_entry.add_to_hass(hass) mock_entry.add_to_hass(hass)
with patch( entity_registry = async_get(hass)
"homeassistant.components.buienradar.sensor.BrSensor.entity_registry_enabled_default" for cond in CONDITIONS:
) as enabled_by_default_mock: entity_registry.async_get_or_create(
enabled_by_default_mock.return_value = True domain="sensor",
platform="buienradar",
unique_id=f"{TEST_LATITUDE:2.6f}{TEST_LONGITUDE:2.6f}{cond}",
config_entry=mock_entry,
original_name=f"Buienradar {cond}",
)
await hass.async_block_till_done()
await hass.config_entries.async_setup(mock_entry.entry_id) await hass.config_entries.async_setup(mock_entry.entry_id)
await hass.async_block_till_done() await hass.async_block_till_done()
for cond in CONDITIONS: for cond in CONDITIONS:
state = hass.states.get(f"sensor.buienradar_{cond}") state = hass.states.get(f"sensor.buienradar_5_40021651_528850{cond}")
assert state.state == "unknown" assert state.state == "unknown"

2
tests/components/buienradar/test_weather.py
View File

@ -7,7 +7,7 @@ from tests.common import MockConfigEntry
TEST_CFG_DATA = {CONF_LATITUDE: 51.5288504, CONF_LONGITUDE: 5.4002156} TEST_CFG_DATA = {CONF_LATITUDE: 51.5288504, CONF_LONGITUDE: 5.4002156}
async def test_smoke_test_setup_component(hass): async def test_smoke_test_setup_component(aioclient_mock, hass):
"""Smoke test for successfully set-up with default config.""" """Smoke test for successfully set-up with default config."""
mock_entry = MockConfigEntry(domain=DOMAIN, unique_id="TEST_ID", data=TEST_CFG_DATA) mock_entry = MockConfigEntry(domain=DOMAIN, unique_id="TEST_ID", data=TEST_CFG_DATA)