Always return a workarea name (#284)

Thomas55555 · Apr 25, 2024 · db1a599 · db1a599
1 parent 5441329
commit db1a599
Show file tree

Hide file tree

Showing 3 changed files with 9 additions and 5 deletions.
diff --git a/src/aioautomower/model.py b/src/aioautomower/model.py
@@ -295,15 +295,19 @@ class _WorkAreas(DataClassDictMixin):
     """DataClass for WorkAreas values."""
 
     work_area_id: int = field(metadata=field_options(alias="workAreaId"))
-    name: str
+    name: str = field(
+        metadata=field_options(
+            deserialize=lambda x: "my_lawn" if x == "" else x,
+        ),
+    )
     cutting_height: int = field(metadata=field_options(alias="cuttingHeight"))
 
 
 @dataclass
 class WorkArea(DataClassDictMixin):
     """DataClass for WorkAreas values."""
 
-    name: str | None
+    name: str
     cutting_height: int
 
 
@@ -332,7 +336,7 @@ class MowerAttributes(DataClassDictMixin):
         metadata=field_options(
             deserialize=lambda workarea_list: {
                 area.work_area_id: WorkArea(
-                    name=area.name or None, cutting_height=area.cutting_height
+                    name=area.name, cutting_height=area.cutting_height
                 )
                 for area in map(_WorkAreas.from_dict, workarea_list)
             },

diff --git a/tests/__snapshots__/test_high_feature_mower_model.ambr b/tests/__snapshots__/test_high_feature_mower_model.ambr
@@ -52,7 +52,7 @@
 # ---
 # name: test_mower_snapshot[work_areas]
   dict({
-    0: WorkArea(name=None, cutting_height=50),
+    0: WorkArea(name='my_lawn', cutting_height=50),
     123456: WorkArea(name='Front lawn', cutting_height=50),
   })
 # ---
diff --git a/tests/__snapshots__/test_session.ambr b/tests/__snapshots__/test_session.ambr
@@ -52,7 +52,7 @@
 # ---
 # name: test_connect[work_areas]
   dict({
-    0: WorkArea(name=None, cutting_height=50),
+    0: WorkArea(name='my_lawn', cutting_height=50),
     123456: WorkArea(name='Front lawn', cutting_height=50),
   })
 # ---