From 18fbe664d719d3413eab3cc996abbe505aba4576 Mon Sep 17 00:00:00 2001
From: Alexandre Bodin <bodin.alex@gmail.com>
Date: Wed, 15 Mar 2023 19:08:49 +0100
Subject: [PATCH] Fix tests

---
 .../server/controllers/__tests__/user.test.js | 33 +++++++++++++++++++
 .../__tests__/single-types.test.js            | 15 +++++++++
 2 files changed, 48 insertions(+)

diff --git a/packages/core/admin/server/controllers/__tests__/user.test.js b/packages/core/admin/server/controllers/__tests__/user.test.js
index f8845989a7..c2c4041719 100644
--- a/packages/core/admin/server/controllers/__tests__/user.test.js
+++ b/packages/core/admin/server/controllers/__tests__/user.test.js
@@ -174,13 +174,29 @@ describe('User Controller', () => {
         pagination,
       }));
 
+      const state = {
+        userAbility: {
+          can: jest.fn(),
+          cannot: jest.fn(() => false),
+        },
+      };
+
       const sanitizeUser = jest.fn((user) => user);
       const ctx = createContext({});
+      ctx.state = state;
+
+      const createPermissionsManager = jest.fn(() => ({
+        ability: state.userAbility,
+        sanitizeQuery: (query) => query,
+      }));
 
       global.strapi = {
         admin: {
           services: {
             user: { findPage, sanitizeUser },
+            permission: {
+              createPermissionsManager,
+            },
           },
         },
       };
@@ -199,13 +215,30 @@ describe('User Controller', () => {
         pagination,
       }));
 
+      const state = {
+        userAbility: {
+          can: jest.fn(),
+          cannot: jest.fn(() => false),
+        },
+      };
+
       const sanitizeUser = jest.fn((user) => user);
       const ctx = createContext({ query: { _q: 'foo' } });
 
+      ctx.state = state;
+
+      const createPermissionsManager = jest.fn(() => ({
+        ability: state.userAbility,
+        sanitizeQuery: (query) => query,
+      }));
+
       global.strapi = {
         admin: {
           services: {
             user: { findPage, sanitizeUser },
+            permission: {
+              createPermissionsManager,
+            },
           },
         },
       };
diff --git a/packages/core/content-manager/server/controllers/__tests__/single-types.test.js b/packages/core/content-manager/server/controllers/__tests__/single-types.test.js
index ed27ba93a8..6915ec2d6e 100644
--- a/packages/core/content-manager/server/controllers/__tests__/single-types.test.js
+++ b/packages/core/content-manager/server/controllers/__tests__/single-types.test.js
@@ -23,6 +23,9 @@ describe('Single Types', () => {
         create: jest.fn(() => false),
       },
       buildReadQuery: jest.fn((query) => query),
+      sanitizedQuery: {
+        read: (q) => q,
+      },
     };
 
     global.strapi = {
@@ -101,6 +104,9 @@ describe('Single Types', () => {
       sanitizeCreateInput: (obj) => obj,
       sanitizeOutput: (obj) => obj,
       buildReadQuery: jest.fn((query) => query),
+      sanitizedQuery: {
+        update: (q) => q,
+      },
     };
 
     const createFn = jest.fn(() => ({}));
@@ -215,6 +221,9 @@ describe('Single Types', () => {
       },
       sanitizeOutput: jest.fn((obj) => obj),
       buildReadQuery: jest.fn((query) => query),
+      sanitizedQuery: {
+        delete: (q) => q,
+      },
     };
 
     const deleteFn = jest.fn(() => ({}));
@@ -309,6 +318,9 @@ describe('Single Types', () => {
       },
       sanitizeOutput: jest.fn((obj) => obj),
       buildReadQuery: jest.fn((query) => query),
+      sanitizedQuery: {
+        publish: (q) => q,
+      },
     };
 
     const publishFn = jest.fn(() => ({}));
@@ -403,6 +415,9 @@ describe('Single Types', () => {
       },
       sanitizeOutput: jest.fn((obj) => obj),
       buildReadQuery: jest.fn((query) => query),
+      sanitizedQuery: {
+        unpublish: (q) => q,
+      },
     };
 
     const unpublishFn = jest.fn(() => ({}));