add the 'wrapInstance' helper for jest tests

Author: Alyar

packages/devextreme/js/__internal/grids/grid_core/__tests__/__mock__/helpers/wrapInstance.ts

@@ -0,0 +1,16 @@
+import { jest } from '@jest/globals';
+
+export default function wrapInstance<T extends object>(instance: T): T {
+  const proto = Object.getPrototypeOf(instance);
+
+  (Object.getOwnPropertyNames(proto) as (keyof T)[]).forEach((key): void => {
+    const originalValue = instance[key];
+    if (typeof originalValue === 'function' && key !== 'constructor') {
+      const originalMethod = originalValue as (...a: unknown[]) => unknown;
+
+      instance[key] = jest.fn(originalMethod.bind(instance)) as T[typeof key];
+    }
+  });
+
+  return instance;
+}

packages/devextreme/js/__internal/grids/grid_core/ai_column/ai_prompt_editor/ai_prompt_editor.integration.test.ts

@@ -7,7 +7,7 @@ import $ from '@js/core/renderer';
 import Popup from '@js/ui/popup';
 import ProgressBar from '@js/ui/progress_bar';
 import TextArea from '@js/ui/text_area';
-import { AiPromptEditorModel } from '@ts/grids/grid_core//__tests__/__mock__/model/ai_prompt_editor';
+import { AiPromptEditorModel } from '@ts/grids/grid_core/__tests__/__mock__/model/ai_prompt_editor';
 
 import { AiPromptEditor } from './ai_prompt_editor';
 import type { AiPromptEditorOptions } from './types';

packages/devextreme/js/__internal/grids/grid_core/ai_column/m_ai_column_view.test.ts

@@ -11,7 +11,8 @@ import fx from '@js/common/core/animation/fx';
 import type { dxElementWrapper } from '@js/core/renderer';
 import $ from '@js/core/renderer';
 import Callbacks from '@js/core/utils/callbacks';
-import { AiPromptEditorModel } from '@ts/grids/grid_core//__tests__/__mock__/model/ai_prompt_editor';
+import wrapInstanceWithMocks from '@ts/grids/grid_core/__tests__/__mock__/helpers/wrapInstance';
+import { AiPromptEditorModel } from '@ts/grids/grid_core/__tests__/__mock__/model/ai_prompt_editor';
 
 import type { Column } from '../columns_controller/m_columns_controller';
 import { AiPromptEditor } from './ai_prompt_editor/ai_prompt_editor';
@@ -20,26 +21,11 @@ import { AiColumnView } from './m_ai_column_view';
 
 jest.mock('./ai_prompt_editor/ai_prompt_editor', (): any => {
   const original = jest.requireActual<any>('./ai_prompt_editor/ai_prompt_editor');
-  const wrapInstanceWithMocks = (instance: AiPromptEditor): AiPromptEditor => {
-    const proto = Object.getPrototypeOf(instance);
-
-    Object.getOwnPropertyNames(proto).forEach((key): void => {
-      const originalMethod = instance[key];
-      if (typeof originalMethod === 'function' && key !== 'constructor') {
-        instance[key] = jest.fn(function mockMethod(...args: any[]) {
-          // eslint-disable-next-line @typescript-eslint/no-unsafe-return
-          return originalMethod.apply(this, args) as any;
-        });
-      }
-    });
-
-    return instance;
-  };
 
   return {
     ...original,
     AiPromptEditor: jest.fn((...args: any[]) => {
-      const instance = new original.AiPromptEditor(...args);
+      const instance: AiPromptEditor = new original.AiPromptEditor(...args);
       return wrapInstanceWithMocks(instance);
     }),
   };