Add missing fields to config model (#116)

airgradienthq · Jun 19, 2024 · d4c6fe1 · d4c6fe1
1 parent c2619a2
commit d4c6fe1
Show file tree

Hide file tree

Showing 3 changed files with 15 additions and 2 deletions.
diff --git a/src/airgradient/models.py b/src/airgradient/models.py
@@ -94,3 +94,9 @@ class Config(DataClassORJSONMixin):
     post_data_to_airgradient: bool = field(
         metadata=field_options(alias="postDataToAirGradient")
     )
+    led_bar_brightness: int = field(metadata=field_options(alias="ledBarBrightness"))
+    display_brightness: int = field(metadata=field_options(alias="displayBrightness"))
+    nox_learning_offset: int = field(metadata=field_options(alias="noxLearningOffset"))
+    tvoc_learning_offset: int = field(
+        metadata=field_options(alias="tvocLearningOffset")
+    )
diff --git a/tests/__snapshots__/test_airgradient.ambr b/tests/__snapshots__/test_airgradient.ambr
@@ -4,10 +4,14 @@
     'co2_automatic_baseline_calibration_days': 8,
     'configuration_control': <ConfigurationControl.NOT_INITIALIZED: 'both'>,
     'country': 'DE',
+    'display_brightness': 0,
+    'led_bar_brightness': 100,
     'led_bar_mode': <LedBarMode.CO2: 'co2'>,
+    'nox_learning_offset': 12,
     'pm_standard': <PmStandard.UGM3: 'ugm3'>,
     'post_data_to_airgradient': True,
     'temperature_unit': <TemperatureUnit.CELSIUS: 'c'>,
+    'tvoc_learning_offset': 12,
   })
 # ---
 # name: test_current_fixtures[current_measures.json]

diff --git a/tests/fixtures/config.json b/tests/fixtures/config.json
@@ -2,12 +2,15 @@
     "country": "DE",
     "pmStandard": "ugm3",
     "ledBarMode": "co2",
-    "displayMode": "on",
     "abcDays": 8,
     "tvocLearningOffset": 12,
     "noxLearningOffset": 12,
     "mqttBrokerUrl": "",
     "temperatureUnit": "c",
     "configurationControl": "both",
-    "postDataToAirGradient": true
+    "postDataToAirGradient": true,
+    "ledBarBrightness": 100,
+    "displayBrightness": 0,
+    "offlineMode": false,
+    "model": "I-9PSL-DE"
 }