在打字稿中使用 jest 模擬 postgres 數據庫連接（Pool、PoolClient、pg）

Question

我有一個簡單的函數來從 Postgres 數據庫中獲取值。 我需要通過模擬數據庫來測試該方法。 我嘗試了提供的各種方法，但都沒有奏效。 總會出現一些錯誤。 我嘗試了以下解決方案：

• 如何使用 jest 在 node.js 中模擬 postgresql (pg)
• 如何使用 Sinon 模擬 pg 池
• 使用 ts-jest/utils 模擬數據庫 Jest
• 如何在不模擬 pg 導入的情況下使用 Node.js / Jest 測試 Postgres 如何使用 Jest 測試異步 await pg 連接？
• 如何使用 Jest 模擬 SQL Server 連接池？
• 如何使用 jest 在 node.js 中模擬 postgresql (pg)

下面是我的文件。

數據庫文件

import { Pool } from 'pg'

const pool = new Pool({
    user: `${process.env.POSTGRES_USER}`,
    database: `${process.env.POSTGRES_DB}`,
    password: `${process.env.POSTGRES_PASSWORD}`,
    port: `${process.env.POSTGRES_PORT}`,
    host: `${process.env.POSTGRES_HOST}`,

})
export default pool;

獲取.ts

import pool from "./db";
import {SO} from "./s2l";

    export async function fetch(code: string): Promise<SO[]>{
        let socalls: SO[] =[]
        const sql = 'Select sg.v1,sg.v2,sg.v3,sg.v4 from table1 sg where code = $1 and sg.r_code not in(\'ABC\', \'XYZ\', \'PQR\') order by sg.datetime asc'
        const values = [code]
        const client =  await pool.connect()
        await client.query(sql, values).then(res => {
            const data = res.rows;
            let so
            data.forEach(row => {
                so ={
                service: {
                    code: row["v1"]
                },
                rCode: row["v2"],
                dVo: row["v3"],
                aVo: row["v4"],
            };
            socalls.push(so)
        });
        }).catch(e => console.error(e))
        .then(() => client.release())
        return Promise.all(socalls);
        }

另外，我需要在查詢中動態傳遞列表。

Answer 1

以下方法對我有用

fetch.test.ts

import { fetch } from "./fetch";
import pool from "./db";

describe("Test fetch of so calls from database", () => {
  afterAll(() => {
    jest.resetAllMocks();
  });
  it("Happy case", async () => {
    (pool as any).connect = jest.fn().mockReturnThis();
    (pool as any).query = jest.fn().mockReturnThis();
    (pool as any).release = jest.fn().mockReturnThis();
    (pool as any).query.mockResolvedValueOnce({
      rows: [
        {
          v2: "EHRBFK76TGSMD",
          v3: "355GG",
          v4: "355GG",
          v1: "Q923892GT",
        },
      ],
    });
    const result = await fetchSiteCalls("Y2K");
    expect(result).toEqual([
      {
        rCode: "EHRBFK76TGSMD",
        dVo: "355GG",
        aVo: "355GG",
        service: { code: "Q923892GT" },
      },
    ]);
  });
});