From fa1a8cf1232b809a566e3e2ed7568a4ae78e4f2d Mon Sep 17 00:00:00 2001
From: Pierre Gayvallet <pierre.gayvallet@elastic.co>
Date: Wed, 9 Sep 2020 21:25:38 +0200
Subject: [PATCH] IndexMigrator: fix non blocking migration wrapper promise
 rejection (#77018) (#77092)

* fix transformNonBlocking

* add test for indexMigrator
---
 .../migrations/core/index_migrator.test.ts    | 24 +++++++++++++++++++
 .../migrations/core/migrate_raw_docs.test.ts  | 14 +++++++++++
 .../migrations/core/migrate_raw_docs.ts       |  8 +++++--
 3 files changed, 44 insertions(+), 2 deletions(-)

diff --git a/src/core/server/saved_objects/migrations/core/index_migrator.test.ts b/src/core/server/saved_objects/migrations/core/index_migrator.test.ts
index df89137a1d798b..13f771c16bc67b 100644
--- a/src/core/server/saved_objects/migrations/core/index_migrator.test.ts
+++ b/src/core/server/saved_objects/migrations/core/index_migrator.test.ts
@@ -369,6 +369,30 @@ describe('IndexMigrator', () => {
       ],
     });
   });
+
+  test('rejects when the migration function throws an error', async () => {
+    const { client } = testOpts;
+    const migrateDoc = jest.fn((doc: SavedObjectUnsanitizedDoc) => {
+      throw new Error('error migrating document');
+    });
+
+    testOpts.documentMigrator = {
+      migrationVersion: { foo: '1.2.3' },
+      migrate: migrateDoc,
+    };
+
+    withIndex(client, {
+      numOutOfDate: 1,
+      docs: [
+        [{ _id: 'foo:1', _source: { type: 'foo', foo: { name: 'Bar' } } }],
+        [{ _id: 'foo:2', _source: { type: 'foo', foo: { name: 'Baz' } } }],
+      ],
+    });
+
+    await expect(new IndexMigrator(testOpts).migrate()).rejects.toThrowErrorMatchingInlineSnapshot(
+      `"error migrating document"`
+    );
+  });
 });
 
 function withIndex(
diff --git a/src/core/server/saved_objects/migrations/core/migrate_raw_docs.test.ts b/src/core/server/saved_objects/migrations/core/migrate_raw_docs.test.ts
index 4c9d2e870a7bb3..83dc042d2b96bc 100644
--- a/src/core/server/saved_objects/migrations/core/migrate_raw_docs.test.ts
+++ b/src/core/server/saved_objects/migrations/core/migrate_raw_docs.test.ts
@@ -90,4 +90,18 @@ describe('migrateRawDocs', () => {
 
     expect(logger.error).toBeCalledTimes(1);
   });
+
+  test('rejects when the transform function throws an error', async () => {
+    const transform = jest.fn<any, any>((doc: any) => {
+      throw new Error('error during transform');
+    });
+    await expect(
+      migrateRawDocs(
+        new SavedObjectsSerializer(new SavedObjectTypeRegistry()),
+        transform,
+        [{ _id: 'a:b', _source: { type: 'a', a: { name: 'AAA' } } }],
+        createSavedObjectsMigrationLoggerMock()
+      )
+    ).rejects.toThrowErrorMatchingInlineSnapshot(`"error during transform"`);
+  });
 });
diff --git a/src/core/server/saved_objects/migrations/core/migrate_raw_docs.ts b/src/core/server/saved_objects/migrations/core/migrate_raw_docs.ts
index 2bdf59d25dc74d..5a5048d8ad88ff 100644
--- a/src/core/server/saved_objects/migrations/core/migrate_raw_docs.ts
+++ b/src/core/server/saved_objects/migrations/core/migrate_raw_docs.ts
@@ -78,10 +78,14 @@ function transformNonBlocking(
 ): (doc: SavedObjectUnsanitizedDoc) => Promise<SavedObjectUnsanitizedDoc> {
   // promises aren't enough to unblock the event loop
   return (doc: SavedObjectUnsanitizedDoc) =>
-    new Promise((resolve) => {
+    new Promise((resolve, reject) => {
       // set immediate is though
       setImmediate(() => {
-        resolve(transform(doc));
+        try {
+          resolve(transform(doc));
+        } catch (e) {
+          reject(e);
+        }
       });
     });
 }