feat(ble): return decoded response payload from infotainment commands

Author: Bre77

tesla_fleet_api/tesla/vehicle/commands.py

@@ -849,15 +849,26 @@ async def _command(
                     }
                 if response.HasField("vehicleData"):
                     return {"response": response.vehicleData}
+                # Any other populated response_msg oneof member (e.g.
+                # getManagedChargingSitesResponse) is returned as the decoded
+                # proto under its field name, alongside result/reason, so an
+                # empty payload is distinguishable from an undecoded one.
+                payload_field = response.WhichOneof("response_msg")
+                data: dict[str, Any] = {}
+                if payload_field is not None:
+                    data[payload_field] = getattr(response, payload_field)
                 if response.HasField("actionStatus"):
                     return {
                         "response": {
                             "result": response.actionStatus.result
                             == OperationStatus_E.OPERATIONSTATUS_OK,
                             "reason": response.actionStatus.result_reason.plain_text
                             or "",
+                            **data,
                         }
                     }
+                if data:
+                    return {"response": {"result": True, "reason": "", **data}}
 
         return {"response": {"result": True, "reason": ""}}
 

tests/test_ble_charging_utility_commands.py

@@ -5,8 +5,18 @@
 parallel flattened API - see ``commands.py`` docstrings.
 """
 
-from tesla_protocol.command.car_server_pb2 import Action, SetRateTariffRequest
-from tesla_protocol.command.universal_message_pb2 import Domain
+from tesla_protocol.command.car_server_pb2 import (
+    Action,
+    GetManagedChargingSitesResponse,
+    ManagedChargingSite,
+    Response,
+    SetRateTariffRequest,
+)
+from tesla_protocol.command.universal_message_pb2 import (
+    Destination,
+    Domain,
+    RoutableMessage,
+)
 
 from ble_mocked_transport import (
     MockedBleTransportTestCase,
@@ -117,3 +127,35 @@ async def test_sends_discharge_limit(self) -> None:
 
         vehicle_action = _decode_vehicle_action(vehicle, send.await_args.args[0])
         self.assertEqual(vehicle_action.setDischargeLimitAction.discharge_limit, 50)
+
+
+class GetManagedChargingSitesDecodeTests(MockedBleTransportTestCase):
+    def _reply(self, sites: list[ManagedChargingSite]) -> RoutableMessage:
+        body = Response(
+            getManagedChargingSitesResponse=GetManagedChargingSitesResponse(sites=sites)
+        )
+        return RoutableMessage(
+            from_destination=Destination(domain=Domain.DOMAIN_INFOTAINMENT),
+            protobuf_message_as_bytes=body.SerializeToString(),
+        )
+
+    async def test_returns_decoded_sites(self) -> None:
+        vehicle, send = self.make_vehicle()
+        send.return_value = self._reply([ManagedChargingSite(public_key="k1")])
+
+        result = await vehicle.get_managed_charging_sites()
+
+        resp = result["response"]
+        self.assertTrue(resp["result"])
+        sites = resp["getManagedChargingSitesResponse"].sites
+        self.assertEqual([s.public_key for s in sites], ["k1"])
+
+    async def test_empty_list_is_distinguishable_from_undecoded(self) -> None:
+        vehicle, send = self.make_vehicle()
+        send.return_value = self._reply([])
+
+        result = await vehicle.get_managed_charging_sites()
+
+        resp = result["response"]
+        self.assertIn("getManagedChargingSitesResponse", resp)
+        self.assertEqual(len(resp["getManagedChargingSitesResponse"].sites), 0)