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
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
|
import re
import aiohttp
from aioresponses.compat import normalize_url
from roborock import HomeData, HomeDataScene, UserData
from roborock.web_api import IotLoginInfo, RoborockApiClient
from tests.mock_data import HOME_DATA_RAW, USER_DATA
async def test_pass_login_flow() -> None:
"""Test that we can login with a password and we get back the correct userdata object."""
my_session = aiohttp.ClientSession()
api = RoborockApiClient(username="test_user@gmail.com", session=my_session)
ud = await api.pass_login("password")
assert ud == UserData.from_dict(USER_DATA)
assert not my_session.closed
async def test_code_login_flow() -> None:
"""Test that we can login with a code and we get back the correct userdata object."""
api = RoborockApiClient(username="test_user@gmail.com")
await api.request_code()
ud = await api.code_login(4123)
assert ud == UserData.from_dict(USER_DATA)
async def test_get_home_data_v2():
"""Test a full standard flow where we get the home data to end it off.
This matches what HA does"""
api = RoborockApiClient(username="test_user@gmail.com")
await api.request_code()
ud = await api.code_login(4123)
hd = await api.get_home_data_v2(ud)
assert hd == HomeData.from_dict(HOME_DATA_RAW)
async def test_nc_prepare():
"""Test adding a device and that nothing breaks"""
api = RoborockApiClient(username="test_user@gmail.com")
await api.request_code()
ud = await api.code_login(4123)
prepare = await api.nc_prepare(ud, "America/New_York")
new_device = await api.add_device(ud, prepare["s"], prepare["t"])
assert new_device["duid"] == "rand_duid"
async def test_get_scenes():
"""Test that we can get scenes"""
api = RoborockApiClient(username="test_user@gmail.com")
ud = await api.pass_login("password")
sc = await api.get_scenes(ud, "123456")
assert sc == [
HomeDataScene.from_dict(
{
"id": 1234567,
"name": "My plan",
}
)
]
async def test_execute_scene(mock_rest):
"""Test that we can execute a scene"""
api = RoborockApiClient(username="test_user@gmail.com")
ud = await api.pass_login("password")
await api.execute_scene(ud, 123456)
mock_rest.assert_any_call("https://api-us.roborock.com/user/scene/123456/execute", "post")
async def test_code_login_v4_flow(mock_rest) -> None:
"""Test that we can login with a code and we get back the correct userdata object."""
api = RoborockApiClient(username="test_user@gmail.com")
await api.request_code_v4()
ud = await api.code_login_v4(4123, "US", 1)
assert ud == UserData.from_dict(USER_DATA)
async def test_url_cycling(mock_rest) -> None:
"""Test that we cycle through the URLs correctly."""
# Clear mock rest so that we can override the patches.
mock_rest.clear()
# 1. Mock US URL to return valid status but None for countrycode
mock_rest.post(
re.compile("https://usiot.roborock.com/api/v1/getUrlByEmail.*"),
status=200,
payload={
"code": 200,
"data": {"url": "https://usiot.roborock.com", "country": None, "countrycode": None},
"msg": "Success",
},
)
# 2. Mock EU URL to return valid status but None for countrycode
mock_rest.post(
re.compile("https://euiot.roborock.com/api/v1/getUrlByEmail.*"),
status=200,
payload={
"code": 200,
"data": {"url": "https://euiot.roborock.com", "country": None, "countrycode": None},
"msg": "Success",
},
)
# 3. Mock CN URL to return the correct, valid data
mock_rest.post(
re.compile("https://cniot.roborock.com/api/v1/getUrlByEmail.*"),
status=200,
payload={
"code": 200,
"data": {"url": "https://cniot.roborock.com", "country": "CN", "countrycode": "86"},
"msg": "Success",
},
)
# The RU URL should not be called, but we can mock it just in case
# to catch unexpected behavior.
mock_rest.post(re.compile("https://ruiot.roborock.com/api/v1/getUrlByEmail.*"), status=500)
client = RoborockApiClient("test@example.com")
result = await client._get_iot_login_info()
assert result is not None
assert isinstance(result, IotLoginInfo)
assert result.base_url == "https://cniot.roborock.com"
assert result.country == "CN"
assert result.country_code == "86"
assert client._iot_login_info == result
# Check that all three urls were called. We have to do this kind of weirdly as aioresponses seems to have a bug.
assert (
len(
mock_rest.requests[
(
"post",
normalize_url(
"https://usiot.roborock.com/api/v1/getUrlByEmail?email=test%2540example.com&needtwostepauth=false"
),
)
]
)
== 1
)
assert (
len(
mock_rest.requests[
(
"post",
normalize_url(
"https://euiot.roborock.com/api/v1/getUrlByEmail?email=test%2540example.com&needtwostepauth=false"
),
)
]
)
== 1
)
assert (
len(
mock_rest.requests[
(
"post",
normalize_url(
"https://cniot.roborock.com/api/v1/getUrlByEmail?email=test%2540example.com&needtwostepauth=false"
),
)
]
)
== 1
)
# Make sure we just have the three we tested for above.
assert len(mock_rest.requests) == 3
async def test_missing_country_login(mock_rest) -> None:
"""Test that we cycle through the URLs correctly."""
mock_rest.clear()
# Make country None, but country code set.
mock_rest.post(
re.compile("https://usiot.roborock.com/api/v1/getUrlByEmail.*"),
status=200,
payload={
"code": 200,
"data": {"url": "https://usiot.roborock.com", "country": None, "countrycode": 1},
"msg": "Success",
},
)
# v4 is not mocked, so it would fail it were called.
mock_rest.post(
re.compile(r"https://.*iot\.roborock\.com/api/v1/loginWithCode.*"),
status=200,
payload={"code": 200, "data": USER_DATA, "msg": "success"},
)
mock_rest.post(
re.compile(r"https://.*iot\.roborock\.com/api/v1/sendEmailCode.*"),
status=200,
payload={"code": 200, "data": None, "msg": "success"},
)
client = RoborockApiClient("test@example.com")
await client.request_code_v4()
ud = await client.code_login_v4(4123)
assert ud is not None
# Ensure we have no surprise REST calls.
assert len(mock_rest.requests) == 3
|
