From ae885ea35ffb4a7ba1f76ea5cd09431f821dfbd5 Mon Sep 17 00:00:00 2001
From: "J. Nick Koston" <nick@koston.org>
Date: Sun, 26 Nov 2023 14:41:57 -0600
Subject: [PATCH] fix test with cython

---
 tests/test_client.py | 5 ++---
 1 file changed, 2 insertions(+), 3 deletions(-)

diff --git a/tests/test_client.py b/tests/test_client.py
index cb602c5..cd75e98 100644
--- a/tests/test_client.py
+++ b/tests/test_client.py
@@ -1487,7 +1487,6 @@ async def test_bluetooth_device_connect(
 ) -> None:
     """Test bluetooth_device_connect."""
     client, connection, transport, protocol = api_client
-    send = client._connection.send_messages = MagicMock()
     states = []
 
     def on_bluetooth_connection_state(connected: bool, mtu: int, error: int) -> None:
@@ -1512,8 +1511,8 @@ async def test_bluetooth_device_connect(
 
     await connect_task
     assert states == [(True, 23, 0)]
-    send.assert_called_once_with(
-        (
+    transport.write.assert_called_once_with(
+        generate_plaintext_packet(
             BluetoothDeviceRequest(
                 address=1234,
                 request_type=method,