时间戳

首页 > 解决方案 > Javascript nodeJs Postgres pg 单元测试

javascript - Javascript nodeJs Postgres pg 单元测试

问题描述

我正在使用PG库连接到 Postgres DB，

现在，我想为数据库访问编写单元测试，但我不知道该怎么做。实际上，我需要一些模拟 Postgres 服务器或其他东西来模拟 Postgres。

我正在使用Mocha进行测试

下面是我访问数据库的课程之一

import { Pool } from "pg";
export class DbAccess implements IdbAccess {
  private static readonly postgres = new Pool();

  constructor(@inject(TYPES.ILogger) private readonly logger: ILogger) {}

  public saveConsumer(consumer: IAPIGWConsumer): Promise<IAPIGWConsumer> {
    return this.queryOne`
        INSERT INTO users (consumer_id, email)
        VALUES (${consumer.consumer_id}, ${consumer.email})
        ON CONFLICT (consumer_id) DO UPDATE SET email = ${consumer.email}
        RETURNING *
    `;
  }
}

我会很感激任何帮助，谢谢。

标签： javascriptnode.jspostgresqltypescriptmocha.js

解决方案

如果你想模拟或存根一些包/方法/模块，你需要安装一个模拟/存根库，例如sinon.js，jest.js来实现这一点。

这是使用sinon.js. 为了简单明了，我删除了不必要的部分，例如 DI。

db.ts：

import { Pool } from "pg";

export class DbAccess {
  private static readonly postgres = new Pool();

  public saveConsumer(consumer) {
    return DbAccess.postgres.query(`
        INSERT INTO users (consumer_id, email)
        VALUES (${consumer.consumer_id}, ${consumer.email})
        ON CONFLICT (consumer_id) DO UPDATE SET email = ${consumer.email}
        RETURNING *
    `);
  }
}

db.test.ts：

import pg from "pg";
import sinon from "sinon";
import { expect } from "chai";

describe("59624370", () => {
  afterEach(() => {
    sinon.restore();
  });
  it("should pass", async () => {
    const mPool = { query: sinon.stub().resolves({ rows: [] }) };
    const poolStub = sinon.stub(pg, "Pool").callsFake(() => mPool);
    const { DbAccess } = require("./db");
    const db = new DbAccess();
    const consumer = { consumer_id: 1, email: "example@gmail.com" };
    const actual = await db.saveConsumer(consumer);
    expect(actual).to.be.eql({ rows: [] });
    sinon.assert.calledOnce(poolStub);
    sinon.assert.calledOnce(mPool.query);
  });
});

带有覆盖率报告的单元测试结果：

 59624370
    ✓ should pass (132ms)


  1 passing (141ms)

------------|----------|----------|----------|----------|-------------------|
File        |  % Stmts | % Branch |  % Funcs |  % Lines | Uncovered Line #s |
------------|----------|----------|----------|----------|-------------------|
All files   |      100 |      100 |      100 |      100 |                   |
 db.test.ts |      100 |      100 |      100 |      100 |                   |
 db.ts      |      100 |      100 |      100 |      100 |                   |
------------|----------|----------|----------|----------|-------------------|

源代码：https ://github.com/mrdulin/mocha-chai-sinon-codelab/tree/master/src/stackoverflow/59624370

推荐阅读