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
|
import asyncio
from pymystrom.switch import MyStromSwitch
import pymystrom.switch as switch_module
async def _fake_request(
self, uri, method="GET", data=None, json_data=None, params=None
):
uri_str = str(uri)
if uri_str.endswith("/report"):
return {"relay": True, "power": 1.23, "Ws": 0.5}
if uri_str.endswith("/api/v1/info"):
# Legacy v1 firmware without 'type'
return {"version": "2.68.10", "mac": "AA:BB:CC:DD:EE:FF"}
if uri_str.endswith("/info.json"):
return {"version": "2.68.10", "mac": "AA:BB:CC:DD:EE:FF"}
return {}
def test_get_state_missing_type():
# Patch the request function used by MyStromSwitch
original_request = switch_module.request
switch_module.request = _fake_request
try:
sw = MyStromSwitch("127.0.0.1")
asyncio.run(sw.get_state())
assert sw.relay is True
assert sw.consumption == 1.2
assert sw.consumedWs == 0.5
assert sw.firmware == "2.68.10"
assert sw.mac == "AA:BB:CC:DD:EE:FF"
# Missing 'type' should be tolerated and map to None
assert sw.device_type is None
finally:
# Restore original request function
switch_module.request = original_request
|
