hass-core/tests/components/airly/__init__.py
Maciej Bieniek 35d44ec90a
Store Airly runtime data in config entry (#117031)
* Store runtime data in config entry

* Fix tests

---------

Co-authored-by: Maciej Bieniek <478555+bieniu@users.noreply.github.com>
2024-05-07 22:04:37 +02:00

39 lines
1.1 KiB
Python

"""Tests for Airly."""
from homeassistant.components.airly.const import DOMAIN
from tests.common import MockConfigEntry, load_fixture
API_NEAREST_URL = "https://airapi.airly.eu/v2/measurements/nearest?lat=123.000000&lng=456.000000&maxDistanceKM=5.000000"
API_POINT_URL = (
"https://airapi.airly.eu/v2/measurements/point?lat=123.000000&lng=456.000000"
)
HEADERS = {
"X-RateLimit-Limit-day": "100",
"X-RateLimit-Remaining-day": "42",
}
async def init_integration(hass, aioclient_mock) -> MockConfigEntry:
"""Set up the Airly integration in Home Assistant."""
entry = MockConfigEntry(
domain=DOMAIN,
title="Home",
entry_id="3bd2acb0e4f0476d40865546d0d91921",
unique_id="123-456",
data={
"api_key": "foo",
"latitude": 123,
"longitude": 456,
"name": "Home",
},
)
aioclient_mock.get(
API_POINT_URL, text=load_fixture("valid_station.json", DOMAIN), headers=HEADERS
)
entry.add_to_hass(hass)
await hass.config_entries.async_setup(entry.entry_id)
await hass.async_block_till_done()
return entry