add exporter test

elastic · Jan 18, 2021 · 417aff9 · 417aff9
1 parent 3887dbd
commit 417aff9
Showing 1 changed file with 48 additions and 1 deletion.
diff --git a/src/core/server/saved_objects/export/saved_objects_exporter.test.ts b/src/core/server/saved_objects/export/saved_objects_exporter.test.ts
@@ -17,14 +17,15 @@
  * under the License.
  */
 
+import type { SavedObject } from '../../../types';
 import { SavedObjectsExporter } from './saved_objects_exporter';
 import { savedObjectsClientMock } from '../service/saved_objects_client.mock';
 import { SavedObjectTypeRegistry } from '../saved_objects_type_registry';
 import { httpServerMock } from '../../http/http_server.mocks';
 import { Readable } from 'stream';
 import { createPromiseFromStreams, createConcatStream } from '@kbn/utils';
 
-async function readStreamToCompletion(stream: Readable) {
+async function readStreamToCompletion(stream: Readable): Promise<Array<SavedObject<any>>> {
   return createPromiseFromStreams([stream, createConcatStream([])]);
 }
 
@@ -132,6 +133,52 @@ describe('getSortedObjectsForExport()', () => {
           `);
     });
 
+    test('applies the export transforms', async () => {
+      typeRegistry.registerType({
+        name: 'foo',
+        mappings: { properties: {} },
+        namespaceType: 'single',
+        hidden: false,
+        management: {
+          importableAndExportable: true,
+          onExport: (ctx, objects) => {
+            objects.forEach((obj: SavedObject<any>) => {
+              obj.attributes.foo = 'modified';
+            });
+            return objects;
+          },
+        },
+      });
+      exporter = new SavedObjectsExporter({ savedObjectsClient, exportSizeLimit, typeRegistry });
+
+      savedObjectsClient.find.mockResolvedValueOnce({
+        total: 1,
+        saved_objects: [
+          {
+            id: '1',
+            type: 'foo',
+            attributes: {
+              foo: 'initial',
+            },
+            score: 0,
+            references: [],
+          },
+        ],
+        per_page: 1,
+        page: 0,
+      });
+      const exportStream = await exporter.exportByTypes({
+        request,
+        types: ['foo'],
+        excludeExportDetails: true,
+      });
+
+      const response = await readStreamToCompletion(exportStream);
+
+      expect(response).toHaveLength(1);
+      expect(response[0].attributes.foo).toEqual('modified');
+    });
+
     test('omits the `namespaces` property from the export', async () => {
       savedObjectsClient.find.mockResolvedValueOnce({
         total: 2,