Merge branch 'fix-status-off'

Author: ziima

recuair_cli/main.py

@@ -63,11 +63,11 @@ class Status(NamedTuple):
 
     device: str
     name: str
-    temperature_in: int
-    humidity_in: int
-    temperature_out: int
+    temperature_in: Optional[int]
+    humidity_in: Optional[int]
+    temperature_out: Optional[int]
     mode: str
-    co2_ppm: int
+    co2_ppm: Optional[int]
     filter: int
     fan: int
     light: int
@@ -78,6 +78,14 @@ def _strip_unit(value: str) -> str:
     return value.strip().partition(" ")[0]
 
 
+def _int_or_none(value: str) -> Optional[int]:
+    """Convert value to int or return None."""
+    if value == "-":
+        return None
+    else:
+        return int(value)
+
+
 async def get_status(client: httpx.AsyncClient, device: str) -> Status:
     """Return device status."""
     try:
@@ -119,11 +127,11 @@ async def get_status(client: httpx.AsyncClient, device: str) -> Status:
         return Status(
             device=device,
             name=cast(Tag, content.find(class_="deviceName")).text,
-            temperature_in=int(_strip_unit(temp_in)),
-            humidity_in=int(_strip_unit(humi_in)),
-            temperature_out=int(_strip_unit(temp_out)),
+            temperature_in=_int_or_none(_strip_unit(temp_in)),
+            humidity_in=_int_or_none(_strip_unit(humi_in)),
+            temperature_out=_int_or_none(_strip_unit(temp_out)),
             mode=mode_raw.strip(),
-            co2_ppm=int(_strip_unit(co2_raw)),
+            co2_ppm=_int_or_none(_strip_unit(co2_raw)),
             filter=100 - int(_strip_unit(filter_raw)),
             fan=100 - int(_strip_unit(fan_raw)),
             light=int(light_raw),

recuair_cli/tests/data/response-off.html

@@ -14,6 +14,7 @@
 
 class GetStatusTest(IsolatedAsyncioTestCase):
     async def test(self):
+        # Test response.
         with respx.mock() as rsps:
             with open(_DATA / "response.html", "rb") as file:
                 rsps.get("http://example/").mock(Response(200, content=file.read()))
@@ -35,6 +36,29 @@ async def test(self):
             )
             self.assertEqual(status, result)
 
+    async def test_off(self):
+        # Test response from stopped device
+        with respx.mock() as rsps:
+            with open(_DATA / "response-off.html", "rb") as file:
+                rsps.get("http://example/").mock(Response(200, content=file.read()))
+
+            async with httpx.AsyncClient() as client:
+                status = await get_status(client, "example")
+
+            result = Status(
+                device="example",
+                name="Holly",
+                temperature_in=None,
+                humidity_in=None,
+                temperature_out=None,
+                mode="AUTO",
+                co2_ppm=None,
+                filter=2,
+                fan=69,
+                light=5,
+            )
+            self.assertEqual(status, result)
+
     async def test_invalid(self):
         # Test case with '%%content%%' in the response.
         with respx.mock() as rsps: