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
|
from unittest import TestCase
import pytest
from miio import WaterPurifier
from miio.waterpurifier import WaterPurifierStatus
from .dummies import DummyDevice
class DummyWaterPurifier(DummyDevice, WaterPurifier):
def __init__(self, *args, **kwargs):
self.state = {
"power": "on",
"mode": "unknown",
"tds": "unknown",
"filter1_life": -1,
"filter1_state": -1,
"filter_life": -1,
"filter_state": -1,
"life": -1,
"state": -1,
"level": "unknown",
"volume": "unknown",
"filter": "unknown",
"usage": "unknown",
"temperature": "unknown",
"uv_life": -1,
"uv_state": -1,
"elecval_state": "unknown",
}
self.return_values = {
"get_prop": self._get_state,
"set_power": lambda x: self._set_state("power", x),
}
super().__init__(args, kwargs)
@pytest.fixture(scope="class")
def waterpurifier(request):
request.cls.device = DummyWaterPurifier()
# TODO add ability to test on a real device
@pytest.mark.usefixtures("waterpurifier")
class TestWaterPurifier(TestCase):
def is_on(self):
return self.device.status().is_on
def state(self):
return self.device.status()
def test_on(self):
self.device.off() # ensure off
assert self.is_on() is False
self.device.on()
assert self.is_on() is True
def test_off(self):
self.device.on() # ensure on
assert self.is_on() is True
self.device.off()
assert self.is_on() is False
def test_status(self):
self.device._reset_state()
assert repr(self.state()) == repr(WaterPurifierStatus(self.device.start_state))
assert self.is_on() is True
|
