Fix tests by renaming planner_id back to id

L015H4CK · Jul 22, 2024 · e2997d8 · e2997d8
1 parent c11a62c
commit e2997d8
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/app/objects/c_planner.py b/app/objects/c_planner.py
@@ -8,7 +8,7 @@
 
 
 class PlannerSchema(ma.Schema):
-    planner_id = ma.fields.String()
+    planner_id = ma.fields.String(data_key='id')
     name = ma.fields.String()
     module = ma.fields.String()
     params = ma.fields.Dict()
@@ -20,8 +20,8 @@ class PlannerSchema(ma.Schema):
 
     @ma.pre_load
     def fix_id(self, data, **_):
-        if 'id' in data:
-            data['planner_id'] = data.pop('id')
+        if 'planner_id' in data:
+            data['id'] = data.pop('planner_id')
         return data
 
     @ma.post_load()