mirror of
https://github.com/home-assistant/core.git
synced 2025-12-20 02:48:57 +00:00
ceeeb22040
Co-authored-by: Norbert Rittel <norbert@rittel.de> Co-authored-by: Josef Zweck <josef@zweck.dev> Co-authored-by: Joost Lekkerkerker <joostlek@outlook.com>
42 lines
1.2 KiB
Python
42 lines
1.2 KiB
Python
"""Test Droplet initialization."""
|
|
|
|
from unittest.mock import AsyncMock
|
|
|
|
import pytest
|
|
|
|
from homeassistant.config_entries import ConfigEntryState
|
|
from homeassistant.core import HomeAssistant
|
|
|
|
from . import setup_integration
|
|
|
|
from tests.common import MockConfigEntry
|
|
|
|
|
|
async def test_setup_no_version_info(
|
|
hass: HomeAssistant,
|
|
mock_config_entry: MockConfigEntry,
|
|
mock_droplet_discovery: AsyncMock,
|
|
mock_droplet_connection: AsyncMock,
|
|
mock_droplet: AsyncMock,
|
|
caplog: pytest.LogCaptureFixture,
|
|
) -> None:
|
|
"""Test coordinator setup where Droplet never sends version info."""
|
|
mock_droplet.version_info_available.return_value = False
|
|
await setup_integration(hass, mock_config_entry)
|
|
|
|
assert "Failed to get version info from Droplet" in caplog.text
|
|
|
|
|
|
async def test_setup_droplet_offline(
|
|
hass: HomeAssistant,
|
|
mock_config_entry: MockConfigEntry,
|
|
mock_droplet_discovery: AsyncMock,
|
|
mock_droplet_connection: AsyncMock,
|
|
mock_droplet: AsyncMock,
|
|
) -> None:
|
|
"""Test integration setup when Droplet is offline."""
|
|
mock_droplet.connected = False
|
|
await setup_integration(hass, mock_config_entry)
|
|
|
|
assert mock_config_entry.state is ConfigEntryState.SETUP_RETRY
|