mirror of
https://github.com/home-assistant/core.git
synced 2025-07-16 17:57:11 +00:00
Use async_load_fixture in async test functions (a) (#145718)
This commit is contained in:
parent
c3ade400fb
commit
bd5fef1ddb
@ -4,7 +4,7 @@ from homeassistant.components.agent_dvr.const import DOMAIN, SERVER_URL
|
|||||||
from homeassistant.const import CONF_HOST, CONF_PORT, CONTENT_TYPE_JSON
|
from homeassistant.const import CONF_HOST, CONF_PORT, CONTENT_TYPE_JSON
|
||||||
from homeassistant.core import HomeAssistant
|
from homeassistant.core import HomeAssistant
|
||||||
|
|
||||||
from tests.common import MockConfigEntry, load_fixture
|
from tests.common import MockConfigEntry, async_load_fixture
|
||||||
from tests.test_util.aiohttp import AiohttpClientMocker
|
from tests.test_util.aiohttp import AiohttpClientMocker
|
||||||
|
|
||||||
CONF_DATA = {
|
CONF_DATA = {
|
||||||
@ -34,12 +34,12 @@ async def init_integration(
|
|||||||
|
|
||||||
aioclient_mock.get(
|
aioclient_mock.get(
|
||||||
"http://example.local:8090/command.cgi?cmd=getStatus",
|
"http://example.local:8090/command.cgi?cmd=getStatus",
|
||||||
text=load_fixture("agent_dvr/status.json"),
|
text=await async_load_fixture(hass, "status.json", DOMAIN),
|
||||||
headers={"Content-Type": CONTENT_TYPE_JSON},
|
headers={"Content-Type": CONTENT_TYPE_JSON},
|
||||||
)
|
)
|
||||||
aioclient_mock.get(
|
aioclient_mock.get(
|
||||||
"http://example.local:8090/command.cgi?cmd=getObjects",
|
"http://example.local:8090/command.cgi?cmd=getObjects",
|
||||||
text=load_fixture("agent_dvr/objects.json"),
|
text=await async_load_fixture(hass, "objects.json", DOMAIN),
|
||||||
headers={"Content-Type": CONTENT_TYPE_JSON},
|
headers={"Content-Type": CONTENT_TYPE_JSON},
|
||||||
)
|
)
|
||||||
entry = create_entry(hass)
|
entry = create_entry(hass)
|
||||||
|
@ -2,8 +2,7 @@
|
|||||||
|
|
||||||
import pytest
|
import pytest
|
||||||
|
|
||||||
from homeassistant.components.agent_dvr import config_flow
|
from homeassistant.components.agent_dvr.const import DOMAIN, SERVER_URL
|
||||||
from homeassistant.components.agent_dvr.const import SERVER_URL
|
|
||||||
from homeassistant.config_entries import SOURCE_USER
|
from homeassistant.config_entries import SOURCE_USER
|
||||||
from homeassistant.const import CONF_HOST, CONF_PORT, CONTENT_TYPE_JSON
|
from homeassistant.const import CONF_HOST, CONF_PORT, CONTENT_TYPE_JSON
|
||||||
from homeassistant.core import HomeAssistant
|
from homeassistant.core import HomeAssistant
|
||||||
@ -11,7 +10,7 @@ from homeassistant.data_entry_flow import FlowResultType
|
|||||||
|
|
||||||
from . import init_integration
|
from . import init_integration
|
||||||
|
|
||||||
from tests.common import load_fixture
|
from tests.common import async_load_fixture
|
||||||
from tests.test_util.aiohttp import AiohttpClientMocker
|
from tests.test_util.aiohttp import AiohttpClientMocker
|
||||||
|
|
||||||
pytestmark = pytest.mark.usefixtures("mock_setup_entry")
|
pytestmark = pytest.mark.usefixtures("mock_setup_entry")
|
||||||
@ -20,7 +19,7 @@ pytestmark = pytest.mark.usefixtures("mock_setup_entry")
|
|||||||
async def test_show_user_form(hass: HomeAssistant) -> None:
|
async def test_show_user_form(hass: HomeAssistant) -> None:
|
||||||
"""Test that the user set up form is served."""
|
"""Test that the user set up form is served."""
|
||||||
result = await hass.config_entries.flow.async_init(
|
result = await hass.config_entries.flow.async_init(
|
||||||
config_flow.DOMAIN,
|
DOMAIN,
|
||||||
context={"source": SOURCE_USER},
|
context={"source": SOURCE_USER},
|
||||||
)
|
)
|
||||||
|
|
||||||
@ -35,7 +34,7 @@ async def test_user_device_exists_abort(
|
|||||||
await init_integration(hass, aioclient_mock)
|
await init_integration(hass, aioclient_mock)
|
||||||
|
|
||||||
result = await hass.config_entries.flow.async_init(
|
result = await hass.config_entries.flow.async_init(
|
||||||
config_flow.DOMAIN,
|
DOMAIN,
|
||||||
context={"source": SOURCE_USER},
|
context={"source": SOURCE_USER},
|
||||||
data={CONF_HOST: "example.local", CONF_PORT: 8090},
|
data={CONF_HOST: "example.local", CONF_PORT: 8090},
|
||||||
)
|
)
|
||||||
@ -51,7 +50,7 @@ async def test_connection_error(
|
|||||||
aioclient_mock.get("http://example.local:8090/command.cgi?cmd=getStatus", text="")
|
aioclient_mock.get("http://example.local:8090/command.cgi?cmd=getStatus", text="")
|
||||||
|
|
||||||
result = await hass.config_entries.flow.async_init(
|
result = await hass.config_entries.flow.async_init(
|
||||||
config_flow.DOMAIN,
|
DOMAIN,
|
||||||
context={"source": SOURCE_USER},
|
context={"source": SOURCE_USER},
|
||||||
data={CONF_HOST: "example.local", CONF_PORT: 8090},
|
data={CONF_HOST: "example.local", CONF_PORT: 8090},
|
||||||
)
|
)
|
||||||
@ -67,18 +66,18 @@ async def test_full_user_flow_implementation(
|
|||||||
"""Test the full manual user flow from start to finish."""
|
"""Test the full manual user flow from start to finish."""
|
||||||
aioclient_mock.get(
|
aioclient_mock.get(
|
||||||
"http://example.local:8090/command.cgi?cmd=getStatus",
|
"http://example.local:8090/command.cgi?cmd=getStatus",
|
||||||
text=load_fixture("agent_dvr/status.json"),
|
text=await async_load_fixture(hass, "status.json", DOMAIN),
|
||||||
headers={"Content-Type": CONTENT_TYPE_JSON},
|
headers={"Content-Type": CONTENT_TYPE_JSON},
|
||||||
)
|
)
|
||||||
|
|
||||||
aioclient_mock.get(
|
aioclient_mock.get(
|
||||||
"http://example.local:8090/command.cgi?cmd=getObjects",
|
"http://example.local:8090/command.cgi?cmd=getObjects",
|
||||||
text=load_fixture("agent_dvr/objects.json"),
|
text=await async_load_fixture(hass, "objects.json", DOMAIN),
|
||||||
headers={"Content-Type": CONTENT_TYPE_JSON},
|
headers={"Content-Type": CONTENT_TYPE_JSON},
|
||||||
)
|
)
|
||||||
|
|
||||||
result = await hass.config_entries.flow.async_init(
|
result = await hass.config_entries.flow.async_init(
|
||||||
config_flow.DOMAIN,
|
DOMAIN,
|
||||||
context={"source": SOURCE_USER},
|
context={"source": SOURCE_USER},
|
||||||
)
|
)
|
||||||
|
|
||||||
@ -95,5 +94,5 @@ async def test_full_user_flow_implementation(
|
|||||||
assert result["title"] == "DESKTOP"
|
assert result["title"] == "DESKTOP"
|
||||||
assert result["type"] is FlowResultType.CREATE_ENTRY
|
assert result["type"] is FlowResultType.CREATE_ENTRY
|
||||||
|
|
||||||
entries = hass.config_entries.async_entries(config_flow.DOMAIN)
|
entries = hass.config_entries.async_entries(DOMAIN)
|
||||||
assert entries[0].unique_id == "c0715bba-c2d0-48ef-9e3e-bc81c9ea4447"
|
assert entries[0].unique_id == "c0715bba-c2d0-48ef-9e3e-bc81c9ea4447"
|
||||||
|
@ -20,7 +20,7 @@ from . import setup_integration
|
|||||||
from tests.common import (
|
from tests.common import (
|
||||||
MockConfigEntry,
|
MockConfigEntry,
|
||||||
async_fire_time_changed,
|
async_fire_time_changed,
|
||||||
load_fixture,
|
async_load_fixture,
|
||||||
snapshot_platform,
|
snapshot_platform,
|
||||||
)
|
)
|
||||||
|
|
||||||
@ -81,7 +81,7 @@ async def test_cloud_creates_no_button(
|
|||||||
assert len(hass.states.async_all()) == 0
|
assert len(hass.states.async_all()) == 0
|
||||||
|
|
||||||
mock_cloud_airgradient_client.get_config.return_value = Config.from_json(
|
mock_cloud_airgradient_client.get_config.return_value = Config.from_json(
|
||||||
load_fixture("get_config_local.json", DOMAIN)
|
await async_load_fixture(hass, "get_config_local.json", DOMAIN)
|
||||||
)
|
)
|
||||||
|
|
||||||
freezer.tick(timedelta(minutes=5))
|
freezer.tick(timedelta(minutes=5))
|
||||||
@ -91,7 +91,7 @@ async def test_cloud_creates_no_button(
|
|||||||
assert len(hass.states.async_all()) == 2
|
assert len(hass.states.async_all()) == 2
|
||||||
|
|
||||||
mock_cloud_airgradient_client.get_config.return_value = Config.from_json(
|
mock_cloud_airgradient_client.get_config.return_value = Config.from_json(
|
||||||
load_fixture("get_config_cloud.json", DOMAIN)
|
await async_load_fixture(hass, "get_config_cloud.json", DOMAIN)
|
||||||
)
|
)
|
||||||
|
|
||||||
freezer.tick(timedelta(minutes=5))
|
freezer.tick(timedelta(minutes=5))
|
||||||
|
@ -24,7 +24,7 @@ from . import setup_integration
|
|||||||
from tests.common import (
|
from tests.common import (
|
||||||
MockConfigEntry,
|
MockConfigEntry,
|
||||||
async_fire_time_changed,
|
async_fire_time_changed,
|
||||||
load_fixture,
|
async_load_fixture,
|
||||||
snapshot_platform,
|
snapshot_platform,
|
||||||
)
|
)
|
||||||
|
|
||||||
@ -83,7 +83,7 @@ async def test_cloud_creates_no_number(
|
|||||||
assert len(hass.states.async_all()) == 0
|
assert len(hass.states.async_all()) == 0
|
||||||
|
|
||||||
mock_cloud_airgradient_client.get_config.return_value = Config.from_json(
|
mock_cloud_airgradient_client.get_config.return_value = Config.from_json(
|
||||||
load_fixture("get_config_local.json", DOMAIN)
|
await async_load_fixture(hass, "get_config_local.json", DOMAIN)
|
||||||
)
|
)
|
||||||
|
|
||||||
freezer.tick(timedelta(minutes=5))
|
freezer.tick(timedelta(minutes=5))
|
||||||
@ -93,7 +93,7 @@ async def test_cloud_creates_no_number(
|
|||||||
assert len(hass.states.async_all()) == 2
|
assert len(hass.states.async_all()) == 2
|
||||||
|
|
||||||
mock_cloud_airgradient_client.get_config.return_value = Config.from_json(
|
mock_cloud_airgradient_client.get_config.return_value = Config.from_json(
|
||||||
load_fixture("get_config_cloud.json", DOMAIN)
|
await async_load_fixture(hass, "get_config_cloud.json", DOMAIN)
|
||||||
)
|
)
|
||||||
|
|
||||||
freezer.tick(timedelta(minutes=5))
|
freezer.tick(timedelta(minutes=5))
|
||||||
|
@ -23,7 +23,7 @@ from . import setup_integration
|
|||||||
from tests.common import (
|
from tests.common import (
|
||||||
MockConfigEntry,
|
MockConfigEntry,
|
||||||
async_fire_time_changed,
|
async_fire_time_changed,
|
||||||
load_fixture,
|
async_load_fixture,
|
||||||
snapshot_platform,
|
snapshot_platform,
|
||||||
)
|
)
|
||||||
|
|
||||||
@ -77,7 +77,7 @@ async def test_cloud_creates_no_number(
|
|||||||
assert len(hass.states.async_all()) == 1
|
assert len(hass.states.async_all()) == 1
|
||||||
|
|
||||||
mock_cloud_airgradient_client.get_config.return_value = Config.from_json(
|
mock_cloud_airgradient_client.get_config.return_value = Config.from_json(
|
||||||
load_fixture("get_config_local.json", DOMAIN)
|
await async_load_fixture(hass, "get_config_local.json", DOMAIN)
|
||||||
)
|
)
|
||||||
|
|
||||||
freezer.tick(timedelta(minutes=5))
|
freezer.tick(timedelta(minutes=5))
|
||||||
@ -87,7 +87,7 @@ async def test_cloud_creates_no_number(
|
|||||||
assert len(hass.states.async_all()) == 7
|
assert len(hass.states.async_all()) == 7
|
||||||
|
|
||||||
mock_cloud_airgradient_client.get_config.return_value = Config.from_json(
|
mock_cloud_airgradient_client.get_config.return_value = Config.from_json(
|
||||||
load_fixture("get_config_cloud.json", DOMAIN)
|
await async_load_fixture(hass, "get_config_cloud.json", DOMAIN)
|
||||||
)
|
)
|
||||||
|
|
||||||
freezer.tick(timedelta(minutes=5))
|
freezer.tick(timedelta(minutes=5))
|
||||||
|
@ -18,7 +18,7 @@ from . import setup_integration
|
|||||||
from tests.common import (
|
from tests.common import (
|
||||||
MockConfigEntry,
|
MockConfigEntry,
|
||||||
async_fire_time_changed,
|
async_fire_time_changed,
|
||||||
load_fixture,
|
async_load_fixture,
|
||||||
snapshot_platform,
|
snapshot_platform,
|
||||||
)
|
)
|
||||||
|
|
||||||
@ -46,14 +46,14 @@ async def test_create_entities(
|
|||||||
) -> None:
|
) -> None:
|
||||||
"""Test creating entities."""
|
"""Test creating entities."""
|
||||||
mock_airgradient_client.get_current_measures.return_value = Measures.from_json(
|
mock_airgradient_client.get_current_measures.return_value = Measures.from_json(
|
||||||
load_fixture("measures_after_boot.json", DOMAIN)
|
await async_load_fixture(hass, "measures_after_boot.json", DOMAIN)
|
||||||
)
|
)
|
||||||
with patch("homeassistant.components.airgradient.PLATFORMS", [Platform.SENSOR]):
|
with patch("homeassistant.components.airgradient.PLATFORMS", [Platform.SENSOR]):
|
||||||
await setup_integration(hass, mock_config_entry)
|
await setup_integration(hass, mock_config_entry)
|
||||||
|
|
||||||
assert len(hass.states.async_all()) == 0
|
assert len(hass.states.async_all()) == 0
|
||||||
mock_airgradient_client.get_current_measures.return_value = Measures.from_json(
|
mock_airgradient_client.get_current_measures.return_value = Measures.from_json(
|
||||||
load_fixture("current_measures_indoor.json", DOMAIN)
|
await async_load_fixture(hass, "current_measures_indoor.json", DOMAIN)
|
||||||
)
|
)
|
||||||
freezer.tick(timedelta(minutes=1))
|
freezer.tick(timedelta(minutes=1))
|
||||||
async_fire_time_changed(hass)
|
async_fire_time_changed(hass)
|
||||||
|
@ -25,7 +25,7 @@ from . import setup_integration
|
|||||||
from tests.common import (
|
from tests.common import (
|
||||||
MockConfigEntry,
|
MockConfigEntry,
|
||||||
async_fire_time_changed,
|
async_fire_time_changed,
|
||||||
load_fixture,
|
async_load_fixture,
|
||||||
snapshot_platform,
|
snapshot_platform,
|
||||||
)
|
)
|
||||||
|
|
||||||
@ -83,7 +83,7 @@ async def test_cloud_creates_no_switch(
|
|||||||
assert len(hass.states.async_all()) == 0
|
assert len(hass.states.async_all()) == 0
|
||||||
|
|
||||||
mock_cloud_airgradient_client.get_config.return_value = Config.from_json(
|
mock_cloud_airgradient_client.get_config.return_value = Config.from_json(
|
||||||
load_fixture("get_config_local.json", DOMAIN)
|
await async_load_fixture(hass, "get_config_local.json", DOMAIN)
|
||||||
)
|
)
|
||||||
|
|
||||||
freezer.tick(timedelta(minutes=5))
|
freezer.tick(timedelta(minutes=5))
|
||||||
@ -93,7 +93,7 @@ async def test_cloud_creates_no_switch(
|
|||||||
assert len(hass.states.async_all()) == 1
|
assert len(hass.states.async_all()) == 1
|
||||||
|
|
||||||
mock_cloud_airgradient_client.get_config.return_value = Config.from_json(
|
mock_cloud_airgradient_client.get_config.return_value = Config.from_json(
|
||||||
load_fixture("get_config_cloud.json", DOMAIN)
|
await async_load_fixture(hass, "get_config_cloud.json", DOMAIN)
|
||||||
)
|
)
|
||||||
|
|
||||||
freezer.tick(timedelta(minutes=5))
|
freezer.tick(timedelta(minutes=5))
|
||||||
|
@ -3,7 +3,7 @@
|
|||||||
from homeassistant.components.airly.const import DOMAIN
|
from homeassistant.components.airly.const import DOMAIN
|
||||||
from homeassistant.core import HomeAssistant
|
from homeassistant.core import HomeAssistant
|
||||||
|
|
||||||
from tests.common import MockConfigEntry, load_fixture
|
from tests.common import MockConfigEntry, async_load_fixture
|
||||||
from tests.test_util.aiohttp import AiohttpClientMocker
|
from tests.test_util.aiohttp import AiohttpClientMocker
|
||||||
|
|
||||||
API_NEAREST_URL = "https://airapi.airly.eu/v2/measurements/nearest?lat=123.000000&lng=456.000000&maxDistanceKM=5.000000"
|
API_NEAREST_URL = "https://airapi.airly.eu/v2/measurements/nearest?lat=123.000000&lng=456.000000&maxDistanceKM=5.000000"
|
||||||
@ -34,7 +34,9 @@ async def init_integration(
|
|||||||
)
|
)
|
||||||
|
|
||||||
aioclient_mock.get(
|
aioclient_mock.get(
|
||||||
API_POINT_URL, text=load_fixture("valid_station.json", DOMAIN), headers=HEADERS
|
API_POINT_URL,
|
||||||
|
text=await async_load_fixture(hass, "valid_station.json", DOMAIN),
|
||||||
|
headers=HEADERS,
|
||||||
)
|
)
|
||||||
entry.add_to_hass(hass)
|
entry.add_to_hass(hass)
|
||||||
await hass.config_entries.async_setup(entry.entry_id)
|
await hass.config_entries.async_setup(entry.entry_id)
|
||||||
|
@ -12,7 +12,7 @@ from homeassistant.data_entry_flow import FlowResultType
|
|||||||
|
|
||||||
from . import API_NEAREST_URL, API_POINT_URL
|
from . import API_NEAREST_URL, API_POINT_URL
|
||||||
|
|
||||||
from tests.common import MockConfigEntry, load_fixture, patch
|
from tests.common import MockConfigEntry, async_load_fixture, patch
|
||||||
from tests.test_util.aiohttp import AiohttpClientMocker
|
from tests.test_util.aiohttp import AiohttpClientMocker
|
||||||
|
|
||||||
CONFIG = {
|
CONFIG = {
|
||||||
@ -55,7 +55,9 @@ async def test_invalid_location(
|
|||||||
hass: HomeAssistant, aioclient_mock: AiohttpClientMocker
|
hass: HomeAssistant, aioclient_mock: AiohttpClientMocker
|
||||||
) -> None:
|
) -> None:
|
||||||
"""Test that errors are shown when location is invalid."""
|
"""Test that errors are shown when location is invalid."""
|
||||||
aioclient_mock.get(API_POINT_URL, text=load_fixture("no_station.json", "airly"))
|
aioclient_mock.get(
|
||||||
|
API_POINT_URL, text=await async_load_fixture(hass, "no_station.json", DOMAIN)
|
||||||
|
)
|
||||||
|
|
||||||
aioclient_mock.get(
|
aioclient_mock.get(
|
||||||
API_NEAREST_URL,
|
API_NEAREST_URL,
|
||||||
@ -74,9 +76,13 @@ async def test_invalid_location_for_point_and_nearest(
|
|||||||
) -> None:
|
) -> None:
|
||||||
"""Test an abort when the location is wrong for the point and nearest methods."""
|
"""Test an abort when the location is wrong for the point and nearest methods."""
|
||||||
|
|
||||||
aioclient_mock.get(API_POINT_URL, text=load_fixture("no_station.json", "airly"))
|
aioclient_mock.get(
|
||||||
|
API_POINT_URL, text=await async_load_fixture(hass, "no_station.json", DOMAIN)
|
||||||
|
)
|
||||||
|
|
||||||
aioclient_mock.get(API_NEAREST_URL, text=load_fixture("no_station.json", "airly"))
|
aioclient_mock.get(
|
||||||
|
API_NEAREST_URL, text=await async_load_fixture(hass, "no_station.json", DOMAIN)
|
||||||
|
)
|
||||||
|
|
||||||
with patch("homeassistant.components.airly.async_setup_entry", return_value=True):
|
with patch("homeassistant.components.airly.async_setup_entry", return_value=True):
|
||||||
result = await hass.config_entries.flow.async_init(
|
result = await hass.config_entries.flow.async_init(
|
||||||
@ -91,7 +97,9 @@ async def test_duplicate_error(
|
|||||||
hass: HomeAssistant, aioclient_mock: AiohttpClientMocker
|
hass: HomeAssistant, aioclient_mock: AiohttpClientMocker
|
||||||
) -> None:
|
) -> None:
|
||||||
"""Test that errors are shown when duplicates are added."""
|
"""Test that errors are shown when duplicates are added."""
|
||||||
aioclient_mock.get(API_POINT_URL, text=load_fixture("valid_station.json", "airly"))
|
aioclient_mock.get(
|
||||||
|
API_POINT_URL, text=await async_load_fixture(hass, "valid_station.json", DOMAIN)
|
||||||
|
)
|
||||||
MockConfigEntry(domain=DOMAIN, unique_id="123-456", data=CONFIG).add_to_hass(hass)
|
MockConfigEntry(domain=DOMAIN, unique_id="123-456", data=CONFIG).add_to_hass(hass)
|
||||||
|
|
||||||
result = await hass.config_entries.flow.async_init(
|
result = await hass.config_entries.flow.async_init(
|
||||||
@ -106,7 +114,9 @@ async def test_create_entry(
|
|||||||
hass: HomeAssistant, aioclient_mock: AiohttpClientMocker
|
hass: HomeAssistant, aioclient_mock: AiohttpClientMocker
|
||||||
) -> None:
|
) -> None:
|
||||||
"""Test that the user step works."""
|
"""Test that the user step works."""
|
||||||
aioclient_mock.get(API_POINT_URL, text=load_fixture("valid_station.json", "airly"))
|
aioclient_mock.get(
|
||||||
|
API_POINT_URL, text=await async_load_fixture(hass, "valid_station.json", DOMAIN)
|
||||||
|
)
|
||||||
|
|
||||||
with patch("homeassistant.components.airly.async_setup_entry", return_value=True):
|
with patch("homeassistant.components.airly.async_setup_entry", return_value=True):
|
||||||
result = await hass.config_entries.flow.async_init(
|
result = await hass.config_entries.flow.async_init(
|
||||||
@ -126,10 +136,13 @@ async def test_create_entry_with_nearest_method(
|
|||||||
) -> None:
|
) -> None:
|
||||||
"""Test that the user step works with nearest method."""
|
"""Test that the user step works with nearest method."""
|
||||||
|
|
||||||
aioclient_mock.get(API_POINT_URL, text=load_fixture("no_station.json", "airly"))
|
aioclient_mock.get(
|
||||||
|
API_POINT_URL, text=await async_load_fixture(hass, "no_station.json", DOMAIN)
|
||||||
|
)
|
||||||
|
|
||||||
aioclient_mock.get(
|
aioclient_mock.get(
|
||||||
API_NEAREST_URL, text=load_fixture("valid_station.json", "airly")
|
API_NEAREST_URL,
|
||||||
|
text=await async_load_fixture(hass, "valid_station.json", DOMAIN),
|
||||||
)
|
)
|
||||||
|
|
||||||
with patch("homeassistant.components.airly.async_setup_entry", return_value=True):
|
with patch("homeassistant.components.airly.async_setup_entry", return_value=True):
|
||||||
|
@ -15,7 +15,7 @@ from homeassistant.helpers import device_registry as dr, entity_registry as er
|
|||||||
|
|
||||||
from . import API_POINT_URL, init_integration
|
from . import API_POINT_URL, init_integration
|
||||||
|
|
||||||
from tests.common import MockConfigEntry, async_fire_time_changed, load_fixture
|
from tests.common import MockConfigEntry, async_fire_time_changed, async_load_fixture
|
||||||
from tests.test_util.aiohttp import AiohttpClientMocker
|
from tests.test_util.aiohttp import AiohttpClientMocker
|
||||||
|
|
||||||
|
|
||||||
@ -69,7 +69,9 @@ async def test_config_without_unique_id(
|
|||||||
},
|
},
|
||||||
)
|
)
|
||||||
|
|
||||||
aioclient_mock.get(API_POINT_URL, text=load_fixture("valid_station.json", "airly"))
|
aioclient_mock.get(
|
||||||
|
API_POINT_URL, text=await async_load_fixture(hass, "valid_station.json", DOMAIN)
|
||||||
|
)
|
||||||
entry.add_to_hass(hass)
|
entry.add_to_hass(hass)
|
||||||
await hass.config_entries.async_setup(entry.entry_id)
|
await hass.config_entries.async_setup(entry.entry_id)
|
||||||
assert entry.state is ConfigEntryState.LOADED
|
assert entry.state is ConfigEntryState.LOADED
|
||||||
@ -92,7 +94,9 @@ async def test_config_with_turned_off_station(
|
|||||||
},
|
},
|
||||||
)
|
)
|
||||||
|
|
||||||
aioclient_mock.get(API_POINT_URL, text=load_fixture("no_station.json", "airly"))
|
aioclient_mock.get(
|
||||||
|
API_POINT_URL, text=await async_load_fixture(hass, "no_station.json", DOMAIN)
|
||||||
|
)
|
||||||
entry.add_to_hass(hass)
|
entry.add_to_hass(hass)
|
||||||
await hass.config_entries.async_setup(entry.entry_id)
|
await hass.config_entries.async_setup(entry.entry_id)
|
||||||
assert entry.state is ConfigEntryState.SETUP_RETRY
|
assert entry.state is ConfigEntryState.SETUP_RETRY
|
||||||
@ -124,7 +128,7 @@ async def test_update_interval(
|
|||||||
|
|
||||||
aioclient_mock.get(
|
aioclient_mock.get(
|
||||||
API_POINT_URL,
|
API_POINT_URL,
|
||||||
text=load_fixture("valid_station.json", "airly"),
|
text=await async_load_fixture(hass, "valid_station.json", DOMAIN),
|
||||||
headers=HEADERS,
|
headers=HEADERS,
|
||||||
)
|
)
|
||||||
entry.add_to_hass(hass)
|
entry.add_to_hass(hass)
|
||||||
@ -159,7 +163,7 @@ async def test_update_interval(
|
|||||||
|
|
||||||
aioclient_mock.get(
|
aioclient_mock.get(
|
||||||
"https://airapi.airly.eu/v2/measurements/point?lat=66.660000&lng=111.110000",
|
"https://airapi.airly.eu/v2/measurements/point?lat=66.660000&lng=111.110000",
|
||||||
text=load_fixture("valid_station.json", "airly"),
|
text=await async_load_fixture(hass, "valid_station.json", DOMAIN),
|
||||||
headers=HEADERS,
|
headers=HEADERS,
|
||||||
)
|
)
|
||||||
entry.add_to_hass(hass)
|
entry.add_to_hass(hass)
|
||||||
@ -216,7 +220,9 @@ async def test_migrate_device_entry(
|
|||||||
},
|
},
|
||||||
)
|
)
|
||||||
|
|
||||||
aioclient_mock.get(API_POINT_URL, text=load_fixture("valid_station.json", "airly"))
|
aioclient_mock.get(
|
||||||
|
API_POINT_URL, text=await async_load_fixture(hass, "valid_station.json", DOMAIN)
|
||||||
|
)
|
||||||
config_entry.add_to_hass(hass)
|
config_entry.add_to_hass(hass)
|
||||||
|
|
||||||
device_entry = device_registry.async_get_or_create(
|
device_entry = device_registry.async_get_or_create(
|
||||||
|
@ -7,6 +7,7 @@ from unittest.mock import patch
|
|||||||
from airly.exceptions import AirlyError
|
from airly.exceptions import AirlyError
|
||||||
from syrupy.assertion import SnapshotAssertion
|
from syrupy.assertion import SnapshotAssertion
|
||||||
|
|
||||||
|
from homeassistant.components.airly.const import DOMAIN
|
||||||
from homeassistant.const import ATTR_ENTITY_ID, STATE_UNAVAILABLE, Platform
|
from homeassistant.const import ATTR_ENTITY_ID, STATE_UNAVAILABLE, Platform
|
||||||
from homeassistant.core import HomeAssistant
|
from homeassistant.core import HomeAssistant
|
||||||
from homeassistant.helpers import entity_registry as er
|
from homeassistant.helpers import entity_registry as er
|
||||||
@ -15,7 +16,7 @@ from homeassistant.util.dt import utcnow
|
|||||||
|
|
||||||
from . import API_POINT_URL, init_integration
|
from . import API_POINT_URL, init_integration
|
||||||
|
|
||||||
from tests.common import async_fire_time_changed, load_fixture
|
from tests.common import async_fire_time_changed, async_load_fixture
|
||||||
from tests.test_util.aiohttp import AiohttpClientMocker
|
from tests.test_util.aiohttp import AiohttpClientMocker
|
||||||
|
|
||||||
|
|
||||||
@ -62,7 +63,9 @@ async def test_availability(
|
|||||||
assert state.state == STATE_UNAVAILABLE
|
assert state.state == STATE_UNAVAILABLE
|
||||||
|
|
||||||
aioclient_mock.clear_requests()
|
aioclient_mock.clear_requests()
|
||||||
aioclient_mock.get(API_POINT_URL, text=load_fixture("valid_station.json", "airly"))
|
aioclient_mock.get(
|
||||||
|
API_POINT_URL, text=await async_load_fixture(hass, "valid_station.json", DOMAIN)
|
||||||
|
)
|
||||||
future = utcnow() + timedelta(minutes=120)
|
future = utcnow() + timedelta(minutes=120)
|
||||||
async_fire_time_changed(hass, future)
|
async_fire_time_changed(hass, future)
|
||||||
await hass.async_block_till_done()
|
await hass.async_block_till_done()
|
||||||
|
Loading…
x
Reference in New Issue
Block a user