jeans/core
jeans/core
1
0
Fork 0
mirror of https://github.com/home-assistant/core.git synced 2025-07-27 23:27:37 +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, SERVICE_API_CALL,
) )
from homeassistant.const import ATTR_NAME 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.common import MockConfigEntry, async_capture_events
from tests.test_util.aiohttp import AiohttpClientMocker from tests.test_util.aiohttp import AiohttpClientMocker
@ -24,13 +24,13 @@ TEST_USER_NAME = "test_user"
@pytest.fixture @pytest.fixture
def capture_api_call_success(hass): def capture_api_call_success(hass: HomeAssistant) -> list[Event]:
"""Capture api_call events.""" """Capture api_call events."""
return async_capture_events(hass, EVENT_API_CALL_SUCCESS) return async_capture_events(hass, EVENT_API_CALL_SUCCESS)
@pytest.fixture @pytest.fixture
def habitica_entry(hass): def habitica_entry(hass: HomeAssistant) -> MockConfigEntry:
"""Test entry for the following tests.""" """Test entry for the following tests."""
entry = MockConfigEntry( entry = MockConfigEntry(
domain=DOMAIN, domain=DOMAIN,
@ -98,8 +98,9 @@ def common_requests(aioclient_mock: AiohttpClientMocker) -> AiohttpClientMocker:
return aioclient_mock return aioclient_mock
@pytest.mark.usefixtures("common_requests")
async def test_entry_setup_unload( async def test_entry_setup_unload(
hass: HomeAssistant, habitica_entry, common_requests hass: HomeAssistant, habitica_entry: MockConfigEntry
) -> None: ) -> None:
"""Test integration setup and unload.""" """Test integration setup and unload."""
assert await hass.config_entries.async_setup(habitica_entry.entry_id) 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) assert not hass.services.has_service(DOMAIN, SERVICE_API_CALL)
@pytest.mark.usefixtures("common_requests")
async def test_service_call( 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: ) -> None:
"""Test integration setup, service call and unload.""" """Test integration setup, service call and unload."""