Add test baseline (pytest), energy networking-status helper, and regen protobufs

AGENTS.md

@@ -14,9 +14,13 @@ uv run pyright tesla_fleet_api
 # Linting
 uv run ruff check tesla_fleet_api
 uv run ruff format tesla_fleet_api
+
+# Tests
+uv run pytest tests
 ```
 
-No test suite exists in this repo.
+Tests live in `tests/` and use `unittest.IsolatedAsyncioTestCase` (collected and
+run natively by pytest — `pytest-asyncio` is not required).
 
 ## API References
 

pyproject.toml

@@ -29,7 +29,7 @@ dependencies = [
 "Homepage" = "https://github.com/Teslemetry/python-tesla-fleet-api"
 
 [dependency-groups]
-dev = ["pyright>=1.1.399"]
+dev = ["pyright>=1.1.399", "pytest>=8.0"]
 
 [tool.setuptools.packages.find]
 include = ["tesla_fleet_api*"]
@@ -44,3 +44,6 @@ exclude = ["tesla_fleet_api/tesla/vehicle/proto/*"]
 include = ["tesla_fleet_api"]
 typeCheckingMode = "strict"
 exclude = ["tesla_fleet_api/tesla/vehicle/proto/*"]
+
+[tool.pytest.ini_options]
+testpaths = ["tests"]

tesla_fleet_api.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: tesla_fleet_api
-Version: 1.4.6
+Version: 1.4.7
 Summary: Tesla Fleet API library for Python
 Author-email: Brett Adams <hello@teslemetry.com>
 License-Expression: Apache-2.0

tesla_fleet_api/tesla/energysite.py

@@ -1,15 +1,17 @@
 from __future__ import annotations
+
 import base64
-from typing import Any, TYPE_CHECKING
+from typing import TYPE_CHECKING, Any
+
 from tesla_fleet_api.const import (
-    Method,
-    EnergyOperationMode,
+    AuthorizedClientKeyType,
+    AuthorizedClientType,
+    EnergyDeviceIdentifierType,
     EnergyExportMode,
     EnergyIslandMode,
+    EnergyOperationMode,
+    Method,
     TeslaEnergyPeriod,
-    EnergyDeviceIdentifierType,
-    AuthorizedClientKeyType,
-    AuthorizedClientType,
 )
 
 if TYPE_CHECKING:
@@ -53,9 +55,21 @@ async def get_system_info(self) -> dict[str, Any]:
         """Get energy device system information including firmware version, device type, part number, serial number, and DIN."""
         return await self._command("common", "get_system_info_request")
 
+    async def raw_networking_status(self) -> dict[str, Any]:
+        """Get raw energy device networking status including WiFi, Ethernet, and cellular connectivity."""
+        return await self._command("common", "get_networking_status_request")
+
     async def get_networking_status(self) -> dict[str, Any]:
         """Get energy device networking status including WiFi, Ethernet, and cellular connectivity."""
-        return await self._command("common", "get_networking_status_request")
+        result = await self.raw_networking_status()
+        return (
+            result.get("response", {})
+            .get("message", {})
+            .get("Payload", {})
+            .get("Common", {})
+            .get("Message", {})
+            .get("GetNetworkingStatusResponse", {})
+        )
 
     async def wifi_scan(self) -> dict[str, Any]:
         """Scan for available WiFi networks from the energy gateway."""
@@ -160,7 +174,11 @@ async def schedule_backup_event(
             priority: Priority level for the backup event.
         """
         params: dict[str, Any] = {}
-        if start_time is not None or duration_seconds is not None or priority is not None:
+        if (
+            start_time is not None
+            or duration_seconds is not None
+            or priority is not None
+        ):
             scheduling_info: dict[str, Any] = {}
             if start_time is not None:
                 scheduling_info["start_time"] = start_time

tesla_fleet_api/tesla/vehicle/proto/common_pb2.pyi

@@ -3,6 +3,7 @@ from google.protobuf import descriptor as _descriptor
 from google.protobuf import message as _message
 from collections.abc import Mapping as _Mapping
 from typing import ClassVar as _ClassVar, Optional as _Optional, Union as _Union
+
 DESCRIPTOR: _descriptor.FileDescriptor
 
 class Invalid(int, metaclass=_enum_type_wrapper.EnumTypeWrapper):
@@ -32,22 +33,18 @@ StwHeatLevel_High: StwHeatLevel
 
 class Void(_message.Message):
     __slots__ = ()
-
-    def __init__(self) -> None:
-        ...
+    def __init__(self) -> None: ...
 
 class LatLong(_message.Message):
-    __slots__ = ('latitude', 'longitude')
+    __slots__ = ("latitude", "longitude")
     LATITUDE_FIELD_NUMBER: _ClassVar[int]
     LONGITUDE_FIELD_NUMBER: _ClassVar[int]
     latitude: float
     longitude: float
-
-    def __init__(self, latitude: _Optional[float]=..., longitude: _Optional[float]=...) -> None:
-        ...
+    def __init__(self, latitude: _Optional[float] = ..., longitude: _Optional[float] = ...) -> None: ...
 
 class ChargePortLatchState(_message.Message):
-    __slots__ = ('SNA', 'Disengaged', 'Engaged', 'Blocking')
+    __slots__ = ("SNA", "Disengaged", "Engaged", "Blocking")
     SNA_FIELD_NUMBER: _ClassVar[int]
     DISENGAGED_FIELD_NUMBER: _ClassVar[int]
     ENGAGED_FIELD_NUMBER: _ClassVar[int]
@@ -56,32 +53,26 @@ class ChargePortLatchState(_message.Message):
     Disengaged: Void
     Engaged: Void
     Blocking: Void
-
-    def __init__(self, SNA: _Optional[_Union[Void, _Mapping]]=..., Disengaged: _Optional[_Union[Void, _Mapping]]=..., Engaged: _Optional[_Union[Void, _Mapping]]=..., Blocking: _Optional[_Union[Void, _Mapping]]=...) -> None:
-        ...
+    def __init__(self, SNA: _Optional[_Union[Void, _Mapping]] = ..., Disengaged: _Optional[_Union[Void, _Mapping]] = ..., Engaged: _Optional[_Union[Void, _Mapping]] = ..., Blocking: _Optional[_Union[Void, _Mapping]] = ...) -> None: ...
 
 class PreconditioningTimes(_message.Message):
-    __slots__ = ('all_week', 'weekdays')
+    __slots__ = ("all_week", "weekdays")
     ALL_WEEK_FIELD_NUMBER: _ClassVar[int]
     WEEKDAYS_FIELD_NUMBER: _ClassVar[int]
     all_week: Void
     weekdays: Void
-
-    def __init__(self, all_week: _Optional[_Union[Void, _Mapping]]=..., weekdays: _Optional[_Union[Void, _Mapping]]=...) -> None:
-        ...
+    def __init__(self, all_week: _Optional[_Union[Void, _Mapping]] = ..., weekdays: _Optional[_Union[Void, _Mapping]] = ...) -> None: ...
 
 class OffPeakChargingTimes(_message.Message):
-    __slots__ = ('all_week', 'weekdays')
+    __slots__ = ("all_week", "weekdays")
     ALL_WEEK_FIELD_NUMBER: _ClassVar[int]
     WEEKDAYS_FIELD_NUMBER: _ClassVar[int]
     all_week: Void
     weekdays: Void
-
-    def __init__(self, all_week: _Optional[_Union[Void, _Mapping]]=..., weekdays: _Optional[_Union[Void, _Mapping]]=...) -> None:
-        ...
+    def __init__(self, all_week: _Optional[_Union[Void, _Mapping]] = ..., weekdays: _Optional[_Union[Void, _Mapping]] = ...) -> None: ...
 
 class ChargeSchedule(_message.Message):
-    __slots__ = ('id', 'name', 'days_of_week', 'start_enabled', 'start_time', 'end_enabled', 'end_time', 'one_time', 'enabled', 'latitude', 'longitude')
+    __slots__ = ("id", "name", "days_of_week", "start_enabled", "start_time", "end_enabled", "end_time", "one_time", "enabled", "latitude", "longitude")
     ID_FIELD_NUMBER: _ClassVar[int]
     NAME_FIELD_NUMBER: _ClassVar[int]
     DAYS_OF_WEEK_FIELD_NUMBER: _ClassVar[int]
@@ -104,12 +95,10 @@ class ChargeSchedule(_message.Message):
     enabled: bool
     latitude: float
     longitude: float
-
-    def __init__(self, id: _Optional[int]=..., name: _Optional[str]=..., days_of_week: _Optional[int]=..., start_enabled: bool=..., start_time: _Optional[int]=..., end_enabled: bool=..., end_time: _Optional[int]=..., one_time: bool=..., enabled: bool=..., latitude: _Optional[float]=..., longitude: _Optional[float]=...) -> None:
-        ...
+    def __init__(self, id: _Optional[int] = ..., name: _Optional[str] = ..., days_of_week: _Optional[int] = ..., start_enabled: _Optional[bool] = ..., start_time: _Optional[int] = ..., end_enabled: _Optional[bool] = ..., end_time: _Optional[int] = ..., one_time: _Optional[bool] = ..., enabled: _Optional[bool] = ..., latitude: _Optional[float] = ..., longitude: _Optional[float] = ...) -> None: ...
 
 class PreconditionSchedule(_message.Message):
-    __slots__ = ('id', 'name', 'days_of_week', 'precondition_time', 'one_time', 'enabled', 'latitude', 'longitude')
+    __slots__ = ("id", "name", "days_of_week", "precondition_time", "one_time", "enabled", "latitude", "longitude")
     ID_FIELD_NUMBER: _ClassVar[int]
     NAME_FIELD_NUMBER: _ClassVar[int]
     DAYS_OF_WEEK_FIELD_NUMBER: _ClassVar[int]
@@ -126,6 +115,4 @@ class PreconditionSchedule(_message.Message):
     enabled: bool
     latitude: float
     longitude: float
-
-    def __init__(self, id: _Optional[int]=..., name: _Optional[str]=..., days_of_week: _Optional[int]=..., precondition_time: _Optional[int]=..., one_time: bool=..., enabled: bool=..., latitude: _Optional[float]=..., longitude: _Optional[float]=...) -> None:
-        ...
+    def __init__(self, id: _Optional[int] = ..., name: _Optional[str] = ..., days_of_week: _Optional[int] = ..., precondition_time: _Optional[int] = ..., one_time: _Optional[bool] = ..., enabled: _Optional[bool] = ..., latitude: _Optional[float] = ..., longitude: _Optional[float] = ...) -> None: ...