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
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
|
import asyncio
from unittest.mock import AsyncMock, patch
import pytest
from volvooncall import Connection
def mocked_request(method, url, rel=None, **kwargs):
if "customeraccounts" in url:
return {"username": "foobar", "accountVehicleRelations": ["rel/1"]}
if "rel" in url:
return {"vehicle": "vehicle/1", "status": "Verified"}
if "attributes" in url:
return {"registrationNumber": "FOO123"}
if "status" in url:
return {
"engineRunning": False,
"engineStartSupported": True,
"ERS": {"status": "off"},
}
if "position" in url:
return {}
if "engine/start" in url:
return {"service": "engine/start", "status": "Started"}
return {"error": "Unauthorized"}
@patch(
"volvooncall.Connection._request",
new_callable=AsyncMock,
side_effect=mocked_request,
)
async def get_vehicle(mock):
connection = Connection(session=None, username="", password="")
await connection.update()
assert mock.called
return next(connection.vehicles, None)
@pytest.mark.asyncio
async def test_basic():
vehicle = await get_vehicle()
assert vehicle is not None
assert vehicle.registration_number == "FOO123"
@pytest.mark.asyncio
async def test_engine():
vehicle = await get_vehicle()
assert not vehicle.is_engine_running
@pytest.mark.asyncio
async def test_ers():
vehicle = await get_vehicle()
dashboard = vehicle.dashboard()
engine_instruments = [
instrument
for instrument in dashboard.instruments
if instrument.attr == "is_engine_running"
]
# a binary sensor and a switch should be present
assert len(engine_instruments) == 2
# should be off
assert all(not engine.state for engine in engine_instruments)
async def get_started_vehicle():
def mocked_request_ers(method, url, rel=None, **kwargs):
if "status" in url:
return {
"engineRunning": False,
"engineStartSupported": True,
"ERS": {"status": "on"},
}
return mocked_request(method, url, rel, **kwargs)
vehicle = await get_vehicle()
with patch(
"volvooncall.Connection._request",
new_callable=AsyncMock,
side_effect=mocked_request_ers,
) as mock:
await vehicle.start_engine()
assert mock.called
return vehicle
@pytest.mark.asyncio
async def test_ers_start():
vehicle = await get_started_vehicle()
assert vehicle.is_engine_running
@pytest.mark.asyncio
async def test_ers_start_dashboard():
vehicle = await get_started_vehicle()
dashboard = vehicle.dashboard()
engine_instruments = [
instrument
for instrument in dashboard.instruments
if instrument.attr == "is_engine_running"
]
# a binary sensor and a switch should be present
assert len(engine_instruments) == 2
# shold be on
assert all(engine.state for engine in engine_instruments)
if __name__ == "__main__":
loop = asyncio.get_event_loop()
loop.run_until_complete(test_basic())
|
