diff --git a/tests/components/blebox/test_binary_sensor.py b/tests/components/blebox/test_binary_sensor.py index 36258198d91..c9181762f3e 100644 --- a/tests/components/blebox/test_binary_sensor.py +++ b/tests/components/blebox/test_binary_sensor.py @@ -1,20 +1,21 @@ """Blebox binary_sensor entities test.""" -from unittest.mock import PropertyMock +from unittest.mock import AsyncMock, PropertyMock import blebox_uniapi import pytest from homeassistant.components.binary_sensor import BinarySensorDeviceClass from homeassistant.const import ATTR_DEVICE_CLASS, STATE_ON +from homeassistant.core import HomeAssistant from homeassistant.helpers import device_registry as dr from .conftest import async_setup_entity, mock_feature @pytest.fixture(name="rainsensor") -def airsensor_fixture(): +def airsensor_fixture() -> tuple[AsyncMock, str]: """Return a default air quality fixture.""" - feature = mock_feature( + feature: AsyncMock = mock_feature( "binary_sensors", blebox_uniapi.binary_sensor.Rain, unique_id="BleBox-windRainSensor-ea68e74f4f49-0.rain", @@ -24,10 +25,10 @@ def airsensor_fixture(): product = feature.product type(product).name = PropertyMock(return_value="My rain sensor") type(product).model = PropertyMock(return_value="rainSensor") - return (feature, "binary_sensor.windrainsensor_0_rain") + return feature, "binary_sensor.windrainsensor_0_rain" -async def test_init(rainsensor, hass, config): +async def test_init(rainsensor: AsyncMock, hass: HomeAssistant, config: dict): """Test binary_sensor initialisation.""" _, entity_id = rainsensor entry = await async_setup_entity(hass, config, entity_id)