Include Bluetooth connection slot allocations in connections free mes…

…sage (#1038)

aioesphomeapi/api.proto

@@ -1404,6 +1404,7 @@ message BluetoothConnectionsFreeResponse {
 
   uint32 free = 1;
   uint32 limit = 2;
+  repeated uint64 allocated = 3;
 }
 
 message BluetoothGATTErrorResponse {

aioesphomeapi/client.py

@@ -521,7 +521,8 @@ def subscribe_bluetooth_le_raw_advertisements(
         return partial(self._unsub_bluetooth_advertisements, unsub_callback)
 
     def subscribe_bluetooth_connections_free(
-        self, on_bluetooth_connections_free_update: Callable[[int, int], None]
+        self,
+        on_bluetooth_connections_free_update: Callable[[int, int, list[int]], None],
     ) -> Callable[[], None]:
         return self._get_connection().send_message_callback_response(
             SubscribeBluetoothConnectionsFreeRequest(),

aioesphomeapi/client_callbacks.py

@@ -71,10 +71,10 @@ def on_bluetooth_le_advertising_response(
 
 
 def on_bluetooth_connections_free_response(
-    on_bluetooth_connections_free_update: Callable[[int, int], None],
+    on_bluetooth_connections_free_update: Callable[[int, int, list[int]], None],
     msg: BluetoothConnectionsFreeResponse,
 ) -> None:
-    on_bluetooth_connections_free_update(msg.free, msg.limit)
+    on_bluetooth_connections_free_update(msg.free, msg.limit, list(msg.allocated))
 
 
 def on_bluetooth_gatt_notify_data_response(

tests/test_client.py

@@ -1858,15 +1858,19 @@ async def test_subscribe_bluetooth_connections_free(
     client, connection, transport, protocol = api_client
     connections = []
 
-    def on_bluetooth_connections_free(free: int, limit: int) -> None:
-        connections.append((free, limit))
+    def on_bluetooth_connections_free(
+        free: int, limit: int, allocated: list[int]
+    ) -> None:
+        connections.append((free, limit, allocated))
 
     unsub = client.subscribe_bluetooth_connections_free(on_bluetooth_connections_free)
     await asyncio.sleep(0)
-    response: message.Message = BluetoothConnectionsFreeResponse(free=2, limit=3)
+    response: message.Message = BluetoothConnectionsFreeResponse(
+        free=2, limit=3, allocated=[1234, 5678]
+    )
     mock_data_received(protocol, generate_plaintext_packet(response))
 
-    assert connections == [(2, 3)]
+    assert connections == [(2, 3, [1234, 5678])]
     unsub()