Improve type hints in blueprint tests (#119263)

This commit is contained in:
epenet 2024-06-10 08:49:18 +02:00 committed by GitHub
parent ea3097f84c
commit 2d2f5de191
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
2 changed files with 27 additions and 25 deletions

View file

@ -1,6 +1,7 @@
"""Test websocket API."""
from pathlib import Path
from typing import Any
from unittest.mock import Mock, patch
import pytest
@ -15,19 +16,23 @@ from tests.typing import WebSocketGenerator
@pytest.fixture
def automation_config():
def automation_config() -> dict[str, Any]:
"""Automation config."""
return {}
@pytest.fixture
def script_config():
def script_config() -> dict[str, Any]:
"""Script config."""
return {}
@pytest.fixture(autouse=True)
async def setup_bp(hass, automation_config, script_config):
async def setup_bp(
hass: HomeAssistant,
automation_config: dict[str, Any],
script_config: dict[str, Any],
) -> None:
"""Fixture to set up the blueprint component."""
assert await async_setup_component(hass, "blueprint", {})
@ -135,11 +140,11 @@ async def test_import_blueprint(
}
@pytest.mark.usefixtures("setup_bp")
async def test_import_blueprint_update(
hass: HomeAssistant,
aioclient_mock: AiohttpClientMocker,
hass_ws_client: WebSocketGenerator,
setup_bp,
) -> None:
"""Test importing blueprints."""
raw_data = Path(
@ -182,7 +187,6 @@ async def test_import_blueprint_update(
async def test_save_blueprint(
hass: HomeAssistant,
aioclient_mock: AiohttpClientMocker,
hass_ws_client: WebSocketGenerator,
) -> None:
"""Test saving blueprints."""
@ -236,7 +240,6 @@ async def test_save_blueprint(
async def test_save_existing_file(
hass: HomeAssistant,
aioclient_mock: AiohttpClientMocker,
hass_ws_client: WebSocketGenerator,
) -> None:
"""Test saving blueprints."""
@ -262,7 +265,6 @@ async def test_save_existing_file(
async def test_save_existing_file_override(
hass: HomeAssistant,
aioclient_mock: AiohttpClientMocker,
hass_ws_client: WebSocketGenerator,
) -> None:
"""Test saving blueprints."""
@ -298,7 +300,6 @@ async def test_save_existing_file_override(
async def test_save_file_error(
hass: HomeAssistant,
aioclient_mock: AiohttpClientMocker,
hass_ws_client: WebSocketGenerator,
) -> None:
"""Test saving blueprints with OS error."""
@ -323,7 +324,6 @@ async def test_save_file_error(
async def test_save_invalid_blueprint(
hass: HomeAssistant,
aioclient_mock: AiohttpClientMocker,
hass_ws_client: WebSocketGenerator,
) -> None:
"""Test saving invalid blueprints."""
@ -352,7 +352,6 @@ async def test_save_invalid_blueprint(
async def test_delete_blueprint(
hass: HomeAssistant,
aioclient_mock: AiohttpClientMocker,
hass_ws_client: WebSocketGenerator,
) -> None:
"""Test deleting blueprints."""
@ -377,7 +376,6 @@ async def test_delete_blueprint(
async def test_delete_non_exist_file_blueprint(
hass: HomeAssistant,
aioclient_mock: AiohttpClientMocker,
hass_ws_client: WebSocketGenerator,
) -> None:
"""Test deleting non existing blueprints."""
@ -417,7 +415,6 @@ async def test_delete_non_exist_file_blueprint(
)
async def test_delete_blueprint_in_use_by_automation(
hass: HomeAssistant,
aioclient_mock: AiohttpClientMocker,
hass_ws_client: WebSocketGenerator,
) -> None:
"""Test deleting a blueprint which is in use."""
@ -463,7 +460,6 @@ async def test_delete_blueprint_in_use_by_automation(
)
async def test_delete_blueprint_in_use_by_script(
hass: HomeAssistant,
aioclient_mock: AiohttpClientMocker,
hass_ws_client: WebSocketGenerator,
) -> None:
"""Test deleting a blueprint which is in use."""