add engine tests

packages/core/data-transfer/lib/__tests__/test-utils.ts

@@ -27,8 +27,8 @@ export const getStrapiFactory =
   >(
     properties?: T
   ) =>
-  () => {
-    return { ...properties } as Strapi.Strapi;
+  (additionalProperties?: T) => {
+    return { ...properties, ...additionalProperties } as Strapi.Strapi;
   };
 
 /**

packages/core/data-transfer/lib/engine/__tests__/engine.test.ts

@@ -1,11 +1,105 @@
 import { Readable } from 'stream';
 import { Duplex } from 'stream-chain';
 import { createTransferEngine } from '..';
-import { IDestinationProvider, ISourceProvider, ITransferEngineOptions } from '../../../types';
+import {
+  IDestinationProvider,
+  ISourceProvider,
+  ITransferEngine,
+  ITransferEngineOptions,
+} from '../../../types';
 
-import { collect, createMockedQueryBuilder, getStrapiFactory } from '../../__tests__/test-utils';
+import { getStrapiFactory } from '../../__tests__/test-utils';
 
-const strapiFactory = getStrapiFactory();
+const strapiFactory = getStrapiFactory({});
+
+const providerStages = ['bootstrap', 'close'];
+
+const sourceStages = [
+  ...providerStages,
+  'streamEntities',
+  'streamLinks',
+  'streamMedia',
+  'streamConfiguration',
+  'streamSchemas',
+];
+const destinationStages = [
+  ...providerStages,
+  'getEntitiesStream',
+  'getLinksStream',
+  'getMediaStream',
+  'getConfigurationStream',
+  'getSchemasStream',
+];
+
+// add some helpers to jest
+expect.extend({
+  toBeValidTransferEngine(engine: ITransferEngine) {
+    try {
+      expect(engine).toBeDefined();
+
+      // All required methods exists
+      expect(engine.integrityCheck).toBeInstanceOf(Function);
+      expect(engine.transfer).toBeInstanceOf(Function);
+      expect(engine.bootstrap).toBeInstanceOf(Function);
+      expect(engine.close).toBeInstanceOf(Function);
+      expect(engine.transferSchemas).toBeInstanceOf(Function);
+      expect(engine.transferEntities).toBeInstanceOf(Function);
+      expect(engine.transferLinks).toBeInstanceOf(Function);
+      expect(engine.transferMedia).toBeInstanceOf(Function);
+      expect(engine.transferConfiguration).toBeInstanceOf(Function);
+      expect(engine.close).toBeInstanceOf(Function);
+      expect(engine.transfer).toBeInstanceOf(Function);
+    } catch (e) {
+      return {
+        pass: false,
+        message: () => `Expected engine to be valid: ${e.message}`,
+      };
+    }
+    return {
+      pass: true,
+      message: () => `Expected engine not to be valid`,
+    };
+  },
+  toHaveSourceStagesCalledOnce(provider: ISourceProvider) {
+    try {
+      sourceStages.forEach((stage) => {
+        if (provider[stage]) {
+          expect(provider[stage]).toHaveBeenCalledTimes(1);
+        }
+      });
+    } catch (e) {
+      return {
+        pass: false,
+        message: () => `Expected provider to have all stages called ${e.message}`,
+      };
+    }
+
+    return {
+      pass: true,
+      message: () => `Expected provider not to have all stages called`,
+    };
+  },
+  toHaveDestinationStagesCalledOnce(provider: IDestinationProvider) {
+    try {
+      destinationStages.forEach((stage) => {
+        // only check the stages that exist
+        if (provider[stage]) {
+          expect(provider[stage]).toHaveBeenCalledTimes(1);
+        }
+      });
+    } catch (e) {
+      return {
+        pass: false,
+        message: () => `Expected provider to have all stages called`,
+      };
+    }
+
+    return {
+      pass: true,
+      message: () => `Expected provider not to have all stages called`,
+    };
+  },
+});
 
 describe('Transfer engine', () => {
   // TODO: if these are needed for any other tests, a factory should be added to test-utils
@@ -44,29 +138,18 @@ describe('Transfer engine', () => {
         exclude: [],
       } as ITransferEngineOptions;
       const engine = createTransferEngine(mockedSource, mockedDestination, engineOptions);
-      expect(engine).toBeDefined();
-
-      // All required methods exists
-      expect(engine.integrityCheck).toBeInstanceOf(Function);
-      expect(engine.transfer).toBeInstanceOf(Function);
-      expect(engine.bootstrap).toBeInstanceOf(Function);
-      expect(engine.close).toBeInstanceOf(Function);
-      expect(engine.transferSchemas).toBeInstanceOf(Function);
-      expect(engine.transferEntities).toBeInstanceOf(Function);
-      expect(engine.transferLinks).toBeInstanceOf(Function);
-      expect(engine.transferMedia).toBeInstanceOf(Function);
-      expect(engine.transferConfiguration).toBeInstanceOf(Function);
-      expect(engine.close).toBeInstanceOf(Function);
-      expect(engine.transfer).toBeInstanceOf(Function);
+      expect(engine).toBeValidTransferEngine();
     });
   });
 
   describe('bootstrap', () => {
     test('works for providers without a bootstrap', async () => {
       const engine = createTransferEngine(mockedSource, mockedDestination, defaultOptions);
-
+      expect(engine).toBeValidTransferEngine();
       await engine.transfer();
+      expect(mockedSource).toHaveSourceStagesCalledOnce();
     });
+
     test('bootstraps all providers with a bootstrap', async () => {
       const source = {
         ...mockedSource,
@@ -77,11 +160,10 @@ describe('Transfer engine', () => {
         bootstrap: jest.fn().mockReturnValue(new Duplex()),
       };
       const engine = createTransferEngine(source, destination, defaultOptions);
-
+      expect(engine).toBeValidTransferEngine();
       await engine.transfer();
 
-      expect(source.bootstrap).toHaveBeenCalledTimes(1);
-      expect(destination.bootstrap).toHaveBeenCalledTimes(1);
+      expect(mockedSource).toHaveSourceStagesCalledOnce();
     });
   });
 
@@ -97,12 +179,14 @@ describe('Transfer engine', () => {
         strategy: 'restore',
       });
       await restoreEngine.transfer();
+      expect(restoreEngine).toBeValidTransferEngine();
 
       const mergeEngine = createTransferEngine(mockedSource, mockedDestination, {
         ...engineOptions,
         strategy: 'merge',
       });
       await mergeEngine.transfer();
+      expect(mergeEngine).toBeValidTransferEngine();
 
       await expect(
         (async () => {
@@ -115,5 +199,24 @@ describe('Transfer engine', () => {
         })()
       ).rejects.toThrow();
     });
+
+    test('calls all provider stages', async () => {
+      const engineOptions = {
+        versionMatching: 'exact',
+        exclude: [],
+        strategy: 'restore',
+      } as unknown as ITransferEngineOptions;
+      const engine = createTransferEngine(mockedSource, mockedDestination, engineOptions);
+
+      await engine.transfer();
+
+      expect(mockedSource).toHaveSourceStagesCalledOnce();
+      expect(mockedDestination).toHaveDestinationStagesCalledOnce();
+    });
+
+    test.todo('returns provider results');
+  });
+  describe('progressStream', () => {
+    test.todo('returns correct result count for each stage');
   });
 });