jeans/core
jeans/core
1
0
Fork 0
mirror of https://github.com/home-assistant/core.git synced 2025-04-23 08:47:57 +00:00
Code Issues Packages Projects Releases Wiki Activity

Improve type hints in habitica tests (#121212)

Browse Source
This commit is contained in:
epenet 2024-07-04 16:27:57 +02:00 committed by GitHub
parent 9a1f7f020c
commit 411630429d
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 9 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
tests/components/habitica/test_init.py
View File

@ -14,7 +14,7 @@ from homeassistant.components.habitica.const import (
SERVICE_API_CALL,
)
from homeassistant.const import ATTR_NAME
from homeassistant.core import HomeAssistant
from homeassistant.core import Event, HomeAssistant
from tests.common import MockConfigEntry, async_capture_events
from tests.test_util.aiohttp import AiohttpClientMocker
@ -24,13 +24,13 @@ TEST_USER_NAME = "test_user"
@pytest.fixture
def capture_api_call_success(hass):
def capture_api_call_success(hass: HomeAssistant) -> list[Event]:
"""Capture api_call events."""
return async_capture_events(hass, EVENT_API_CALL_SUCCESS)
@pytest.fixture
def habitica_entry(hass):
def habitica_entry(hass: HomeAssistant) -> MockConfigEntry:
"""Test entry for the following tests."""
entry = MockConfigEntry(
domain=DOMAIN,
@ -98,8 +98,9 @@ def common_requests(aioclient_mock: AiohttpClientMocker) -> AiohttpClientMocker:
return aioclient_mock
@pytest.mark.usefixtures("common_requests")
async def test_entry_setup_unload(
hass: HomeAssistant, habitica_entry, common_requests
hass: HomeAssistant, habitica_entry: MockConfigEntry
) -> None:
"""Test integration setup and unload."""
assert await hass.config_entries.async_setup(habitica_entry.entry_id)
@ -112,8 +113,11 @@ async def test_entry_setup_unload(
assert not hass.services.has_service(DOMAIN, SERVICE_API_CALL)
@pytest.mark.usefixtures("common_requests")
async def test_service_call(
hass: HomeAssistant, habitica_entry, common_requests, capture_api_call_success
hass: HomeAssistant,
habitica_entry: MockConfigEntry,
capture_api_call_success: list[Event],
) -> None:
"""Test integration setup, service call and unload."""