mirror of
https://github.com/home-assistant/core.git
synced 2025-07-21 04:07:08 +00:00
Improve type hints in flo tests (#120730)
This commit is contained in:
parent
2e031d0194
commit
d0ce0d562e
@ -16,7 +16,7 @@ from tests.test_util.aiohttp import AiohttpClientMocker
|
|||||||
|
|
||||||
|
|
||||||
@pytest.fixture
|
@pytest.fixture
|
||||||
def config_entry(hass):
|
def config_entry() -> MockConfigEntry:
|
||||||
"""Config entry version 1 fixture."""
|
"""Config entry version 1 fixture."""
|
||||||
return MockConfigEntry(
|
return MockConfigEntry(
|
||||||
domain=FLO_DOMAIN,
|
domain=FLO_DOMAIN,
|
||||||
|
@ -1,5 +1,7 @@
|
|||||||
"""Test Flo by Moen binary sensor entities."""
|
"""Test Flo by Moen binary sensor entities."""
|
||||||
|
|
||||||
|
import pytest
|
||||||
|
|
||||||
from homeassistant.components.flo.const import DOMAIN as FLO_DOMAIN
|
from homeassistant.components.flo.const import DOMAIN as FLO_DOMAIN
|
||||||
from homeassistant.const import (
|
from homeassistant.const import (
|
||||||
ATTR_FRIENDLY_NAME,
|
ATTR_FRIENDLY_NAME,
|
||||||
@ -13,9 +15,12 @@ from homeassistant.setup import async_setup_component
|
|||||||
|
|
||||||
from .common import TEST_PASSWORD, TEST_USER_ID
|
from .common import TEST_PASSWORD, TEST_USER_ID
|
||||||
|
|
||||||
|
from tests.common import MockConfigEntry
|
||||||
|
|
||||||
|
|
||||||
|
@pytest.mark.usefixtures("aioclient_mock_fixture")
|
||||||
async def test_binary_sensors(
|
async def test_binary_sensors(
|
||||||
hass: HomeAssistant, config_entry, aioclient_mock_fixture
|
hass: HomeAssistant, config_entry: MockConfigEntry
|
||||||
) -> None:
|
) -> None:
|
||||||
"""Test Flo by Moen sensors."""
|
"""Test Flo by Moen sensors."""
|
||||||
config_entry.add_to_hass(hass)
|
config_entry.add_to_hass(hass)
|
||||||
|
@ -5,6 +5,8 @@ import json
|
|||||||
import time
|
import time
|
||||||
from unittest.mock import patch
|
from unittest.mock import patch
|
||||||
|
|
||||||
|
import pytest
|
||||||
|
|
||||||
from homeassistant import config_entries
|
from homeassistant import config_entries
|
||||||
from homeassistant.components.flo.const import DOMAIN
|
from homeassistant.components.flo.const import DOMAIN
|
||||||
from homeassistant.const import CONTENT_TYPE_JSON
|
from homeassistant.const import CONTENT_TYPE_JSON
|
||||||
@ -16,7 +18,8 @@ from .common import TEST_EMAIL_ADDRESS, TEST_PASSWORD, TEST_TOKEN, TEST_USER_ID
|
|||||||
from tests.test_util.aiohttp import AiohttpClientMocker
|
from tests.test_util.aiohttp import AiohttpClientMocker
|
||||||
|
|
||||||
|
|
||||||
async def test_form(hass: HomeAssistant, aioclient_mock_fixture) -> None:
|
@pytest.mark.usefixtures("aioclient_mock_fixture")
|
||||||
|
async def test_form(hass: HomeAssistant) -> None:
|
||||||
"""Test we get the form."""
|
"""Test we get the form."""
|
||||||
|
|
||||||
result = await hass.config_entries.flow.async_init(
|
result = await hass.config_entries.flow.async_init(
|
||||||
|
@ -5,6 +5,7 @@ from unittest.mock import patch
|
|||||||
|
|
||||||
from aioflo.errors import RequestError
|
from aioflo.errors import RequestError
|
||||||
from freezegun.api import FrozenDateTimeFactory
|
from freezegun.api import FrozenDateTimeFactory
|
||||||
|
import pytest
|
||||||
|
|
||||||
from homeassistant.components.flo.const import DOMAIN as FLO_DOMAIN
|
from homeassistant.components.flo.const import DOMAIN as FLO_DOMAIN
|
||||||
from homeassistant.components.flo.coordinator import FloDeviceDataUpdateCoordinator
|
from homeassistant.components.flo.coordinator import FloDeviceDataUpdateCoordinator
|
||||||
@ -14,14 +15,14 @@ from homeassistant.setup import async_setup_component
|
|||||||
|
|
||||||
from .common import TEST_PASSWORD, TEST_USER_ID
|
from .common import TEST_PASSWORD, TEST_USER_ID
|
||||||
|
|
||||||
from tests.common import async_fire_time_changed
|
from tests.common import MockConfigEntry, async_fire_time_changed
|
||||||
from tests.test_util.aiohttp import AiohttpClientMocker
|
from tests.test_util.aiohttp import AiohttpClientMocker
|
||||||
|
|
||||||
|
|
||||||
|
@pytest.mark.usefixtures("aioclient_mock_fixture")
|
||||||
async def test_device(
|
async def test_device(
|
||||||
hass: HomeAssistant,
|
hass: HomeAssistant,
|
||||||
config_entry,
|
config_entry: MockConfigEntry,
|
||||||
aioclient_mock_fixture,
|
|
||||||
aioclient_mock: AiohttpClientMocker,
|
aioclient_mock: AiohttpClientMocker,
|
||||||
freezer: FrozenDateTimeFactory,
|
freezer: FrozenDateTimeFactory,
|
||||||
) -> None:
|
) -> None:
|
||||||
@ -90,10 +91,10 @@ async def test_device(
|
|||||||
assert aioclient_mock.call_count == call_count + 6
|
assert aioclient_mock.call_count == call_count + 6
|
||||||
|
|
||||||
|
|
||||||
|
@pytest.mark.usefixtures("aioclient_mock_fixture")
|
||||||
async def test_device_failures(
|
async def test_device_failures(
|
||||||
hass: HomeAssistant,
|
hass: HomeAssistant,
|
||||||
config_entry,
|
config_entry: MockConfigEntry,
|
||||||
aioclient_mock_fixture,
|
|
||||||
aioclient_mock: AiohttpClientMocker,
|
aioclient_mock: AiohttpClientMocker,
|
||||||
freezer: FrozenDateTimeFactory,
|
freezer: FrozenDateTimeFactory,
|
||||||
) -> None:
|
) -> None:
|
||||||
|
@ -1,5 +1,7 @@
|
|||||||
"""Test init."""
|
"""Test init."""
|
||||||
|
|
||||||
|
import pytest
|
||||||
|
|
||||||
from homeassistant.components.flo.const import DOMAIN as FLO_DOMAIN
|
from homeassistant.components.flo.const import DOMAIN as FLO_DOMAIN
|
||||||
from homeassistant.const import CONF_PASSWORD, CONF_USERNAME
|
from homeassistant.const import CONF_PASSWORD, CONF_USERNAME
|
||||||
from homeassistant.core import HomeAssistant
|
from homeassistant.core import HomeAssistant
|
||||||
@ -7,10 +9,11 @@ from homeassistant.setup import async_setup_component
|
|||||||
|
|
||||||
from .common import TEST_PASSWORD, TEST_USER_ID
|
from .common import TEST_PASSWORD, TEST_USER_ID
|
||||||
|
|
||||||
|
from tests.common import MockConfigEntry
|
||||||
|
|
||||||
async def test_setup_entry(
|
|
||||||
hass: HomeAssistant, config_entry, aioclient_mock_fixture
|
@pytest.mark.usefixtures("aioclient_mock_fixture")
|
||||||
) -> None:
|
async def test_setup_entry(hass: HomeAssistant, config_entry: MockConfigEntry) -> None:
|
||||||
"""Test migration of config entry from v1."""
|
"""Test migration of config entry from v1."""
|
||||||
config_entry.add_to_hass(hass)
|
config_entry.add_to_hass(hass)
|
||||||
assert await async_setup_component(
|
assert await async_setup_component(
|
||||||
|
@ -1,5 +1,7 @@
|
|||||||
"""Test Flo by Moen sensor entities."""
|
"""Test Flo by Moen sensor entities."""
|
||||||
|
|
||||||
|
import pytest
|
||||||
|
|
||||||
from homeassistant.components.flo.const import DOMAIN as FLO_DOMAIN
|
from homeassistant.components.flo.const import DOMAIN as FLO_DOMAIN
|
||||||
from homeassistant.components.sensor import ATTR_STATE_CLASS, SensorStateClass
|
from homeassistant.components.sensor import ATTR_STATE_CLASS, SensorStateClass
|
||||||
from homeassistant.const import ATTR_ENTITY_ID, CONF_PASSWORD, CONF_USERNAME
|
from homeassistant.const import ATTR_ENTITY_ID, CONF_PASSWORD, CONF_USERNAME
|
||||||
@ -9,12 +11,12 @@ from homeassistant.util.unit_system import US_CUSTOMARY_SYSTEM
|
|||||||
|
|
||||||
from .common import TEST_PASSWORD, TEST_USER_ID
|
from .common import TEST_PASSWORD, TEST_USER_ID
|
||||||
|
|
||||||
|
from tests.common import MockConfigEntry
|
||||||
from tests.test_util.aiohttp import AiohttpClientMocker
|
from tests.test_util.aiohttp import AiohttpClientMocker
|
||||||
|
|
||||||
|
|
||||||
async def test_sensors(
|
@pytest.mark.usefixtures("aioclient_mock_fixture")
|
||||||
hass: HomeAssistant, config_entry, aioclient_mock_fixture
|
async def test_sensors(hass: HomeAssistant, config_entry: MockConfigEntry) -> None:
|
||||||
) -> None:
|
|
||||||
"""Test Flo by Moen sensors."""
|
"""Test Flo by Moen sensors."""
|
||||||
hass.config.units = US_CUSTOMARY_SYSTEM
|
hass.config.units = US_CUSTOMARY_SYSTEM
|
||||||
config_entry.add_to_hass(hass)
|
config_entry.add_to_hass(hass)
|
||||||
@ -85,10 +87,10 @@ async def test_sensors(
|
|||||||
)
|
)
|
||||||
|
|
||||||
|
|
||||||
|
@pytest.mark.usefixtures("aioclient_mock_fixture")
|
||||||
async def test_manual_update_entity(
|
async def test_manual_update_entity(
|
||||||
hass: HomeAssistant,
|
hass: HomeAssistant,
|
||||||
config_entry,
|
config_entry: MockConfigEntry,
|
||||||
aioclient_mock_fixture,
|
|
||||||
aioclient_mock: AiohttpClientMocker,
|
aioclient_mock: AiohttpClientMocker,
|
||||||
) -> None:
|
) -> None:
|
||||||
"""Test manual update entity via service homeasasistant/update_entity."""
|
"""Test manual update entity via service homeasasistant/update_entity."""
|
||||||
|
@ -19,15 +19,16 @@ from homeassistant.setup import async_setup_component
|
|||||||
|
|
||||||
from .common import TEST_PASSWORD, TEST_USER_ID
|
from .common import TEST_PASSWORD, TEST_USER_ID
|
||||||
|
|
||||||
|
from tests.common import MockConfigEntry
|
||||||
from tests.test_util.aiohttp import AiohttpClientMocker
|
from tests.test_util.aiohttp import AiohttpClientMocker
|
||||||
|
|
||||||
SWITCH_ENTITY_ID = "switch.smart_water_shutoff_shutoff_valve"
|
SWITCH_ENTITY_ID = "switch.smart_water_shutoff_shutoff_valve"
|
||||||
|
|
||||||
|
|
||||||
|
@pytest.mark.usefixtures("aioclient_mock_fixture")
|
||||||
async def test_services(
|
async def test_services(
|
||||||
hass: HomeAssistant,
|
hass: HomeAssistant,
|
||||||
config_entry,
|
config_entry: MockConfigEntry,
|
||||||
aioclient_mock_fixture,
|
|
||||||
aioclient_mock: AiohttpClientMocker,
|
aioclient_mock: AiohttpClientMocker,
|
||||||
) -> None:
|
) -> None:
|
||||||
"""Test Flo services."""
|
"""Test Flo services."""
|
||||||
|
@ -1,5 +1,7 @@
|
|||||||
"""Tests for the switch domain for Flo by Moen."""
|
"""Tests for the switch domain for Flo by Moen."""
|
||||||
|
|
||||||
|
import pytest
|
||||||
|
|
||||||
from homeassistant.components.flo.const import DOMAIN as FLO_DOMAIN
|
from homeassistant.components.flo.const import DOMAIN as FLO_DOMAIN
|
||||||
from homeassistant.components.switch import DOMAIN
|
from homeassistant.components.switch import DOMAIN
|
||||||
from homeassistant.const import CONF_PASSWORD, CONF_USERNAME, STATE_OFF, STATE_ON
|
from homeassistant.const import CONF_PASSWORD, CONF_USERNAME, STATE_OFF, STATE_ON
|
||||||
@ -8,9 +10,12 @@ from homeassistant.setup import async_setup_component
|
|||||||
|
|
||||||
from .common import TEST_PASSWORD, TEST_USER_ID
|
from .common import TEST_PASSWORD, TEST_USER_ID
|
||||||
|
|
||||||
|
from tests.common import MockConfigEntry
|
||||||
|
|
||||||
|
|
||||||
|
@pytest.mark.usefixtures("aioclient_mock_fixture")
|
||||||
async def test_valve_switches(
|
async def test_valve_switches(
|
||||||
hass: HomeAssistant, config_entry, aioclient_mock_fixture
|
hass: HomeAssistant, config_entry: MockConfigEntry
|
||||||
) -> None:
|
) -> None:
|
||||||
"""Test Flo by Moen valve switches."""
|
"""Test Flo by Moen valve switches."""
|
||||||
config_entry.add_to_hass(hass)
|
config_entry.add_to_hass(hass)
|
||||||
|
Loading…
x
Reference in New Issue
Block a user