Add methods for requesting tests and calibrations (#114)

airgradienthq · Jun 18, 2024 · c5bd413 · c5bd413
1 parent 1a27570
commit c5bd413
Show file tree

Hide file tree

Showing 2 changed files with 16 additions and 0 deletions.
diff --git a/src/airgradient/airgradient.py b/src/airgradient/airgradient.py
@@ -107,6 +107,14 @@ async def set_led_bar_mode(self, led_bar_mode: LedBarMode) -> None:
         """Set LED bar mode on AirGradient device."""
         await self._set_config("ledBarMode", led_bar_mode)
 
+    async def request_co2_calibration(self) -> None:
+        """Request CO2 calibration on AirGradient device."""
+        await self._set_config("co2CalibrationRequested", value=True)
+
+    async def request_led_bar_test(self) -> None:
+        """Request LED bar test on AirGradient device."""
+        await self._set_config("ledBarTestRequested", value=True)
+
     async def close(self) -> None:
         """Close open client session."""
         if self.session and self._close_session:

diff --git a/tests/test_airgradient.py b/tests/test_airgradient.py
@@ -126,6 +126,14 @@ async def test_config(
             lambda client: client.set_led_bar_mode(LedBarMode.CO2),
             {"ledBarMode": "co2"},
         ),
+        (
+            lambda client: client.request_co2_calibration(),
+            {"co2CalibrationRequested": True},
+        ),
+        (
+            lambda client: client.request_led_bar_test(),
+            {"ledBarTestRequested": True},
+        ),
     ],
 )
 async def test_setting_config(