Fix linting and adjust tests

DurgNomis-drol · Nov 9, 2023 · 6984d79 · 6984d79
1 parent 296f99a
commit 6984d79
Show file tree

Hide file tree

Showing 2 changed files with 34 additions and 4 deletions.
diff --git a/mytoyota/models/vehicle.py b/mytoyota/models/vehicle.py
@@ -85,7 +85,7 @@ def is_connected_services_enabled(self) -> bool:
             ):
                 vin_specific_connected_service = None
                 for device in self._connected_services["connectedService"]["devices"]:
-                    if (device.get("vin") == self.vin):
+                    if device.get("vin") == self.vin:
                         vin_specific_connected_service = device
                         break
                 if vin_specific_connected_service["state"] == "ACTIVE":

diff --git a/tests/test_vehicle.py b/tests/test_vehicle.py
@@ -92,7 +92,17 @@ def test_vehicle_init(self):
         for veh in vehicle_fixtures:
             vehicle = Vehicle(
                 vehicle_info=veh,
-                connected_services={"connectedService": {"status": "ACTIVE"}},
+                connected_services={
+                    "connectedService": {
+                        "devices": [
+                            {
+                                "brand": "TOYOTA",
+                                "state": "ACTIVE",
+                                "vin": veh.get("vin"),
+                            }
+                        ]
+                    }
+                },
             )
 
             assert vehicle.vin == veh.get("vin")
@@ -134,7 +144,17 @@ def test_vehicle_init_status(self):
 
         vehicle = Vehicle(
             vehicle_info=vehicle_fixtures[0],
-            connected_services={"connectedService": {"status": "ACTIVE"}},
+            connected_services={
+                "connectedService": {
+                    "devices": [
+                        {
+                            "brand": "TOYOTA",
+                            "state": "ACTIVE",
+                            "vin": vehicle_fixtures[0].get("vin"),
+                        }
+                    ]
+                }
+            },
             odometer=odometer_fixture,
             status=status_fixture,
         )
@@ -175,7 +195,17 @@ def test_vehicle_init_status_legacy(self):
 
         vehicle = Vehicle(
             vehicle_info=vehicle_fixtures[0],
-            connected_services={"connectedService": {"status": "ACTIVE"}},
+            connected_services={
+                "connectedService": {
+                    "devices": [
+                        {
+                            "brand": "TOYOTA",
+                            "state": "ACTIVE",
+                            "vin": vehicle_fixtures[0].get("vin"),
+                        }
+                    ]
+                }
+            },
             odometer=odometer_fixture,
             status_legacy=status_fixture,
         )