mirror of
https://github.com/home-assistant/core.git
synced 2026-07-03 04:36:04 +01:00
TimL
c24186e571
Co-authored-by: Copilot Autofix powered by AI <175728472+Copilot@users.noreply.github.com> Co-authored-by: J. Nick Koston <nick@koston.org>
90 lines
3.0 KiB
Python
90 lines
3.0 KiB
Python
"""Tests for the SMLIGHT Bluetooth platform."""
|
|
|
|
from unittest.mock import ANY, MagicMock
|
|
|
|
from pysmlight import Info
|
|
from pysmlight.models import BleFeatures
|
|
import pytest
|
|
|
|
from homeassistant.config_entries import ConfigEntryState
|
|
from homeassistant.const import CONF_HOST
|
|
from homeassistant.core import HomeAssistant
|
|
|
|
from tests.common import MockConfigEntry
|
|
|
|
|
|
@pytest.mark.usefixtures("mock_ultima_client")
|
|
async def test_bluetooth_scanner_lifecycle(
|
|
hass: HomeAssistant,
|
|
mock_config_entry: MockConfigEntry,
|
|
mock_connect_scanner: MagicMock,
|
|
mock_bluetooth_scanner: MagicMock,
|
|
) -> None:
|
|
"""Test setting up and unloading SMLIGHT Bluetooth scanner (lifecycle)."""
|
|
mock_config_entry.add_to_hass(hass)
|
|
await hass.config_entries.async_setup(mock_config_entry.entry_id)
|
|
await hass.async_block_till_done()
|
|
|
|
assert mock_config_entry.state is ConfigEntryState.LOADED
|
|
mock_connect_scanner.assert_called_once_with(
|
|
source=mock_config_entry.unique_id,
|
|
name=mock_config_entry.title,
|
|
host=mock_config_entry.data[CONF_HOST],
|
|
port=5050,
|
|
)
|
|
|
|
client_data = mock_connect_scanner.return_value
|
|
client_data.client.start.assert_called_once()
|
|
mock_bluetooth_scanner.assert_called_once_with(
|
|
hass,
|
|
client_data.scanner,
|
|
source_domain="smlight",
|
|
source_model="SLZB-Ultima3",
|
|
source_config_entry_id=mock_config_entry.entry_id,
|
|
source_device_id=ANY,
|
|
)
|
|
|
|
await hass.config_entries.async_unload(mock_config_entry.entry_id)
|
|
await hass.async_block_till_done()
|
|
|
|
assert mock_config_entry.state is ConfigEntryState.NOT_LOADED
|
|
client_data.client.stop.assert_called_once()
|
|
|
|
|
|
async def test_bluetooth_not_started_for_disabled_settings(
|
|
hass: HomeAssistant,
|
|
mock_config_entry: MockConfigEntry,
|
|
mock_connect_scanner: MagicMock,
|
|
mock_smlight_client: MagicMock,
|
|
) -> None:
|
|
"""Test that bluetooth scanner is not started for SLZB device with disabled settings."""
|
|
mock_smlight_client.get_info.side_effect = None
|
|
mock_smlight_client.get_info.return_value = Info(
|
|
MAC="AA:BB:CC:DD:EE:FF",
|
|
model="SLZB-MR3U",
|
|
u_device=True,
|
|
ble=BleFeatures(proxy_enabled=False),
|
|
)
|
|
|
|
mock_config_entry.add_to_hass(hass)
|
|
await hass.config_entries.async_setup(mock_config_entry.entry_id)
|
|
await hass.async_block_till_done()
|
|
|
|
assert mock_config_entry.state is ConfigEntryState.LOADED
|
|
mock_connect_scanner.assert_not_called()
|
|
|
|
|
|
@pytest.mark.usefixtures("mock_smlight_client")
|
|
async def test_bluetooth_not_started_for_classic_device(
|
|
hass: HomeAssistant,
|
|
mock_config_entry: MockConfigEntry,
|
|
mock_connect_scanner: MagicMock,
|
|
) -> None:
|
|
"""Test that bluetooth scanner is not started for classic (non-U) devices."""
|
|
mock_config_entry.add_to_hass(hass)
|
|
await hass.config_entries.async_setup(mock_config_entry.entry_id)
|
|
await hass.async_block_till_done()
|
|
|
|
assert mock_config_entry.state is ConfigEntryState.LOADED
|
|
mock_connect_scanner.assert_not_called()
|