full fixtures

app/api/migrations/migrations/161-denormalize_published_state/index.ts

@@ -10,6 +10,8 @@ export default {
 
   reindex: false,
 
+  batchSize: 1000,
+
   async up(db: Db) {
     process.stdout.write(`${this.name}...\r\n`);
   },

app/api/migrations/migrations/161-denormalize_published_state/specs/161-denormalize_published_state.spec.ts

@@ -13,6 +13,7 @@ const initTest = async (fixture: Fixture) => {
   await testingDB.setupFixturesAndContext(fixture);
   db = testingDB.mongodb!;
   migration.reindex = false;
+  migration.batchSize = 2;
   await migration.up(db);
   metadata = (await db.collection<Entity>('entities').find().toArray()).map(e => e.metadata || {});
 };
@@ -45,9 +46,6 @@ describe('migration test', () => {
         {
           text_property: [{ value: 'A', label: 'A' }],
         },
-        {
-          relationship: [{ value: 'unpublishedDoc', label: 'unpublishedDoc' }],
-        },
       ]);
     });
 

app/api/migrations/migrations/161-denormalize_published_state/specs/fixtures.ts

@@ -1,5 +1,5 @@
 import { ObjectId } from 'mongodb';
-import { Fixture, Template } from '../types';
+import { Fixture } from '../types';
 
 const sourceTemplate = {
   _id: new ObjectId(),
@@ -14,60 +14,149 @@ const sourceTemplate = {
   ],
 };
 
-const baseTemplates: Record<string, Template> = {
-  source: sourceTemplate,
-  withRelationship: {
-    _id: new ObjectId(),
-    name: 'Template With Relationship',
-    properties: [
-      {
-        _id: new ObjectId(),
-        label: 'Relationship To Source',
-        name: 'relationship_to_source',
-        type: 'relationship',
-        content: sourceTemplate._id.toString(),
-      },
-    ],
-  },
-};
-
-const unpublishedEntityTitle = 'unpublishedDoc';
+const unpublishedSourceTitle1 = 'unpublishedDoc1';
 
 const unchangedFixtures: Fixture = {
-  templates: Object.values(baseTemplates),
+  templates: [sourceTemplate],
   entities: [
     {
       _id: new ObjectId(),
-      title: unpublishedEntityTitle,
-      sharedId: unpublishedEntityTitle,
+      title: unpublishedSourceTitle1,
+      sharedId: unpublishedSourceTitle1,
       language: 'en',
-      template: baseTemplates.source._id,
+      template: sourceTemplate._id,
       published: false,
       metadata: {
         text_property: [{ value: 'A', label: 'A' }],
       },
     },
+  ],
+};
+
+const sourceTemplate2 = {
+  _id: new ObjectId(),
+  name: 'source_template_2',
+  properties: [
     {
       _id: new ObjectId(),
-      title: 'nonChangingEntity',
-      sharedId: 'nonChangingEntity',
-      language: 'en',
-      template: baseTemplates.withRelationship._id,
-      published: false,
-      metadata: {
-        relationship: [{ value: unpublishedEntityTitle, label: unpublishedEntityTitle }],
-      },
+      label: 'Text Property',
+      name: 'text_property',
+      type: 'text' as 'text',
     },
   ],
 };
 
+const templateWithRelationship = {
+  _id: new ObjectId(),
+  name: 'Template With Relationship',
+  properties: [
+    {
+      _id: new ObjectId(),
+      label: 'Relationship To Source',
+      name: 'relationship_to_source',
+      type: 'relationship',
+      content: sourceTemplate._id.toString(),
+    },
+  ],
+};
+
+const templateWithTwoRelationships = {
+  _id: new ObjectId(),
+  name: 'Template With Two Relationships',
+  properties: [
+    {
+      _id: new ObjectId(),
+      label: 'Relationship To Source',
+      name: 'relationship_to_source',
+      type: 'relationship' as 'relationship',
+      content: sourceTemplate._id.toString(),
+    },
+    {
+      _id: new ObjectId(),
+      label: 'Relationship To Source 2',
+      name: 'relationship_to_source_2',
+      type: 'relationship' as 'relationship',
+      content: sourceTemplate2._id.toString(),
+    },
+  ],
+};
+
+const unpublishedSourceTitle2 = 'unpublishedDoc2';
+
+const publishedSourceTitle1 = 'publishedDoc1';
+
+const publishedSourceTitle2 = 'publishedDoc2';
+
 const fixtures: Fixture = {
-  templates: {
-    ...unchangedFixtures.templates,
-  },
-  entities: {
+  templates: [...unchangedFixtures.templates, sourceTemplate2, templateWithTwoRelationships],
+  entities: [
     ...unchangedFixtures.entities,
-  },
+    {
+      _id: new ObjectId(),
+      title: unpublishedSourceTitle2,
+      sharedId: unpublishedSourceTitle2,
+      language: 'en',
+      template: sourceTemplate2._id,
+      published: false,
+      metadata: {
+        text_property: [{ value: 'B', label: 'B' }],
+      },
+    },
+    {
+      _id: new ObjectId(),
+      title: publishedSourceTitle1,
+      sharedId: publishedSourceTitle1,
+      language: 'en',
+      template: sourceTemplate._id,
+      published: true,
+      metadata: {
+        text_property: [{ value: 'C', label: 'C' }],
+      },
+    },
+    {
+      _id: new ObjectId(),
+      title: publishedSourceTitle2,
+      sharedId: publishedSourceTitle2,
+      language: 'en',
+      template: sourceTemplate2._id,
+      published: true,
+      metadata: {
+        text_property: [{ value: 'D', label: 'D' }],
+      },
+    },
+    {
+      _id: new ObjectId(),
+      title: 'docWithPublishedRelationship',
+      sharedId: 'docWithPublishedRelationship',
+      language: 'en',
+      template: templateWithRelationship._id,
+      published: true,
+      metadata: {
+        relationship_to_source: [
+          { value: publishedSourceTitle1, label: publishedSourceTitle1 },
+          { value: unpublishedSourceTitle1, label: unpublishedSourceTitle1 },
+        ],
+      },
+    },
+    {
+      id: new ObjectId(),
+      title: 'docWithTwoRelationships',
+      sharedId: 'docWithTwoRelationships',
+      language: 'en',
+      template: templateWithTwoRelationships._id,
+      published: true,
+      metadata: {
+        relationship_to_source: [
+          { value: publishedSourceTitle1, label: publishedSourceTitle1 },
+          { value: unpublishedSourceTitle1, label: unpublishedSourceTitle1 },
+        ],
+        relationship_to_source_2: [
+          { value: publishedSourceTitle2, label: publishedSourceTitle2 },
+          { value: unpublishedSourceTitle2, label: unpublishedSourceTitle2 },
+        ],
+      },
+    },
+  ],
 };
 
 export { fixtures, unchangedFixtures };