evlli/hass-core
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Ensure bleak_retry_connector uses HaBleakClientWrapper (#79132)

Browse source
This commit is contained in:
J. Nick Koston 2022-09-27 08:06:10 -10:00 • committed by GitHub
parent 8eaa22cf90
commit 5eb50f63fd
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
2 changed files with 49 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

30
tests/components/bluetooth/test_usage.py
View file

@ -5,6 +5,7 @@ from unittest.mock import patch
import bleak
from bleak.backends.device import BLEDevice
import bleak_retry_connector
from homeassistant.components.bluetooth.models import (
HaBleakClientWrapper,
@ -75,3 +76,32 @@ async def test_bleak_client_reports_with_address(hass, enable_bluetooth, caplog)
assert not isinstance(instance, HaBleakClientWrapper)
assert "BleakClient with an address instead of a BLEDevice" not in caplog.text
async def test_bleak_retry_connector_client_reports_with_address(
hass, enable_bluetooth, caplog
):
"""Test we report when we pass an address to BleakClientWithServiceCache."""
install_multiple_bleak_catcher()
with patch.object(
_get_manager(),
"async_ble_device_from_address",
return_value=MOCK_BLE_DEVICE,
):
instance = bleak_retry_connector.BleakClientWithServiceCache(
"00:00:00:00:00:00"
)
assert "BleakClient with an address instead of a BLEDevice" in caplog.text
assert isinstance(instance, HaBleakClientWrapper)
uninstall_multiple_bleak_catcher()
caplog.clear()
instance = bleak_retry_connector.BleakClientWithServiceCache("00:00:00:00:00:00")
assert not isinstance(instance, HaBleakClientWrapper)
assert "BleakClient with an address instead of a BLEDevice" not in caplog.text