test that filter migrations are applied to lens migration map

elastic · Jan 5, 2022 · e63a5d7 · e63a5d7
1 parent 97494a7
commit e63a5d7
Showing 1 changed file with 52 additions and 0 deletions.
diff --git a/x-pack/plugins/lens/server/migrations/saved_object_migrations.test.ts b/x-pack/plugins/lens/server/migrations/saved_object_migrations.test.ts
@@ -1511,4 +1511,56 @@ describe('Lens migrations', () => {
       expect(result.attributes.state.filters).toEqual(expectedFilters);
     });
   });
+
+  test('should properly apply a filter migration within a lens visualization', () => {
+    const migrationVersion = 'some-version';
+
+    const lensVisualizationDoc = {
+      attributes: {
+        state: {
+          filters: [
+            {
+              filter: 1,
+              migrated: false,
+            },
+            {
+              filter: 2,
+              migrated: false,
+            },
+          ],
+        },
+      },
+    };
+
+    const migrationFunctionsObject = getAllMigrations({
+      [migrationVersion]: (filterState) => {
+        return {
+          ...filterState,
+          migrated: true,
+        };
+      },
+    });
+
+    const migratedLensDoc = migrationFunctionsObject[migrationVersion](
+      lensVisualizationDoc as SavedObjectUnsanitizedDoc,
+      {} as SavedObjectMigrationContext
+    );
+
+    expect(migratedLensDoc).toEqual({
+      attributes: {
+        state: {
+          filters: [
+            {
+              filter: 1,
+              migrated: true,
+            },
+            {
+              filter: 2,
+              migrated: true,
+            },
+          ],
+        },
+      },
+    });
+  });
 });