mirror of
https://github.com/flobz/psa_car_controller.git
synced 2026-08-22 09:26:16 +00:00
256 lines
12 KiB
Python
256 lines
12 KiB
Python
# flake8: noqa
|
|
import json
|
|
import os
|
|
import unittest
|
|
from datetime import datetime, timedelta
|
|
from psa_connectedcar import ApiClient
|
|
import psa_connectedcar as psacc
|
|
import reverse_geocode
|
|
from libs.car import Car, Cars
|
|
from libs.charging import Charging
|
|
from libs.elec_price import ElecPrice
|
|
from my_psacc import MyPSACC
|
|
from ecomix import Ecomix
|
|
from libs.car_model import CarModel
|
|
from mylogger import my_logger
|
|
from otp.otp import load_otp, save_otp
|
|
from charge_control import ChargeControls
|
|
from trip import Trips
|
|
from libs.utils import get_temp
|
|
from web.db import Database
|
|
from web.figures import get_figures, get_battery_curve_fig, get_altitude_fig
|
|
import pytz
|
|
from deepdiff import DeepDiff
|
|
|
|
latitude = 47.2183
|
|
longitude = -1.55362
|
|
date3 = datetime.utcnow().replace(2021, 3, 1, 12, 00, 00, 00, tzinfo=pytz.UTC)
|
|
date2 = date3 - timedelta(minutes=20)
|
|
date1 = date3 - timedelta(minutes=40)
|
|
date0 = date3 - timedelta(minutes=60)
|
|
DATA_DIR = os.path.dirname(os.path.realpath(__file__)) + "/data/"
|
|
|
|
|
|
def compare_dict(result, expected):
|
|
diff = DeepDiff(expected, result)
|
|
if diff != {}:
|
|
raise AssertionError(str(diff))
|
|
return True
|
|
|
|
|
|
dummy_value = 0
|
|
|
|
|
|
def callback_test():
|
|
global dummy_value
|
|
dummy_value += 1
|
|
|
|
|
|
class TestUnit(unittest.TestCase):
|
|
def __init__(self, methodName='runTest'):
|
|
super().__init__(methodName)
|
|
self.test_online = os.environ.get("TEST_ONLINE", "0") == "1"
|
|
self.vehicule_list = Cars()
|
|
self.vehicule_list.extend(
|
|
[Car("VR3UHZKX", "vid", "Peugeot"), Car("VXXXXX", "XXXX", "Peugeot", label="SUV 3008")])
|
|
|
|
@staticmethod
|
|
def get_new_test_db():
|
|
try:
|
|
os.remove(DATA_DIR + "tmp.db")
|
|
except:
|
|
pass
|
|
Database.DEFAULT_DB_FILE = DATA_DIR + "tmp.db"
|
|
Database.db_initialized = False
|
|
conn = Database.get_db()
|
|
return conn
|
|
|
|
def test_car(self):
|
|
car1 = Car("VRAAAAAAA", "1sdfdksnfk222", "Peugeot", "208", 46, 0)
|
|
car2 = Car("VR3UHZKX", "1sdfdksnfk222", "Peugeot")
|
|
cars = Cars([car1, car2])
|
|
cars.save_cars(name=DATA_DIR + "test_car.json")
|
|
Cars.load_cars(name=DATA_DIR + "test_car.json")
|
|
|
|
def test_otp_config(self):
|
|
otp_config = load_otp(filename=DATA_DIR + "otp_test.bin")
|
|
assert otp_config is not None
|
|
save_otp(otp_config, filename=DATA_DIR + "otp_test2.bin")
|
|
|
|
def test_mypsacc(self):
|
|
if self.test_online:
|
|
myp = MyPSACC.load_config("config.json")
|
|
myp.refresh_token()
|
|
myp.get_vehicles()
|
|
car = myp.vehicles_list[0]
|
|
myp.abrp.abrp_enable_vin.add(car.vin)
|
|
res = myp.get_vehicle_info(myp.vehicles_list[0].vin)
|
|
myp.abrp.call(car, 22.1)
|
|
myp.save_config()
|
|
assert isinstance(get_temp(str(latitude), str(longitude), myp.weather_api), float)
|
|
|
|
def test_car_model(self):
|
|
assert CarModel.find_model_by_vin("VR3UHZKXZL").name == "e-208"
|
|
assert CarModel.find_model_by_vin("VR3UKZKXZM").name == "e-2008"
|
|
assert CarModel.find_model_by_vin("VXKUHZKXZL").name == "corsa-e"
|
|
|
|
def test_c02_signal_cache(self):
|
|
start = datetime.now() - timedelta(minutes=30)
|
|
end = datetime.now()
|
|
Ecomix._cache = {'FR': [[start - timedelta(days=1), 100],
|
|
[start + timedelta(minutes=1), 10],
|
|
[start + timedelta(minutes=2), 20],
|
|
[start + timedelta(minutes=3), 30]]}
|
|
assert Ecomix.get_co2_from_signal_cache(start, end, "FR") == 20
|
|
|
|
def test_c02_signal(self):
|
|
if self.test_online:
|
|
key = "d186c74bfbcd1da8"
|
|
Ecomix.co2_signal_key = key
|
|
def_country = "FR"
|
|
Ecomix.get_data_from_co2_signal(latitude, longitude, def_country)
|
|
res = Ecomix.get_co2_from_signal_cache(datetime.now() - timedelta(minutes=5), datetime.now(), def_country)
|
|
assert isinstance(res, float)
|
|
|
|
def test_charge_control(self):
|
|
charge_control = ChargeControls()
|
|
charge_control.file_name = "test_charge_control.json"
|
|
charge_control.save_config(force=True)
|
|
|
|
def test_battery_curve(self):
|
|
from libs.car import Car
|
|
from libs.charging import Charging
|
|
try:
|
|
os.remove("tmp.db")
|
|
except:
|
|
pass
|
|
Database.DEFAULT_DB_FILE = "tmp.db"
|
|
conn = Database.get_db()
|
|
list(map(dict, conn.execute('PRAGMA database_list').fetchall()))
|
|
vin = "VR3UHZKXZL"
|
|
car = Car(vin, "id", "Peugeot")
|
|
Charging.record_charging(car, "InProgress", date0, 50, latitude, longitude, "FR", "slow")
|
|
Charging.record_charging(car, "InProgress", date1, 75, latitude, longitude, "FR", "slow")
|
|
Charging.record_charging(car, "InProgress", date2, 85, latitude, longitude, "FR", "slow")
|
|
Charging.record_charging(car, "InProgress", date3, 90, latitude, longitude, "FR", "slow")
|
|
|
|
res = Database.get_battery_curve(Database.get_db(), date0, vin)
|
|
assert len(res) == 3
|
|
|
|
def test_sdk(self):
|
|
|
|
res = {
|
|
'lastPosition': {'type': 'Feature', 'geometry': {'type': 'Point', 'coordinates': [9.65457, 49.96119, 21]},
|
|
'properties': {'updatedAt': '2021-03-29T05:16:10Z', 'heading': 126,
|
|
'type': 'Estimated'}}, 'preconditionning': {
|
|
'airConditioning': {'updatedAt': '2021-04-01T16:17:01Z', 'status': 'Disabled', 'programs': [
|
|
{'enabled': False, 'slot': 1, 'recurrence': 'Daily', 'start': 'PT21H40M',
|
|
'occurence': {'day': ['Sat']}}]}},
|
|
'energy': [{'updatedAt': '2021-02-23T22:29:03Z', 'type': 'Fuel', 'level': 0},
|
|
{'updatedAt': '2021-04-01T16:17:01Z', 'type': 'Electric', 'level': 70, 'autonomy': 192,
|
|
'charging': {'plugged': False, 'status': 'Disconnected', 'remainingTime': 'PT0S',
|
|
'chargingRate': 0, 'chargingMode': 'No', 'nextDelayedTime': 'PT21H30M'}}],
|
|
'createdAt': '2021-04-01T16:17:01Z',
|
|
'battery': {'voltage': 99, 'current': 0, 'createdAt': '2021-04-01T16:17:01Z'},
|
|
'kinetic': {'createdAt': '2021-03-29T05:16:10Z', 'moving': False},
|
|
'privacy': {'createdAt': '2021-04-01T16:17:01Z', 'state': 'None'},
|
|
'service': {'type': 'Electric', 'updatedAt': '2021-02-23T21:10:29Z'}, '_links': {'self': {
|
|
'href': 'https://api.groupe-psa.com/connectedcar/v4/user/vehicles/myid/status'},
|
|
'vehicles': {
|
|
'href': 'https://api.groupe-psa.com/connectedcar/v4/user/vehicles/myid'}},
|
|
'timed.odometer': {'createdAt': None, 'mileage': 1107.1}, 'updatedAt': '2021-04-01T16:17:01Z'}
|
|
api = ApiClient()
|
|
status: psacc.models.status.Status = api._ApiClient__deserialize(res, "Status")
|
|
geocode_res = reverse_geocode.search([(status.last_position.geometry.coordinates[:2])[::-1]])[0]
|
|
assert geocode_res["country_code"] == "DE"
|
|
TestUnit.get_new_test_db()
|
|
car = Car("XX", "vid", "Peugeot")
|
|
car.status = status
|
|
myp = MyPSACC.load_config(DATA_DIR + "config.json")
|
|
myp.record_info(car)
|
|
assert "features" in json.loads(Database.get_recorded_position())
|
|
# electric should be first
|
|
assert car.status.energy[0].type == 'Electric'
|
|
|
|
def test_record_position_charging(self):
|
|
TestUnit.get_new_test_db()
|
|
ElecPrice.CONFIG_FILENAME = DATA_DIR + "config.ini"
|
|
car = self.vehicule_list[0]
|
|
Database.record_position(None, car.vin, 11, latitude, longitude - 0.05, None, date0, 40, None, False)
|
|
Database.record_position(None, car.vin, 20, latitude, longitude, 32, date1, 35, None, False)
|
|
Database.record_position(None, car.vin, 30, latitude, longitude, 42, date2, 30, None, False)
|
|
Database.add_altitude_to_db(Database.get_db())
|
|
data = json.loads(Database.get_recorded_position())
|
|
assert data["features"][1]["geometry"]["coordinates"] == [float(longitude), float(latitude)]
|
|
trips = Trips.get_trips(self.vehicule_list)[car.vin]
|
|
trip = trips[0]
|
|
map(trip.add_temperature, [10, 13, 15])
|
|
res = trip.get_info()
|
|
assert compare_dict(res, {'consumption_km': 24.21052631578947,
|
|
'start_at': date0,
|
|
'consumption_by_temp': None,
|
|
'positions': {'lat': [latitude], 'long': [longitude]},
|
|
'duration': 40.0, 'speed_average': 28.5, 'distance': 19.0, 'mileage': 30.0,
|
|
'altitude_diff': 2, 'id': 1, 'consumption': 4.6})
|
|
|
|
Charging.elec_price = ElecPrice.read_config()
|
|
start_level = 40
|
|
end_level = 85
|
|
Charging.record_charging(car, "InProgress", date0, start_level, latitude, longitude, None, "slow")
|
|
Charging.record_charging(car, "InProgress", date1, 70, latitude, longitude, "FR", "slow")
|
|
Charging.record_charging(car, "InProgress", date1, 70, latitude, longitude, "FR", "slow")
|
|
Charging.record_charging(car, "InProgress", date2, 80, latitude, longitude, "FR", "slow")
|
|
Charging.record_charging(car, "Stopped", date3, end_level, latitude, longitude, "FR", "slow")
|
|
chargings = Charging.get_chargings()
|
|
co2 = chargings[0]["co2"]
|
|
assert isinstance(co2, float)
|
|
assert compare_dict(chargings, [{'start_at': date0,
|
|
'stop_at': date3,
|
|
'VIN': 'VR3UHZKX',
|
|
'start_level': 40,
|
|
'end_level': 85,
|
|
'co2': co2,
|
|
'kw': 20.7,
|
|
'price': 3.84,
|
|
'charging_mode': 'slow'}])
|
|
assert get_figures(car)
|
|
row = {"start_at": date0.strftime("%Y-%m-%dT%H:%M:%S+00:00"),
|
|
"stop_at": date3.strftime("%Y-%m-%dT%H:%M:%S+00:00"), "start_level": start_level, "end_level": end_level}
|
|
assert get_battery_curve_fig(row, car) is not None
|
|
assert get_altitude_fig(trip) is not None
|
|
|
|
def test_fuel_car(self):
|
|
TestUnit.get_new_test_db()
|
|
ElecPrice.CONFIG_FILENAME = DATA_DIR + "config.ini"
|
|
car = self.vehicule_list[1]
|
|
Database.record_position(None, car.vin, 11, latitude, longitude, 22, date0, 40, 30, False)
|
|
Database.record_position(None, car.vin, 20, latitude, longitude, 22, date1, 35, 29, False)
|
|
Database.record_position(None, car.vin, 30, latitude, longitude, 22, date2, 30, 28, False)
|
|
trips = Trips.get_trips(self.vehicule_list)
|
|
res = trips[car.vin].get_trips_as_dict()
|
|
assert compare_dict(res, [{'consumption_km': 5.684210526315789,
|
|
'start_at': date0,
|
|
'consumption_by_temp': None,
|
|
'positions': {'lat': [latitude],
|
|
'long': [longitude]},
|
|
'duration': 40.0,
|
|
'speed_average': 28.5,
|
|
'distance': 19.0,
|
|
'mileage': 30.0,
|
|
'altitude_diff': 0,
|
|
'id': 1,
|
|
'consumption': 1.08,
|
|
'consumption_fuel_km': 10.53}])
|
|
|
|
def test_db_callback(self):
|
|
old_dummy_value = dummy_value
|
|
TestUnit.get_new_test_db()
|
|
Database.set_db_callback(callback_test)
|
|
assert old_dummy_value == dummy_value
|
|
Database.record_position(None, "xx", 11, latitude, longitude - 0.05, None, date0, 40, None, False)
|
|
assert old_dummy_value != dummy_value
|
|
|
|
if __name__ == '__main__':
|
|
my_logger(handler_level=os.environ.get("DEBUG_LEVEL", 20))
|
|
unittest.main()
|