Allow mock.property() to work with process.env variables

Author: Sakina Farukh Ahemad

lib/internal/test_runner/mock/mock.js

@@ -706,6 +706,16 @@ class MockTracker {
     validateObject(object, 'object');
     validateStringOrSymbol(propertyName, 'propertyName');
 
+    if (object === process.env) {
+    Object.defineProperty(object, propertyName, {
+      value,
+      writable: true,
+      enumerable: true,
+      configurable: true
+    });
+    return object;
+  }
+
     const ctx = arguments.length > 2 ?
       new MockPropertyContext(object, propertyName, value) :
       new MockPropertyContext(object, propertyName);

lib/test.js

@@ -1,75 +1,31 @@
 'use strict';
 
-const {
-  ObjectAssign,
-  ObjectDefineProperty,
-} = primordials;
+// Flags: --expose-internals
+require('../common');
+const assert = require('assert');
 
-const { test, suite, before, after, beforeEach, afterEach } = require('internal/test_runner/harness');
-const { run } = require('internal/test_runner/runner');
+// This test verifies mocking and restoring process.env.TEST_ENV
 
-module.exports = test;
-ObjectAssign(module.exports, {
-  after,
-  afterEach,
-  before,
-  beforeEach,
-  describe: suite,
-  it: test,
-  run,
-  suite,
-  test,
-});
+const hadOriginal = Object.prototype.hasOwnProperty.call(process.env, 'TEST_ENV');
+const original = process.env.TEST_ENV;
 
-let lazyMock;
+// Mock environment variable
+process.env.TEST_ENV = 'mocked';
+assert.strictEqual(process.env.TEST_ENV, 'mocked');
+console.log('Mocked process.env.TEST_ENV successfully');
 
-ObjectDefineProperty(module.exports, 'mock', {
-  __proto__: null,
-  configurable: true,
-  enumerable: true,
-  get() {
-    if (lazyMock === undefined) {
-      const { MockTracker } = require('internal/test_runner/mock/mock');
+// Restore original variable
+if (hadOriginal) {
+  process.env.TEST_ENV = original;
+} else {
+  delete process.env.TEST_ENV;
+}
 
-      lazyMock = new MockTracker();
-    }
 
-    return lazyMock;
-  },
-});
+if (hadOriginal) {
+  assert.strictEqual(process.env.TEST_ENV, original);
+} else {
+  assert.ok(!('TEST_ENV' in process.env));
+}
 
-let lazySnapshot;
-
-ObjectDefineProperty(module.exports, 'snapshot', {
-  __proto__: null,
-  configurable: true,
-  enumerable: true,
-  get() {
-    if (lazySnapshot === undefined) {
-      const {
-        setDefaultSnapshotSerializers,
-        setResolveSnapshotPath,
-      } = require('internal/test_runner/snapshot');
-
-      lazySnapshot = {
-        __proto__: null,
-        setDefaultSnapshotSerializers,
-        setResolveSnapshotPath,
-      };
-    }
-
-    return lazySnapshot;
-  },
-});
-
-ObjectDefineProperty(module.exports, 'assert', {
-  __proto__: null,
-  configurable: true,
-  enumerable: true,
-  get() {
-    const { register } = require('internal/test_runner/assert');
-    const assert = { __proto__: null, register };
-    ObjectDefineProperty(module.exports, 'assert', assert);
-    return assert;
-  },
-});
+console.log('Restored original process.env.TEST_ENV');

test/parallel/test-env-mock-process-env.js

@@ -0,0 +1,32 @@
+'use strict';
+
+// Flags: --expose-internals
+require('../common');
+const assert = require('assert');
+
+// This test verifies mocking and restoring process.env.TEST_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';
+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;
+}
+
+//  Validation
+if (hadOriginal) {
+  assert.strictEqual(process.env.TEST_ENV, original);
+} else {
+  assert.ok(!('TEST_ENV' in process.env));
+}
+
+console.log('Restored original process.env.TEST_ENV');
+