minor refactoring

Author: notanengineercom

src/SubstituteBase.ts

@@ -7,7 +7,7 @@ export class SubstituteJS {
     this._lastRegisteredSubstituteJSMethodOrProperty = value;
   }
   get lastRegisteredSubstituteJSMethodOrProperty() {
-    return typeof this._lastRegisteredSubstituteJSMethodOrProperty === 'undefined' ? 'root' : this._lastRegisteredSubstituteJSMethodOrProperty;
+    return this._lastRegisteredSubstituteJSMethodOrProperty ?? 'root';
   }
   [Symbol.toPrimitive]() {
     return `[class ${this.constructor.name}] -> ${this.lastRegisteredSubstituteJSMethodOrProperty}`;

src/Utilities.ts

@@ -1,7 +1,4 @@
 import { Argument, AllArguments } from './Arguments';
-import { GetPropertyState } from './states/GetPropertyState';
-import { InitialState } from './states/InitialState';
-import { Context } from './Context';
 import * as util from 'util';
 
 export type Call = any[] // list of args
@@ -11,7 +8,7 @@ export enum PropertyType {
     property = 'property'
 }
 
-export enum SubstituteMethods {
+export enum SubstituteMethod {
     received = 'received',
     didNotReceive = 'didNotReceive',
     mimicks = 'mimicks',
@@ -21,6 +18,14 @@ export enum SubstituteMethods {
     rejects = 'rejects'
 }
 
+export function isSubstituteMethod(property: PropertyKey): property is SubstituteMethod {
+    return property === SubstituteMethod.returns ||
+        property === SubstituteMethod.mimicks ||
+        property === SubstituteMethod.throws ||
+        property === SubstituteMethod.resolves ||
+        property === SubstituteMethod.rejects;
+}
+
 const seenObject = Symbol();
 
 export function stringifyArguments(args: any[]) {