From 70d1b600d5732fda0de0ff055446474fb71246e4 Mon Sep 17 00:00:00 2001
From: Marc-Roig <marc12info@gmail.com>
Date: Mon, 17 Apr 2023 14:03:30 +0200
Subject: [PATCH] fix: rw migration tests

---
 .../admin/ee/review-workflows.test.api.js     | 33 +++++++++++--------
 1 file changed, 19 insertions(+), 14 deletions(-)

diff --git a/api-tests/core/admin/ee/review-workflows.test.api.js b/api-tests/core/admin/ee/review-workflows.test.api.js
index 7aed1d803b..fbfd56bd05 100644
--- a/api-tests/core/admin/ee/review-workflows.test.api.js
+++ b/api-tests/core/admin/ee/review-workflows.test.api.js
@@ -76,12 +76,6 @@ describeOnCondition(edition === 'EE')('Review workflows', () => {
     requests.admin = await createAuthRequest({ strapi });
   };
 
-  const getRWMorphTableResults = async (connection) =>
-    connection
-      .select('*')
-      .from('strapi_workflows_stages_related_morphs')
-      .where('related_type', productUID);
-
   beforeAll(async () => {
     await builder.addContentTypes([model]).build();
     // eslint-disable-next-line node/no-extraneous-require
@@ -371,13 +365,16 @@ describeOnCondition(edition === 'EE')('Review workflows', () => {
 
       expect(response.body.data.schema.reviewWorkflows).toBeTruthy();
 
-      const morphTableResults = await getRWMorphTableResults(strapi.db.getConnection());
+      const {
+        body: { results },
+      } = await requests.admin({
+        method: 'GET',
+        url: '/content-manager/collection-types/api::product.product',
+      });
 
-      expect(morphTableResults.length).toEqual(3);
-      for (let i = 0; i < morphTableResults.length; i += 1) {
-        const entry = morphTableResults[i];
-        expect(entry.related_id).toEqual(i + 1);
-        expect(entry.order).toEqual(1);
+      expect(results.length).toEqual(3);
+      for (let i = 0; i < results.length; i += 1) {
+        expect(results[i][ENTITY_STAGE_ATTRIBUTE]).toBeDefined();
       }
     });
 
@@ -395,8 +392,16 @@ describeOnCondition(edition === 'EE')('Review workflows', () => {
       });
       expect(response.body.data.schema.reviewWorkflows).toBeFalsy();
 
-      const morphTableResults = await getRWMorphTableResults(strapi.db.getConnection());
-      expect(morphTableResults.length).toEqual(0);
+      const {
+        body: { results },
+      } = await requests.admin({
+        method: 'GET',
+        url: '/content-manager/collection-types/api::product.product',
+      });
+
+      for (let i = 0; i < results.length; i += 1) {
+        expect(results[i][ENTITY_STAGE_ATTRIBUTE]).toBeUndefined();
+      }
     });
   });