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

Adjust imap tests

Browse Source
This commit is contained in:
Erik 2025-05-07 17:12:24 +02:00
parent 473b77279f
commit fdcaf2897a
1 changed files with 7 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
tests/components/imap/test_init.py
View File

@ -1,6 +1,9 @@
"""Test the imap entry initialization.""" """Test the imap entry initialization."""
from __future__ import annotations
import asyncio import asyncio
from collections.abc import Callable
from datetime import datetime, timedelta, timezone from datetime import datetime, timedelta, timezone
from typing import Any from typing import Any
from unittest.mock import AsyncMock, MagicMock, call, patch from unittest.mock import AsyncMock, MagicMock, call, patch
@ -422,8 +425,8 @@ async def test_late_folder_error(
@pytest.mark.parametrize( @pytest.mark.parametrize(
"imap_close", "imap_close",
[ [
AsyncMock(side_effect=AioImapException("Something went wrong")), lambda: AsyncMock(side_effect=AioImapException("Something went wrong")),
AsyncMock(side_effect=TimeoutError), lambda: AsyncMock(side_effect=TimeoutError),
], ],
ids=["AioImapException", "TimeoutError"], ids=["AioImapException", "TimeoutError"],
) )
@ -431,7 +434,7 @@ async def test_handle_cleanup_exception(
hass: HomeAssistant, hass: HomeAssistant,
caplog: pytest.LogCaptureFixture, caplog: pytest.LogCaptureFixture,
mock_imap_protocol: MagicMock, mock_imap_protocol: MagicMock,
imap_close: Exception, imap_close: Callable[[], AsyncMock],
) -> None: ) -> None:
"""Test handling an excepton during cleaning up.""" """Test handling an excepton during cleaning up."""
config_entry = MockConfigEntry(domain=DOMAIN, data=MOCK_CONFIG) config_entry = MockConfigEntry(domain=DOMAIN, data=MOCK_CONFIG)
@ -448,7 +451,7 @@ async def test_handle_cleanup_exception(
assert state.state == "0" assert state.state == "0"
# Fail cleaning up # Fail cleaning up
mock_imap_protocol.close.side_effect = imap_close mock_imap_protocol.close.side_effect = imap_close()
assert await hass.config_entries.async_unload(config_entry.entry_id) assert await hass.config_entries.async_unload(config_entry.entry_id)
await hass.async_block_till_done() await hass.async_block_till_done()