evlli/hass-core
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Improve type hints in canary tests (#120305)

Browse source
This commit is contained in:
epenet 2024-06-24 13:34:44 +02:00 • committed by GitHub
parent e3806d12f4
commit aef2f7d707
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 6 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

9
tests/components/canary/conftest.py
View file

@ -4,16 +4,19 @@ from unittest.mock import MagicMock, patch
from canary.api import Api
import pytest
from typing_extensions import Generator
from homeassistant.core import HomeAssistant
@pytest.fixture(autouse=True)
def mock_ffmpeg(hass):
def mock_ffmpeg(hass: HomeAssistant) -> None:
"""Mock ffmpeg is loaded."""
hass.config.components.add("ffmpeg")
@pytest.fixture
def canary(hass):
def canary() -> Generator[MagicMock]:
"""Mock the CanaryApi for easier testing."""
with (
patch.object(Api, "login", return_value=True),
@ -38,7 +41,7 @@ def canary(hass):
@pytest.fixture
def canary_config_flow(hass):
def canary_config_flow() -> Generator[MagicMock]:
"""Mock the CanaryApi for easier config flow testing."""
with (
patch.object(Api, "login", return_value=True),