jeans/core
jeans/core
1
0
Fork 0
mirror of https://github.com/home-assistant/core.git synced 2025-07-25 22:27:07 +00:00
Code Issues Packages Projects Releases Wiki Activity

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