41 lines
1.4 KiB
Python
41 lines
1.4 KiB
Python
|
import unittest
|
||
|
|
import sys
|
||
|
|
from pathlib import Path
|
||
|
|
|
||
|
|
# Allow importing scripts/tesla.py as a module
|
||
|
|
ROOT = Path(__file__).resolve().parents[1]
|
||
|
|
sys.path.insert(0, str(ROOT / "scripts"))
|
||
|
|
|
||
|
|
import tesla # noqa: E402
|
||
|
|
|
||
|
|
|
||
|
|
class ChargePortStatusTests(unittest.TestCase):
|
||
|
|
def test_charge_port_status_json_shape(self):
|
||
|
|
vehicle = {"display_name": "Test Car", "state": "online"}
|
||
|
|
data = {
|
||
|
|
"charge_state": {
|
||
|
|
"charge_port_door_open": True,
|
||
|
|
"charge_port_latch": "Engaged",
|
||
|
|
"conn_charge_cable": "SAE",
|
||
|
|
"charging_state": "Charging",
|
||
|
|
# extra fields should be ignored
|
||
|
|
"battery_level": 50,
|
||
|
|
},
|
||
|
|
# drive_state/location must not be pulled into the object
|
||
|
|
"drive_state": {"latitude": 1.23, "longitude": 4.56},
|
||
|
|
}
|
||
|
|
|
||
|
|
out = tesla._charge_port_status_json(vehicle, data)
|
||
|
|
self.assertEqual(out["display_name"], "Test Car")
|
||
|
|
self.assertEqual(out["state"], "online")
|
||
|
|
self.assertEqual(out["charge_port_door_open"], True)
|
||
|
|
self.assertEqual(out["charge_port_latch"], "Engaged")
|
||
|
|
self.assertEqual(out["conn_charge_cable"], "SAE")
|
||
|
|
self.assertEqual(out["charging_state"], "Charging")
|
||
|
|
self.assertNotIn("drive_state", out)
|
||
|
|
self.assertNotIn("latitude", out)
|
||
|
|
|
||
|
|
|
||
|
|
if __name__ == "__main__":
|
||
|
|
unittest.main()
|