mirror of
https://github.com/home-assistant/core.git
synced 2025-08-06 12:08:19 +00:00
Fix tests for connect changes
This commit is contained in:
parent
ef4699375c
commit
9bc6257538
@ -124,7 +124,7 @@ async def test_wrapped_bleak_client_local_adapter_only(hass: HomeAssistant) -> N
|
|||||||
"bleak.backends.bluezdbus.client.BleakClientBlueZDBus.is_connected", True
|
"bleak.backends.bluezdbus.client.BleakClientBlueZDBus.is_connected", True
|
||||||
),
|
),
|
||||||
):
|
):
|
||||||
assert await client.connect() is True
|
await client.connect()
|
||||||
assert client.is_connected is True
|
assert client.is_connected is True
|
||||||
client.set_disconnected_callback(lambda client: None)
|
client.set_disconnected_callback(lambda client: None)
|
||||||
await client.disconnect()
|
await client.disconnect()
|
||||||
@ -215,7 +215,7 @@ async def test_wrapped_bleak_client_set_disconnected_callback_after_connected(
|
|||||||
"bleak.backends.bluezdbus.client.BleakClientBlueZDBus.is_connected", True
|
"bleak.backends.bluezdbus.client.BleakClientBlueZDBus.is_connected", True
|
||||||
),
|
),
|
||||||
):
|
):
|
||||||
assert await client.connect() is True
|
await client.connect()
|
||||||
assert client.is_connected is True
|
assert client.is_connected is True
|
||||||
client.set_disconnected_callback(lambda client: None)
|
client.set_disconnected_callback(lambda client: None)
|
||||||
await client.disconnect()
|
await client.disconnect()
|
||||||
|
@ -92,6 +92,10 @@ class FakeBleakClient(BaseFakeBleakClient):
|
|||||||
|
|
||||||
async def connect(self, *args, **kwargs):
|
async def connect(self, *args, **kwargs):
|
||||||
"""Connect."""
|
"""Connect."""
|
||||||
|
|
||||||
|
@property
|
||||||
|
def is_connected(self):
|
||||||
|
"""Connected."""
|
||||||
return True
|
return True
|
||||||
|
|
||||||
|
|
||||||
@ -100,6 +104,10 @@ class FakeBleakClientFailsToConnect(BaseFakeBleakClient):
|
|||||||
|
|
||||||
async def connect(self, *args, **kwargs):
|
async def connect(self, *args, **kwargs):
|
||||||
"""Connect."""
|
"""Connect."""
|
||||||
|
|
||||||
|
@property
|
||||||
|
def is_connected(self):
|
||||||
|
"""Not connected."""
|
||||||
return False
|
return False
|
||||||
|
|
||||||
|
|
||||||
@ -110,6 +118,11 @@ class FakeBleakClientRaisesOnConnect(BaseFakeBleakClient):
|
|||||||
"""Connect."""
|
"""Connect."""
|
||||||
raise ConnectionError("Test exception")
|
raise ConnectionError("Test exception")
|
||||||
|
|
||||||
|
@property
|
||||||
|
def is_connected(self):
|
||||||
|
"""Not connected."""
|
||||||
|
return False
|
||||||
|
|
||||||
|
|
||||||
def _generate_ble_device_and_adv_data(
|
def _generate_ble_device_and_adv_data(
|
||||||
interface: str, mac: str, rssi: int
|
interface: str, mac: str, rssi: int
|
||||||
@ -219,7 +232,8 @@ async def test_test_switch_adapters_when_out_of_slots(
|
|||||||
):
|
):
|
||||||
ble_device = hci0_device_advs["00:00:00:00:00:01"][0]
|
ble_device = hci0_device_advs["00:00:00:00:00:01"][0]
|
||||||
client = bleak.BleakClient(ble_device)
|
client = bleak.BleakClient(ble_device)
|
||||||
assert await client.connect() is True
|
await client.connect()
|
||||||
|
assert client.is_connected is True
|
||||||
assert allocate_slot_mock.call_count == 1
|
assert allocate_slot_mock.call_count == 1
|
||||||
assert release_slot_mock.call_count == 0
|
assert release_slot_mock.call_count == 0
|
||||||
|
|
||||||
@ -251,7 +265,8 @@ async def test_test_switch_adapters_when_out_of_slots(
|
|||||||
):
|
):
|
||||||
ble_device = hci0_device_advs["00:00:00:00:00:03"][0]
|
ble_device = hci0_device_advs["00:00:00:00:00:03"][0]
|
||||||
client = bleak.BleakClient(ble_device)
|
client = bleak.BleakClient(ble_device)
|
||||||
assert await client.connect() is True
|
await client.connect()
|
||||||
|
assert client.is_connected is True
|
||||||
assert release_slot_mock.call_count == 0
|
assert release_slot_mock.call_count == 0
|
||||||
|
|
||||||
cancel_hci0()
|
cancel_hci0()
|
||||||
@ -278,7 +293,8 @@ async def test_release_slot_on_connect_failure(
|
|||||||
):
|
):
|
||||||
ble_device = hci0_device_advs["00:00:00:00:00:01"][0]
|
ble_device = hci0_device_advs["00:00:00:00:00:01"][0]
|
||||||
client = bleak.BleakClient(ble_device)
|
client = bleak.BleakClient(ble_device)
|
||||||
assert await client.connect() is False
|
await client.connect()
|
||||||
|
assert client.is_connected is False
|
||||||
assert allocate_slot_mock.call_count == 1
|
assert allocate_slot_mock.call_count == 1
|
||||||
assert release_slot_mock.call_count == 1
|
assert release_slot_mock.call_count == 1
|
||||||
|
|
||||||
@ -335,13 +351,18 @@ async def test_passing_subclassed_str_as_address(
|
|||||||
|
|
||||||
async def connect(self, *args, **kwargs):
|
async def connect(self, *args, **kwargs):
|
||||||
"""Connect."""
|
"""Connect."""
|
||||||
|
|
||||||
|
@property
|
||||||
|
def is_connected(self):
|
||||||
|
"""Connected."""
|
||||||
return True
|
return True
|
||||||
|
|
||||||
with patch(
|
with patch(
|
||||||
"habluetooth.wrappers.get_platform_client_backend_type",
|
"habluetooth.wrappers.get_platform_client_backend_type",
|
||||||
return_value=FakeBleakClient,
|
return_value=FakeBleakClient,
|
||||||
):
|
):
|
||||||
assert await client.connect() is True
|
await client.connect()
|
||||||
|
assert client.is_connected is True
|
||||||
|
|
||||||
cancel_hci0()
|
cancel_hci0()
|
||||||
cancel_hci1()
|
cancel_hci1()
|
||||||
|
Loading…
x
Reference in New Issue
Block a user