From 1880faf8a005a712f5c3e2e52fdbf04b87067066 Mon Sep 17 00:00:00 2001
From: Dan Rumney <dancrumb@gmail.com>
Date: Thu, 15 Jun 2023 12:52:45 -0500
Subject: [PATCH] fix(packages/grpc-js/test/assert2): move assert2 into its own
 file

Moving from exporting a namespace to just putting assert2 functions into their own files

Fixes #2464
---
 packages/grpc-js/test/assert2.ts | 93 ++++++++++++++++++++++++++++++++
 packages/grpc-js/test/common.ts  | 83 ++--------------------------
 2 files changed, 96 insertions(+), 80 deletions(-)
 create mode 100644 packages/grpc-js/test/assert2.ts

diff --git a/packages/grpc-js/test/assert2.ts b/packages/grpc-js/test/assert2.ts
new file mode 100644
index 000000000..d3912a928
--- /dev/null
+++ b/packages/grpc-js/test/assert2.ts
@@ -0,0 +1,93 @@
+/*
+ * Copyright 2019 gRPC authors.
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *     http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ *
+ */
+
+import * as assert from 'assert';
+
+const toCall = new Map<() => void, number>();
+const afterCallsQueue: Array<() => void> = [];
+
+/**
+ * Assert that the given function doesn't throw an error, and then return
+ * its value.
+ * @param fn The function to evaluate.
+ */
+export function noThrowAndReturn<T>(fn: () => T): T {
+  try {
+    return fn();
+  } catch (e) {
+    assert.throws(() => {
+      throw e;
+    });
+    throw e; // for type safety only
+  }
+}
+
+/**
+ * Helper function that returns true when every function wrapped with
+ * mustCall has been called.
+ */
+function mustCallsSatisfied(): boolean {
+  let result = true;
+  toCall.forEach(value => {
+    result = result && value === 0;
+  });
+  return result;
+}
+
+export function clearMustCalls(): void {
+  afterCallsQueue.length = 0;
+}
+
+/**
+ * Wraps a function to keep track of whether it was called or not.
+ * @param fn The function to wrap.
+ */
+// tslint:disable:no-any
+export function mustCall<T>(fn: (...args: any[]) => T): (...args: any[]) => T {
+  const existingValue = toCall.get(fn);
+  if (existingValue !== undefined) {
+    toCall.set(fn, existingValue + 1);
+  } else {
+    toCall.set(fn, 1);
+  }
+  return (...args: any[]) => {
+    const result = fn(...args);
+    const existingValue = toCall.get(fn);
+    if (existingValue !== undefined) {
+      toCall.set(fn, existingValue - 1);
+    }
+    if (mustCallsSatisfied()) {
+      afterCallsQueue.forEach(fn => fn());
+      afterCallsQueue.length = 0;
+    }
+    return result;
+  };
+}
+
+/**
+ * Calls the given function when every function that was wrapped with
+ * mustCall has been called.
+ * @param fn The function to call once all mustCall-wrapped functions have
+ *           been called.
+ */
+export function afterMustCallsSatisfied(fn: () => void): void {
+  if (!mustCallsSatisfied()) {
+    afterCallsQueue.push(fn);
+  } else {
+    fn();
+  }
+}
diff --git a/packages/grpc-js/test/common.ts b/packages/grpc-js/test/common.ts
index 24cb71650..16b393b55 100644
--- a/packages/grpc-js/test/common.ts
+++ b/packages/grpc-js/test/common.ts
@@ -16,7 +16,7 @@
  */
 
 import * as loader from '@grpc/proto-loader';
-import * as assert from 'assert';
+import * as assert2 from './assert2';
 
 import { GrpcObject, loadPackageDefinition } from '../src/make-client';
 
@@ -32,86 +32,9 @@ export function mockFunction(): never {
   throw new Error('Not implemented');
 }
 
-export namespace assert2 {
-  const toCall = new Map<() => void, number>();
-  const afterCallsQueue: Array<() => void> = [];
-
-  /**
-   * Assert that the given function doesn't throw an error, and then return
-   * its value.
-   * @param fn The function to evaluate.
-   */
-  export function noThrowAndReturn<T>(fn: () => T): T {
-    try {
-      return fn();
-    } catch (e) {
-      assert.throws(() => {
-        throw e;
-      });
-      throw e; // for type safety only
-    }
-  }
-
-  /**
-   * Helper function that returns true when every function wrapped with
-   * mustCall has been called.
-   */
-  function mustCallsSatisfied(): boolean {
-    let result = true;
-    toCall.forEach(value => {
-      result = result && value === 0;
-    });
-    return result;
-  }
-
-  export function clearMustCalls(): void {
-    afterCallsQueue.length = 0;
-  }
-
-  /**
-   * Wraps a function to keep track of whether it was called or not.
-   * @param fn The function to wrap.
-   */
-  // tslint:disable:no-any
-  export function mustCall<T>(
-    fn: (...args: any[]) => T
-  ): (...args: any[]) => T {
-    const existingValue = toCall.get(fn);
-    if (existingValue !== undefined) {
-      toCall.set(fn, existingValue + 1);
-    } else {
-      toCall.set(fn, 1);
-    }
-    return (...args: any[]) => {
-      const result = fn(...args);
-      const existingValue = toCall.get(fn);
-      if (existingValue !== undefined) {
-        toCall.set(fn, existingValue - 1);
-      }
-      if (mustCallsSatisfied()) {
-        afterCallsQueue.forEach(fn => fn());
-        afterCallsQueue.length = 0;
-      }
-      return result;
-    };
-  }
-
-  /**
-   * Calls the given function when every function that was wrapped with
-   * mustCall has been called.
-   * @param fn The function to call once all mustCall-wrapped functions have
-   *           been called.
-   */
-  export function afterMustCallsSatisfied(fn: () => void): void {
-    if (!mustCallsSatisfied()) {
-      afterCallsQueue.push(fn);
-    } else {
-      fn();
-    }
-  }
-}
-
 export function loadProtoFile(file: string): GrpcObject {
   const packageDefinition = loader.loadSync(file, protoLoaderOptions);
   return loadPackageDefinition(packageDefinition);
 }
+
+export { assert2 };