test: add integration test for unstable_unmockModule

Closes #15079

Author: BondarenkoAlex

e2e/__tests__/__snapshots__/nativeEsm.test.ts.snap

@@ -42,12 +42,20 @@ Ran all test suites matching native-esm-wasm.test.js."
 
 exports[`runs test with native ESM 1`] = `
 "Test Suites: 1 passed, 1 total
-Tests:       33 passed, 33 total
+Tests:       31 passed, 31 total
 Snapshots:   0 total
 Time:        <<REPLACED>>
 Ran all test suites matching native-esm.test.js."
 `;
 
+exports[`runs test with native mock ESM 1`] = `
+"Test Suites: 1 passed, 1 total
+Tests:       3 passed, 3 total
+Snapshots:   0 total
+Time:        <<REPLACED>>
+Ran all test suites matching native-esm-mocks.test.js."
+`;
+
 exports[`support re-exports from CJS of core module 1`] = `
 "Test Suites: 1 passed, 1 total
 Tests:       1 passed, 1 total

e2e/__tests__/nativeEsm.test.ts

@@ -51,6 +51,22 @@ test('runs test with native ESM', () => {
   expect(exitCode).toBe(0);
 });
 
+test('runs test with native mock ESM', () => {
+  const {exitCode, stderr, stdout} = runJest(
+    DIR,
+    ['native-esm-mocks.test.js'],
+    {
+      nodeOptions: '--experimental-vm-modules --no-warnings',
+    },
+  );
+
+  const {summary} = extractSummary(stderr);
+
+  expect(summary).toMatchSnapshot();
+  expect(stdout).toBe('');
+  expect(exitCode).toBe(0);
+});
+
 test('supports top-level await', () => {
   const {exitCode, stderr, stdout} = runJest(DIR, ['native-esm.tla.test.js'], {
     nodeOptions: '--experimental-vm-modules --no-warnings',

e2e/native-esm/__tests__/native-esm-mocks.test.js

@@ -0,0 +1,45 @@
+/**
+ * Copyright (c) Meta Platforms, Inc. and affiliates.
+ *
+ * This source code is licensed under the MIT license found in the
+ * LICENSE file in the root directory of this source tree.
+ */
+import {jest as jestObject} from '@jest/globals';
+
+afterEach(() => {
+  jestObject.resetModules();
+});
+
+test('can mock module', async () => {
+  jestObject.unstable_mockModule('../mockedModule.mjs', () => ({foo: 'bar'}), {
+    virtual: true,
+  });
+
+  const importedMock = await import('../mockedModule.mjs');
+
+  expect(Object.keys(importedMock)).toEqual(['foo']);
+  expect(importedMock.foo).toBe('bar');
+});
+
+test('can mock transitive module', async () => {
+  jestObject.unstable_mockModule('../index.js', () => ({foo: 'bar'}));
+
+  const importedMock = await import('../reexport.js');
+
+  expect(Object.keys(importedMock)).toEqual(['foo']);
+  expect(importedMock.foo).toBe('bar');
+});
+
+test('can unmock module', async () => {
+  jestObject.unstable_mockModule('../index.js', () => ({
+    double: () => 1000,
+  }));
+
+  const importedMock = await import('../index.js');
+  expect(importedMock.double()).toBe(1000);
+
+  jestObject.unstable_unmockModule('../index.js');
+
+  const importedMockAfterUnmock = await import('../index.js');
+  expect(importedMockAfterUnmock.double(2)).toBe(4);
+});

e2e/native-esm/__tests__/native-esm.test.js

@@ -224,26 +224,6 @@ test('require of ESM should throw correct error', () => {
   );
 });
 
-test('can mock module', async () => {
-  jestObject.unstable_mockModule('../mockedModule.mjs', () => ({foo: 'bar'}), {
-    virtual: true,
-  });
-
-  const importedMock = await import('../mockedModule.mjs');
-
-  expect(Object.keys(importedMock)).toEqual(['foo']);
-  expect(importedMock.foo).toBe('bar');
-});
-
-test('can mock transitive module', async () => {
-  jestObject.unstable_mockModule('../index.js', () => ({foo: 'bar'}));
-
-  const importedMock = await import('../reexport.js');
-
-  expect(Object.keys(importedMock)).toEqual(['foo']);
-  expect(importedMock.foo).toBe('bar');
-});
-
 test('supports imports using "node:" prefix', () => {
   expect(dns).toBe(prefixDns);
 });