feat: support rs.hoisted

Author: fi3ework

e2e/mock/src/sum.ts

@@ -2,3 +2,4 @@ import { bar } from './bar';
 import { foo } from './foo';
 
 export const sum = foo + bar;
+export { bar, foo };

e2e/mock/tests/hoisted.test.ts

@@ -0,0 +1,20 @@
+import { expect, it, rs } from '@rstest/core';
+import { foo } from '../src/sum';
+
+// `rs` should can be accessed in hoisted function.
+const mocks = rs.hoisted(() => {
+  return {
+    hoistedFn: rs.fn(),
+  };
+});
+
+rs.mock('../src/sum', () => {
+  return { foo: mocks.hoistedFn };
+});
+
+it('hoisted', () => {
+  mocks.hoistedFn(42);
+  expect(mocks.hoistedFn).toHaveBeenCalledOnce();
+  expect(mocks.hoistedFn).toHaveBeenCalledWith(42);
+  expect(foo).toBe(mocks.hoistedFn);
+});

package.json

@@ -58,6 +58,9 @@
     "pnpm": ">=10.12.4"
   },
   "pnpm": {
+    "overrides": {
+      "@rspack/core": "link:../rspack/packages/rspack"
+    },
     "ignoredBuiltDependencies": [
       "@biomejs/biome",
       "nx",

packages/core/src/core/plugins/mockRuntimeCode.js

@@ -169,3 +169,8 @@ __webpack_require__.rstest_reset_modules = () => {
   });
 };
 //#endregion
+
+//#region rs.hoisted
+__webpack_require__.rstest_hoisted = (fn) => {
+  return fn();
+};

packages/core/src/runtime/api/utilities.ts

@@ -82,7 +82,11 @@ export const createRstestUtilities: (
     },
     resetModules: () => {
       // The actual implementation is managed by the built-in Rstest plugin.
-      return rstest;
+      return {} as any;
+    },
+    hoisted: () => {
+      // The actual implementation is managed by the built-in Rstest plugin.
+      return {} as any;
     },
 
     setConfig: (config) => {

packages/core/src/types/mock.ts

@@ -238,6 +238,11 @@ export interface RstestUtilities {
     moduleFactory?: () => T,
   ) => void;
 
+  /**
+   * Hoisted mock function.
+   */
+  hoisted: <T = unknown>(fn: () => T) => T;
+
   /**
    * Removes module from the mocked registry.
    */