Fix types in test

elastic · Apr 8, 2020 · e789020 · e789020
1 parent ccdf919
commit e789020
Showing 1 changed file with 5 additions and 4 deletions.
diff --git a/x-pack/plugins/lens/server/migrations.test.ts b/x-pack/plugins/lens/server/migrations.test.ts
@@ -5,11 +5,12 @@
  */
 
 import { migrations, RawLensSavedXYObject770 } from './migrations';
-import { SimpleSavedObject } from 'src/core/public';
+import { SavedObjectUnsanitizedDoc } from 'src/core/server';
 
 describe('Lens migrations', () => {
   describe('7.7.0 missing dimensions in XY', () => {
-    const migrate = (doc: SimpleSavedObject | RawLensSavedXYObject770) => migrations['7.7.0'](doc);
+    const migrate = (doc: SavedObjectUnsanitizedDoc | RawLensSavedXYObject770) =>
+      migrations['7.7.0'](doc);
 
     const example: RawLensSavedXYObject770 = {
       type: 'lens',
@@ -106,7 +107,7 @@ describe('Lens migrations', () => {
           visualizationType: 'lnsMetric',
         },
       };
-      const result = migrate(target as SimpleSavedObject);
+      const result = migrate(target as SavedObjectUnsanitizedDoc);
       expect(result).toEqual(target);
     });
 
@@ -124,7 +125,7 @@ describe('Lens migrations', () => {
             },
           },
         },
-      } as SimpleSavedObject) as RawLensSavedXYObject770;
+      } as SavedObjectUnsanitizedDoc) as RawLensSavedXYObject770;
 
       expect(result.attributes.state.visualization.layers).toEqual([
         {