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
|
"""Test for Device model."""
import json
import pytest
from syrupy.assertion import SnapshotAssertion
from homewizard_energy.models import Device
from . import load_fixtures
pytestmark = [pytest.mark.asyncio]
@pytest.mark.parametrize(
("model", "fixtures"),
[
("HWE-P1", ["device"]),
("HWE-SKT", ["device"]),
("HWE-WTR", ["device"]),
("HWE-KWH1", ["device"]),
("HWE-KWH3", ["device"]),
("SDM230-wifi", ["device"]),
("SDM630-wifi", ["device"]),
],
)
async def test_device(model: str, fixtures: str, snapshot: SnapshotAssertion):
"""Test Device model."""
for fixture in fixtures:
data = Device.from_dict(json.loads(load_fixtures(f"{model}/{fixture}.json")))
assert data
assert snapshot == data
# pylint: disable=too-many-arguments,too-many-positional-arguments
@pytest.mark.parametrize(
(
"model",
"supports_state",
"supports_identify",
"supports_cloud_enable",
"supports_reboot",
"supports_telegram",
),
[
("HWE-P1", False, True, True, False, True),
("HWE-SKT", True, True, True, False, False),
("HWE-WTR", False, True, True, False, False),
("HWE-KWH1", False, False, True, False, False),
("HWE-KWH3", False, False, True, False, False),
("SDM230-wifi", False, False, True, False, False),
("SDM630-wifi", False, False, True, False, False),
],
)
async def test_device_support_functions(
model: str,
supports_state: bool,
supports_identify: bool,
supports_cloud_enable: bool,
supports_reboot: bool,
supports_telegram: bool,
):
"""Test Device model support functions."""
device = Device.from_dict(json.loads(load_fixtures(f"{model}/device.json")))
assert device
assert device.supports_state() == supports_state
assert device.supports_identify() == supports_identify
assert device.supports_cloud_enable() == supports_cloud_enable
assert device.supports_reboot() == supports_reboot
assert device.supports_telegram() == supports_telegram
|
