test: add mock.property() test for process.env tracking and restore

Author: Sakina Farukh Ahemad

lib/internal/test_runner/mock/mock.js

@@ -707,14 +707,15 @@ class MockTracker {
     validateStringOrSymbol(propertyName, 'propertyName');
 
     if (object === process.env) {
-    Object.defineProperty(object, propertyName, {
-      value,
-      writable: true,
-      enumerable: true,
-      configurable: true
-    });
-    return object;
-  }
+
+    const originalValue = object[propertyName];
+    object[propertyName] = value;
+  
+    return new ProxyConstructor({ restore: () => {
+    object[propertyName] = originalValue;
+    }});
+    }
+
 
     const ctx = arguments.length > 2 ?
       new MockPropertyContext(object, propertyName, value) :

lib/test.js

@@ -1,26 +1,21 @@
 'use strict';
 
-// Flags: --expose-internals
 require('../common');
 const assert = require('assert');
+const { mock } = require('node:test');
 
-// This test verifies mocking and restoring process.env.TEST_ENV
+console.log('Testing mock.property() with process.env...');
 
 const hadOriginal = Object.prototype.hasOwnProperty.call(process.env, 'TEST_ENV');
 const original = process.env.TEST_ENV;
 
-// Mock environment variable
-process.env.TEST_ENV = 'mocked';
+
+const tracker = mock.property(process.env, 'TEST_ENV', 'mocked');
+
 assert.strictEqual(process.env.TEST_ENV, 'mocked');
 console.log('Mocked process.env.TEST_ENV successfully');
 
-// Restore original variable
-if (hadOriginal) {
-  process.env.TEST_ENV = original;
-} else {
-  delete process.env.TEST_ENV;
-}
-
+tracker.restore();
 
 if (hadOriginal) {
   assert.strictEqual(process.env.TEST_ENV, original);