From 6dca9b906cd26d96086809a279933c93668ce59c Mon Sep 17 00:00:00 2001
From: MURAKAMI Masahiko <fossamagna2@gmail.com>
Date: Fri, 18 Nov 2022 18:31:49 +0900
Subject: [PATCH] test_runner: add getter and setter to MockTracker

This commit allows tests in test runner to use the
`getter` and `setter` methods as "syntax sugar" for
`MockTracker.method` with the `options.getter` or
`options.setter` set to true in the options.

Refs: https://github.com/nodejs/node/pull/45326#discussion_r1014727289
---
 doc/api/test.md                      | 17 ++++++++
 lib/internal/test_runner/mock.js     | 24 +++++++++++
 test/parallel/test-runner-mocking.js | 63 ++++++++++++++++++++++++++++
 3 files changed, 104 insertions(+)

diff --git a/doc/api/test.md b/doc/api/test.md
index 4d5c50f51ff8ec..d6e51fb1d51574 100644
--- a/doc/api/test.md
+++ b/doc/api/test.md
@@ -995,6 +995,22 @@ test('spies on an object method', (t) => {
 });
 ```
 
+### `mock.getter(object, methodName[, implementation][, options])`
+
+<!-- YAML
+added: REPLACEME
+-->
+
+This function is syntax sugar for [`MockTracker.method`][] with `options.getter` is `true`.
+
+### `mock.setter(object, methodName[, implementation][, options])`
+
+<!-- YAML
+added: REPLACEME
+-->
+
+This function is syntax sugar for [`MockTracker.method`][] with `options.setter` is `true`.
+
 ### `mock.reset()`
 
 <!-- YAML
@@ -1357,6 +1373,7 @@ added:
 [`--test-only`]: cli.md#--test-only
 [`--test`]: cli.md#--test
 [`MockFunctionContext`]: #class-mockfunctioncontext
+[`MockTracker.method`]: #mockmethodobject-methodname-implementation-options
 [`MockTracker`]: #class-mocktracker
 [`SuiteContext`]: #class-suitecontext
 [`TestContext`]: #class-testcontext
diff --git a/lib/internal/test_runner/mock.js b/lib/internal/test_runner/mock.js
index 6b1d7e0c570a50..0b9de6bf2e1031 100644
--- a/lib/internal/test_runner/mock.js
+++ b/lib/internal/test_runner/mock.js
@@ -202,6 +202,30 @@ class MockTracker {
     return mock;
   }
 
+  getter(
+    object,
+    methodName,
+    implementation = kDefaultFunction,
+    options = kEmptyObject
+  ) {
+    return this.method(object, methodName, implementation, {
+      getter: true,
+      ...options
+    });
+  }
+
+  setter(
+    object,
+    methodName,
+    implementation = kDefaultFunction,
+    options = kEmptyObject
+  ) {
+    return this.method(object, methodName, implementation, {
+      setter: true,
+      ...options
+    });
+  }
+
   reset() {
     this.restoreAll();
     this.#mocks = [];
diff --git a/test/parallel/test-runner-mocking.js b/test/parallel/test-runner-mocking.js
index 6d3233ec0ed495..9d7178865a21ca 100644
--- a/test/parallel/test-runner-mocking.js
+++ b/test/parallel/test-runner-mocking.js
@@ -534,6 +534,69 @@ test('mocks a setter', (t) => {
   assert.strictEqual(obj.prop, 65);
 });
 
+test('mocks a getter with syntax sugar', (t) => {
+  const obj = {
+    prop: 5,
+    get method() {
+      return this.prop;
+    },
+  };
+
+  function mockMethod() {
+    return this.prop - 1;
+  }
+  const getter = t.mock.getter(obj, 'method', mockMethod);
+  assert.strictEqual(getter.mock.calls.length, 0);
+  assert.strictEqual(obj.method, 4);
+
+  const call = getter.mock.calls[0];
+
+  assert.deepStrictEqual(call.arguments, []);
+  assert.strictEqual(call.result, 4);
+  assert.strictEqual(call.target, undefined);
+  assert.strictEqual(call.this, obj);
+
+  assert.strictEqual(getter.mock.restore(), undefined);
+  assert.strictEqual(obj.method, 5);
+});
+
+test('mocks a setter with syntax sugar', (t) => {
+  const obj = {
+    prop: 100,
+    // eslint-disable-next-line accessor-pairs
+    set method(val) {
+      this.prop = val;
+    },
+  };
+
+  function mockMethod(val) {
+    this.prop = -val;
+  }
+
+  assert.strictEqual(obj.prop, 100);
+  obj.method = 88;
+  assert.strictEqual(obj.prop, 88);
+
+  const setter = t.mock.setter(obj, 'method', mockMethod);
+
+  assert.strictEqual(setter.mock.calls.length, 0);
+  obj.method = 77;
+  assert.strictEqual(obj.prop, -77);
+  assert.strictEqual(setter.mock.calls.length, 1);
+
+  const call = setter.mock.calls[0];
+
+  assert.deepStrictEqual(call.arguments, [77]);
+  assert.strictEqual(call.result, undefined);
+  assert.strictEqual(call.target, undefined);
+  assert.strictEqual(call.this, obj);
+
+  assert.strictEqual(setter.mock.restore(), undefined);
+  assert.strictEqual(obj.prop, -77);
+  obj.method = 65;
+  assert.strictEqual(obj.prop, 65);
+});
+
 test('mocked functions match name and length', (t) => {
   function getNameAndLength(fn) {
     return {