From 06b87fca5458bb9813f1bfbb1606d7f5ffabc25c Mon Sep 17 00:00:00 2001
From: Ben Irvin <ben@innerdvations.com>
Date: Wed, 21 Dec 2022 15:34:56 +0100
Subject: [PATCH] fix unit tests

---
 .../__tests__/entities.test.ts                             | 7 +++++--
 .../local-strapi-source-provider/__tests__/index.test.ts   | 3 +++
 2 files changed, 8 insertions(+), 2 deletions(-)

diff --git a/packages/core/data-transfer/lib/providers/local-strapi-source-provider/__tests__/entities.test.ts b/packages/core/data-transfer/lib/providers/local-strapi-source-provider/__tests__/entities.test.ts
index 44308acaf0..ef4c29c15b 100644
--- a/packages/core/data-transfer/lib/providers/local-strapi-source-provider/__tests__/entities.test.ts
+++ b/packages/core/data-transfer/lib/providers/local-strapi-source-provider/__tests__/entities.test.ts
@@ -46,10 +46,13 @@ describe('Local Strapi Source Provider - Entities Streaming', () => {
           { id: 2, age: 84 },
         ],
       });
-
+      const contentTypes = getContentTypes();
       const strapi = getStrapiFactory({
-        contentTypes: getContentTypes(),
+        contentTypes,
         db: { queryBuilder },
+        getModel: jest.fn((uid) => {
+          return contentTypes[uid];
+        }),
       })();
 
       const entitiesStream = createEntitiesStream(strapi);
diff --git a/packages/core/data-transfer/lib/providers/local-strapi-source-provider/__tests__/index.test.ts b/packages/core/data-transfer/lib/providers/local-strapi-source-provider/__tests__/index.test.ts
index fed0c136d8..342c5a9ba2 100644
--- a/packages/core/data-transfer/lib/providers/local-strapi-source-provider/__tests__/index.test.ts
+++ b/packages/core/data-transfer/lib/providers/local-strapi-source-provider/__tests__/index.test.ts
@@ -95,6 +95,9 @@ describe('Local Strapi Source Provider', () => {
           db: {
             queryBuilder,
           },
+          getModel: jest.fn((uid) => {
+            return contentTypes[uid];
+          }),
         }),
       });