如何使用jest在node.js中模拟postgresql (pg)

npm install --save-dev jest pg

// 在测试文件的顶部引入pg和jest的mock功能
const pg = require('pg');
jest.mock('pg');

// 创建模拟的PostgreSQL客户端
const client = new pg.Client();

// 模拟查询的结果
const queryResult = { rows: [{ id: 1, name: 'John' }] };
pg.Client.prototype.query.mockImplementation((query, params, callback) => {
  // 回调函数中返回模拟的查询结果
  callback(null, queryResult);
});

// 模拟连接
pg.Client.prototype.connect.mockImplementation((callback) => {
  // 回调函数中不做任何操作，表示成功连接
  callback(null);
});

test('测试查询数据', async () => {
  // 模拟查询
  const result = await client.query('SELECT * FROM users');

  // 断言结果是否符合预期
  expect(result.rows.length).toBe(1);
  expect(result.rows[0].name).toBe('John');
});

jest