typescript - Bagaimana cara menambahkan typehint untuk assertion chai custom?

Dalam proyek TypeScript, saya membuat klaim kustom untuk pustaka klaim chai seperti berikut:

// ./tests/assertions/assertTimestamp.ts
import moment = require("moment");
import {Moment} from "moment";

const {Assertion} = require("chai");

const parseDateWithTime = (dateWithTime: string, format = "YYYY-MM-DD hh:mm:ss"): Moment => {
    return moment.utc(dateWithTime, format);
};

const formatTimestamp = (timestampInMs: number): string => {
    return moment.utc(timestampInMs).toISOString();
};

Assertion.addMethod("timestampOf", function (humanReadableDate: string, format = "YYYY-MM-DD hh:mm:ss") {
        const expectedValue = parseDateWithTime(humanReadableDate, format);
        const formatted = formatTimestamp(this._obj);

        new Assertion(this._obj)
            .to.eq(
            expectedValue.valueOf(),
            `\nEXPECTED: "${expectedValue.toISOString()}"\nGOT:      "${formatted}"\n`,
        );
    },
);

Saya menggunakannya dengan cara ini:

require("../assertions/assertTimestamp");
import {expect} from "chai";


describe("My test", () => {
   it("should test timestamp", () => {
     expect(1610841600000).to.be.timestampOf("2021-01-16");
   });
});

Namun saya mendapatkan error TypeScript:

TS2339:Property 'timestampOf' does not exist on type 'Assertion'.

Saya tahu saya bisa mengatasi masalah ini melalui:

(expect(1610841600000).to.be as any).timestampOf("2021-01-16");

tapi saya lebih ingin mendaftarkan metode saya di dalam TypeScript, sehingga saya juga bisa memanfaatkan auto-completion di IDE saya.

Jawaban #1

Dalam assertTimestamp.ts Anda dapat menambahkan:

declare global {
    export namespace Chai {
        interface Assertion {
            timestampOf(date: string, format?: string): void;
        }
    }
}

itu akan menambahkan fungsi kustom Anda sambil mempertahankan fungsi yang sudah ada.

Jika Anda menjalankan tes Anda sekarang, itu akan gagal, sesuai yang diharapkan, dengan:

AssertionError: 
EXPECTED: "2021-01-16T00:00:00.000Z"
GOT:      "2021-01-17T00:00:00.000Z"
: expected 1610841600000 to equal 1610755200000
Expected :1610755200000
Actual   :1610841600000