forked from flobz/psa_car_controller
-
Notifications
You must be signed in to change notification settings - Fork 0
/
test_psa.py
62 lines (52 loc) · 3.2 KB
/
test_psa.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
from unittest.mock import MagicMock, patch
from psa_car_controller.psa.connected_car_api import Vehicles, ApiClient
from psa_car_controller.psa.constants import DISCONNECTED
from psa_car_controller.psacc.model.car import Car
from tests.data.car_status import ELECTRIC_CAR_STATUS
import unittest
from paho.mqtt.client import MQTTMessage
from psa_car_controller.psa.RemoteClient import MQTT_EVENT_TOPIC
from tests.utils import get_rc
message_without_precond = b'{"date":"2022-03-30T12:00:52Z","etat_res_elec":0,"precond_state":{},"charging_state":{"program":{' \
b'"hour":22,"minute":30},"available":1,"remaining_time":0,"rate":0,"cable_detected":1,"soc_batt":76,' \
b'"autonomy_zev":178,"type":0,"hmi_state":0,"mode":2},"stolen_state":0,"vin":"vin",' \
b'"reason":0,"signal_quality":3,"sev_stop_date":"2022-03-30T11:10:25Z","fds":[],"sev_state":0,' \
b'"obj_counter":1,"privacy_customer":0,"privacy_applicable":0,"privacy_applicable_max":2,' \
b'"superlock_state":0} '
message_without_charge_info = b'{"date":"2022-03-30T13:18:56Z","etat_res_elec":5,"precond_state":{"available":1,' \
b'"programs":{"program1":{"hour":34,"minute":7,"on":0,"day":[0,0,0,0,0,0,0]},' \
b'"program2":{"hour":34,"minute":7,"on":0,"day":[0,0,0,0,0,0,0]},"program3":{' \
b'"hour":34,"minute":7,"on":0,"day":[0,0,0,0,0,0,0]},"program4":{"hour":34,"minute":7,' \
b'"on":0,"day":[0,0,0,0,0,0,0]}},"asap":0,"status":0,"aff":1},"charging_state":{' \
b'"program":{"hour":22,"minute":30},"available":1,"rate":0,"cable_detected":1,' \
b'"soc_batt":61,"type":0,"aff":1,"hmi_state":0,"mode":2},"stolen_state":0,' \
b'"vin":"VIN","reason":4,"signal_quality":5,' \
b'"sev_stop_date":"2022-03-30T12:41:10Z","fds":["NDR01","NBM01","NCG01","NAO01",' \
b'"NAS01"],"sev_state":1,"obj_counter":2,"privacy_customer":0,"privacy_applicable":0,' \
b'"privacy_applicable_max":2,"superlock_state":0} '
class TestUnit(unittest.TestCase):
@patch('time.sleep', return_value=None)
def test_fix_not_updated_api(self, patched_time_sleep):
# GIVEN
remote_client = get_rc()
vin = "myvin"
car = Car("a", "b", "c")
car.status = ApiClient()._ApiClient__deserialize(ELECTRIC_CAR_STATUS, "Status")
car.status.get_energy('Electric').charging.status = DISCONNECTED
remote_client.vehicles_list.get_car_by_vin = MagicMock(return_value=car)
remote_client.wakeup = MagicMock()
# WHEN
remote_client._fix_not_updated_api({'remaining_time': 1}, vin)
# THEN
remote_client.wakeup.assert_called_once_with(vin)
def test_message_without_precond(self):
remote_client = get_rc()
msg = MQTTMessage(topic=MQTT_EVENT_TOPIC.encode("utf-8"))
msg.payload = message_without_precond
remote_client._on_mqtt_message(None, None, msg)
self.assertEqual(remote_client.precond_programs, {})
def test_message_without_charge_info(self):
remote_client = get_rc()
msg = MQTTMessage(topic=MQTT_EVENT_TOPIC.encode("utf-8"))
msg.payload = message_without_charge_info
remote_client._on_mqtt_message(None, None, msg)