rename rollback info fields

CiscoDevNet · Mar 22, 2024 · a5dae78 · a5dae78
1 parent 04a0490
commit a5dae78
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 6 deletions.
diff --git a/catalystwan/models/configuration/config_migration.py b/catalystwan/models/configuration/config_migration.py
@@ -112,11 +112,15 @@ class UX2Config(BaseModel):
 
 
 class UX2ConfigRollback(BaseModel):
-    config_groups_ids: List[UUID] = Field(default_factory=list)
-    feature_profiles_ids: List[Tuple[UUID, ProfileType]] = Field(default_factory=list)
+    config_group_ids: List[UUID] = Field(
+        default_factory=list, serialization_alias="ConfigGroupIds", validation_alias="ConfigGroupIds"
+    )
+    feature_profile_ids: List[Tuple[UUID, ProfileType]] = Field(
+        default_factory=list, serialization_alias="FeatureProfileIds", validation_alias="FeatureProfileIds"
+    )
 
     def add_config_group(self, config_group_id: UUID) -> None:
-        self.config_groups_ids.append(config_group_id)
+        self.config_group_ids.append(config_group_id)
 
     def add_feature_profile(self, feature_profile_id: UUID, profile_type: ProfileType) -> None:
-        self.feature_profiles_ids.append((feature_profile_id, profile_type))
+        self.feature_profile_ids.append((feature_profile_id, profile_type))
diff --git a/catalystwan/utils/config_migration/reverters/config_reverter.py b/catalystwan/utils/config_migration/reverters/config_reverter.py
@@ -11,9 +11,9 @@ def __init__(self, session) -> None:
 
     def rollback(self, rollback_config: UX2ConfigRollback) -> bool:
         try:
-            for cg_id in rollback_config.config_groups_ids:
+            for cg_id in rollback_config.config_group_ids:
                 self._session.endpoints.configuration_group.delete_config_group(cg_id)
-            for feature_profile_id, type_ in rollback_config.feature_profiles_ids:
+            for feature_profile_id, type_ in rollback_config.feature_profile_ids:
                 api = FeatureProfileAPIFactory.get_api(type_, self._session)
                 if type_ == "policy-object":
                     continue