From 99b5982f1c6e0940acbe578d735aae31ddc57981 Mon Sep 17 00:00:00 2001
From: Marta Bondyra <marta.bondyra@elastic.co>
Date: Thu, 30 Apr 2020 15:42:56 +0200
Subject: [PATCH] fix: fix migration

---
 x-pack/plugins/lens/server/migrations.test.ts | 2 +-
 x-pack/plugins/lens/server/migrations.ts      | 4 ++--
 2 files changed, 3 insertions(+), 3 deletions(-)

diff --git a/x-pack/plugins/lens/server/migrations.test.ts b/x-pack/plugins/lens/server/migrations.test.ts
index 4cc330d40efd..0541d9636577 100644
--- a/x-pack/plugins/lens/server/migrations.test.ts
+++ b/x-pack/plugins/lens/server/migrations.test.ts
@@ -160,7 +160,7 @@ describe('Lens migrations', () => {
   });
 
   describe('7.8.0 auto timestamp', () => {
-    const context = {} as SavedObjectMigrationContext;
+    const context = ({ log: { warning: () => {} } } as unknown) as SavedObjectMigrationContext;
 
     const example = {
       type: 'lens',
diff --git a/x-pack/plugins/lens/server/migrations.ts b/x-pack/plugins/lens/server/migrations.ts
index 51fcd3b6198c..2fa6c904977f 100644
--- a/x-pack/plugins/lens/server/migrations.ts
+++ b/x-pack/plugins/lens/server/migrations.ts
@@ -4,7 +4,7 @@
  * you may not use this file except in compliance with the Elastic License.
  */
 
-import { cloneDeep, flow } from 'lodash';
+import { cloneDeep } from 'lodash';
 import { fromExpression, toExpression, Ast, ExpressionFunctionAST } from '@kbn/interpreter/common';
 import { SavedObjectMigrationFn } from 'src/core/server';
 
@@ -153,5 +153,5 @@ export const migrations: Record<string, SavedObjectMigrationFn> = {
   },
   // The order of these migrations matter, since the timefield migration relies on the aggConfigs
   // sitting directly on the esaggs as an argument and not a nested function (which lens_auto_date was).
-  '7.8.0': flow(removeLensAutoDate, addTimeFieldToEsaggs),
+  '7.8.0': (doc, context) => addTimeFieldToEsaggs(removeLensAutoDate(doc, context), context),
 };