fix: Add documentation + check of expire function

egguy · Mar 10, 2024 · ffd13ff · ffd13ff
1 parent be2399b
commit ffd13ff
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/src/grottext/ha/constants.py b/src/grottext/ha/constants.py
@@ -10,6 +10,7 @@
 MQTT_PASSWORD_CONF_KEY = "ha_mqtt_password"
 MQTT_RETAIN_CONF_KEY = "ha_mqtt_retain"
 
+# https://www.home-assistant.io/integrations/sensor.mqtt/#expire_after
 MQTT_EXPIRE_AFTER = 15 * 60  # 15 minutes
 
 
@@ -19,6 +20,3 @@ class MQTTConfiguration:
     hostname: str
     port: int
     client_id: str = "Grott - HA"
-
-
-MQTT_EXPIRE_AFTER = 15 * 60  # 15 minutes
diff --git a/tests/test_mappings.py b/tests/test_mappings.py
@@ -1,3 +1,4 @@
+from grottext.ha.constants import MQTT_EXPIRE_AFTER
 from grottext.ha.ha_types import to_dict
 from grottext.ha.mappings import mapping
 from grottext.ha.mqtt import MQTTConfigPayload, make_payload
@@ -18,6 +19,7 @@ def test_generate_payload(fake_config):
     assert payload["state_class"] == "measurement"
     assert payload["device_class"] == "power"
     assert payload["unit_of_measurement"] == "W"
+    assert payload["expire_after"] == MQTT_EXPIRE_AFTER
 
 
 def test_generate_log_type_payload(fake_config):