chore(NODE-7428): import from mongodb.ts file in tests instead of src (#4864)

Co-authored-by: Sergey Zelenov <sergey.zelenov@mongodb.com>

Author: baileympearson

.eslintrc.json

@@ -186,6 +186,7 @@
           {
             "patterns": [
               "**/../lib/**",
+              "**/../src/**",
               "mongodb-mock-server"
             ]
           }
@@ -224,7 +225,8 @@
           {
             "patterns": [
               "**/../lib/**",
-              "mongodb-mock-server"
+              "mongodb-mock-server",
+              "**/../src/**"
             ]
           }
         ]
@@ -331,6 +333,15 @@
             "disallowTypeAnnotations": false,
             "fixStyle": "separate-type-imports"
           }
+        ],
+        "no-restricted-imports": [
+          "error",
+          {
+            "patterns": [
+              "**/../lib/**",
+              "**/../src/**"
+            ]
+          }
         ]
       }
     }

src/cmap/auth/mongodb_oidc.ts

@@ -5,11 +5,11 @@ import type { Connection } from '../connection';
 import { type AuthContext, AuthProvider } from './auth_provider';
 import type { MongoCredentials } from './mongo_credentials';
 import { AutomatedCallbackWorkflow } from './mongodb_oidc/automated_callback_workflow';
-import { callback as azureCallback } from './mongodb_oidc/azure_machine_workflow';
-import { callback as gcpCallback } from './mongodb_oidc/gcp_machine_workflow';
-import { callback as k8sCallback } from './mongodb_oidc/k8s_machine_workflow';
+import { azureCallback } from './mongodb_oidc/azure_machine_workflow';
+import { gcpCallback } from './mongodb_oidc/gcp_machine_workflow';
+import { k8sCallback } from './mongodb_oidc/k8s_machine_workflow';
 import { TokenCache } from './mongodb_oidc/token_cache';
-import { callback as testCallback } from './mongodb_oidc/token_machine_workflow';
+import { tokenMachineCallback as testCallback } from './mongodb_oidc/token_machine_workflow';
 
 /** Error when credentials are missing. */
 const MISSING_CREDENTIALS_ERROR = 'AuthContext must provide credentials.';

src/cmap/auth/mongodb_oidc/azure_machine_workflow.ts

@@ -19,7 +19,7 @@ const TOKEN_RESOURCE_MISSING_ERROR =
  * @param params - The OIDC callback parameters.
  * @returns The OIDC response.
  */
-export const callback: OIDCCallbackFunction = async (
+export const azureCallback: OIDCCallbackFunction = async (
   params: OIDCCallbackParams
 ): Promise<OIDCResponse> => {
   const tokenAudience = params.tokenAudience;

src/cmap/auth/mongodb_oidc/gcp_machine_workflow.ts

@@ -18,7 +18,7 @@ const TOKEN_RESOURCE_MISSING_ERROR =
  * @param params - The OIDC callback parameters.
  * @returns The OIDC response.
  */
-export const callback: OIDCCallbackFunction = async (
+export const gcpCallback: OIDCCallbackFunction = async (
   params: OIDCCallbackParams
 ): Promise<OIDCResponse> => {
   const tokenAudience = params.tokenAudience;

src/cmap/auth/mongodb_oidc/k8s_machine_workflow.ts

@@ -17,7 +17,7 @@ const AWS_FILENAME = 'AWS_WEB_IDENTITY_TOKEN_FILE';
  * @param params - The OIDC callback parameters.
  * @returns The OIDC response.
  */
-export const callback: OIDCCallbackFunction = async (): Promise<OIDCResponse> => {
+export const k8sCallback: OIDCCallbackFunction = async (): Promise<OIDCResponse> => {
   let filename: string;
   if (process.env[AZURE_FILENAME]) {
     filename = process.env[AZURE_FILENAME];

src/cmap/auth/mongodb_oidc/token_machine_workflow.ts

@@ -12,7 +12,7 @@ const TOKEN_MISSING_ERROR = 'OIDC_TOKEN_FILE must be set in the environment.';
  * @param params - The OIDC callback parameters.
  * @returns The OIDC response.
  */
-export const callback: OIDCCallbackFunction = async (): Promise<OIDCResponse> => {
+export const tokenMachineCallback: OIDCCallbackFunction = async (): Promise<OIDCResponse> => {
   const tokenFile = process.env.OIDC_TOKEN_FILE;
   if (!tokenFile) {
     throw new MongoAWSError(TOKEN_MISSING_ERROR);

test/action/dependency.test.ts

@@ -5,7 +5,7 @@ import * as path from 'node:path';
 import { expect } from 'chai';
 
 import { dependencies, peerDependencies, peerDependenciesMeta } from '../../package.json';
-import { setDifference } from '../../src/utils';
+import { setDifference } from '../mongodb';
 import { alphabetically, itInNodeProcess, sorted } from '../tools/utils';
 
 const EXPECTED_DEPENDENCIES = sorted(
@@ -139,6 +139,7 @@ describe('package.json', function () {
 
     beforeEach(async function () {
       for (const key of Object.keys(require.cache)) delete require.cache[key];
+      // Intentionally import from `src` because we are testing the package's exports (index.ts)
       // eslint-disable-next-line @typescript-eslint/no-require-imports
       require('../../src');
       imports = Array.from(

test/benchmarks/unitBench/list.bench.js

@@ -1,5 +1,5 @@
 const chalk = require('chalk');
-const { List } = require('../../../src/utils');
+const { List } = require('../../mongodb');
 const { createHistogram } = require('perf_hooks');
 const { process } = require('process');
 

test/csfle-kms-providers.ts

@@ -1,6 +1,6 @@
 import * as process from 'process';
 
-import { type KMSProviders } from './../src';
+import { type KMSProviders } from './mongodb';
 
 const csfleKMSProviders = {
   aws: {

test/integration/auth/auth.prose.test.ts

@@ -2,10 +2,7 @@ import { expect } from 'chai';
 import * as process from 'process';
 import * as sinon from 'sinon';
 
-import { type MongoClient } from '../../../src';
-import { ScramSHA256 } from '../../../src/cmap/auth/scram';
-import { Connection } from '../../../src/cmap/connection';
-import { LEGACY_HELLO_COMMAND } from '../../../src/constants';
+import { Connection, LEGACY_HELLO_COMMAND, type MongoClient, ScramSHA256 } from '../../mongodb';
 import { type TestConfiguration } from '../../tools/runner/config';
 
 function makeConnectionString(config, username, password) {