Improve type hints in zerproc tests (#123925)

This commit is contained in:
epenet
2024-08-14 15:41:15 +02:00
committed by GitHub
parent c65f845329
commit 1227cd8693

View File

@@ -35,13 +35,13 @@ from tests.common import MockConfigEntry, async_fire_time_changed
@pytest.fixture
async def mock_entry(hass):
async def mock_entry() -> MockConfigEntry:
"""Create a mock light entity."""
return MockConfigEntry(domain=DOMAIN)
@pytest.fixture
async def mock_light(hass, mock_entry):
async def mock_light(hass: HomeAssistant, mock_entry: MockConfigEntry) -> MagicMock:
"""Create a mock light entity."""
mock_entry.add_to_hass(hass)