jest.mock() 不起作用，测试仍在从节点模块中提取代码

Question

I am trying to mock the @databricks/sql package using the jest.mock function and it continually pulls from the node modules package instead of the code I pass into the function.

我的测试文件：

import mockResponse from "./mockResponse.json";

const {
  handler,
} = require("../../../backend/function/dataLakeConnect/src/index");

const testEvent = {
  arguments: {
    page: 1,
    clientId: "test",
    startTime: "2022-09-26 00:00:00",
    endTime: "2022-09-27 00:00:00",
  },
};

const exampleResponse = {
  ARN: "x",
  Name: "credentials/databricks/read-write",
  VersionId: "x",
  SecretString:
    '{"DATABRICKS_TOKEN":"test","DATABRICKS_SERVER_HOSTNAME":"test.test.com","DATABRICKS_HTTP_PATH":"/pathy/pathpath/mcgee"}',
  VersionStages: ["x"],
  CreatedDate: "x",
};

jest.mock("@databricks/sql", () => {
  return {
    DBSQLClient: function () {
      return {
        connect: function () {
          return {
            promise: function () {},
          };
        },
        openSession: function () {
          console.log("openSession called");
          return {
            promise: function () {
              return {
                executeStatement: function (sql, params) {
                  return {
                    promise: function () {
                      return {
                        fetchAll: function () {
                          return {
                            promise: function () {
                              return mockResponse;
                            },
                          };
                        },
                      };
                    },
                  };
                },
              };
            },
          };
        },
      };
    },
  };
});

jest.mock("aws-sdk", () => {
  return {
    SecretsManager: function () {
      return {
        getSecretValue: function ({ SecretId }) {
          if (SecretId === "credentials/databricks/read-write") {
            return {
              promise: function () {
                return exampleResponse;
              },
            };
          } else {
            throw new Error("mock error");
          }
        },
      };
    },
  };
});


describe("dataLakeConnect", () => {
  test("should return expected array", async () => {
    let res = await handler(testEvent);
    expect(res).toEqual(mockResponse);
  });
});

我正在尝试测试的 function：

const AWS = require("aws-sdk");
const { DBSQLClient } = require("@databricks/sql");
const client = new AWS.SecretsManager({
  region: "us-east-1",
});
console.log(AWS);
console.log(DBSQLClient);
const databricksClient = new DBSQLClient();
const resultsPerPage = 300;

const getSecret = async (secretName) => {
  try {
    let data = await client.getSecretValue({ SecretId: secretName }).promise();
    if (data.SecretString) {
      return JSON.parse(data.SecretString);
    }
  } catch (err) {
    throw err;
  }
};

exports.handler = async (event) => {
  console.log(`EVENT: ${JSON.stringify(event)}`);
  const { page, clientId, startTime, endTime } = event.arguments;
  const offset = resultsPerPage * (page - 1);
  let sql = `SELECT * FROM test`;

  try {
    const creds = await getSecret("credentials/databricks/read-write");
    // console.log(databricksClient);
    await databricksClient.connect({
      token: creds.DATABRICKS_TOKEN,
      host: creds.DATABRICKS_SERVER_HOSTNAME,
      path: creds.DATABRICKS_HTTP_PATH,
    });
    const session = await databricksClient.openSession();
    const queryOperation = await session.executeStatement(sql, {
      runAsync: true,
      maxRows: 10000,
    });
    const result = await queryOperation.fetchAll({
      progress: false,
      callback: () => {},
    });

    await queryOperation.close();
    await session.close();
    await databricksClient.close();
    return result;
  } catch (err) {
    throw err;
  }
};

我希望此文件中的控制台日志返回模拟代码，并且aws-sdk模拟似乎工作正常，但是， @databricks/sql控制台日志返回原始模块，这会导致测试超时。

我已经尝试将这些模拟放在节点模块旁边的__mocks__目录，就像 jest 建议的那样，但我最终得到错误，说 jest 找不到 package。

关于如何真正让@databricks/sql模拟工作的任何建议？ 我很困惑。

Answer 1

使用您的代码，模拟实际上（从问题中复制了代码）可以工作，但我得到：

dataLakeConnect › should return expected array

    TypeError: session.executeStatement is not a function

      35 |     });
      36 |     const session = await databricksClient.openSession();
    > 37 |     const queryOperation = await session.executeStatement(sql, {
         |                                          ^
      38 |       runAsync: true,
      39 |       maxRows: 10000,
      40 |     });

将模拟更改为：

jest.mock("@databricks/sql", () => ({
  DBSQLClient: function () {
    return {
      connect: () => Promise.resolve({}),
      openSession: () => {
        console.log("openSession called");
        return Promise.resolve({
          executeStatement: () =>
            Promise.resolve({
              fetchAll: () => Promise.resolve(mockResponse),
              close: () => Promise.resolve({}),
            }),
          close: () => Promise.resolve({}),
        });
      },
      close: () => Promise.resolve({}),
    };
  },
}));

测试通过：

 ● Console

    console.log
      { SecretsManager: [Function: SecretsManager] }

      at Object.<anonymous> (src/totest.js:6:9)

    console.log
      [Function: DBSQLClient]

      at Object.<anonymous> (src/totest.js:7:9)

    console.log
      EVENT: {"arguments":{"page":1,"clientId":"test","startTime":"2022-09-26 00:00:00","endTime":"2022-09-27 00:00:00"}}

      at handler (src/totest.js:23:11)

    console.log
      openSession called

      at Object.openSession (src/__tests__/totest.spec.js:54:17)