Files
2026-07-13 13:39:12 +08:00

91 lines
2.9 KiB
TypeScript

import test from "node:test";
import assert from "node:assert/strict";
import fs from "node:fs";
import os from "node:os";
import path from "node:path";
const TEST_DATA_DIR = fs.mkdtempSync(path.join(os.tmpdir(), "omni-db-webhooks-"));
process.env.DATA_DIR = TEST_DATA_DIR;
const coreDb = await import("../../src/lib/db/core.ts");
const webhooksDb = await import("../../src/lib/db/webhooks.ts");
async function resetStorage() {
coreDb.resetDbInstance();
fs.rmSync(TEST_DATA_DIR, { recursive: true, force: true });
fs.mkdirSync(TEST_DATA_DIR, { recursive: true });
}
test.beforeEach(async () => {
await resetStorage();
});
test.after(() => {
coreDb.resetDbInstance();
fs.rmSync(TEST_DATA_DIR, { recursive: true, force: true });
});
test("webhooks create, update, query enabled hooks and delete records", () => {
const created = webhooksDb.createWebhook({
url: "https://example.com/hook",
events: ["request.completed"],
description: "Primary webhook",
});
assert.match(created.secret, /^whsec_/);
assert.equal(webhooksDb.getWebhooks().length, 1);
assert.equal(webhooksDb.getEnabledWebhooks().length, 1);
const updated = webhooksDb.updateWebhook(created.id, {
enabled: false,
events: ["request.failed"],
secret: "custom-secret",
description: "Updated webhook",
});
assert.equal(updated.enabled, false);
assert.deepEqual(updated.events, ["request.failed"]);
assert.equal(updated.secret, "custom-secret");
assert.equal(webhooksDb.getEnabledWebhooks().length, 0);
assert.equal(webhooksDb.deleteWebhook(created.id), true);
assert.equal(webhooksDb.deleteWebhook(created.id), false);
});
test("webhooks record delivery success and failures", () => {
const created = webhooksDb.createWebhook({
url: "https://example.com/hook",
});
webhooksDb.recordWebhookDelivery(created.id, 500, false);
webhooksDb.recordWebhookDelivery(created.id, 502, false);
let stored = webhooksDb.getWebhook(created.id);
assert.equal(stored.failure_count, 2);
assert.equal(stored.last_status, 502);
webhooksDb.recordWebhookDelivery(created.id, 200, true);
stored = webhooksDb.getWebhook(created.id);
assert.equal(stored.failure_count, 0);
assert.equal(stored.last_status, 200);
assert.ok(stored.last_triggered_at);
});
test("webhooks disable only hooks above the failure threshold", () => {
const a = webhooksDb.createWebhook({ url: "https://example.com/a" });
const b = webhooksDb.createWebhook({ url: "https://example.com/b" });
for (let i = 0; i < 3; i++) {
webhooksDb.recordWebhookDelivery(a.id, 500, false);
}
webhooksDb.recordWebhookDelivery(b.id, 500, false);
const disabled = webhooksDb.disableWebhooksWithHighFailures(2);
assert.equal(disabled, 1);
assert.equal(webhooksDb.getWebhook(a.id).enabled, false);
assert.equal(webhooksDb.getWebhook(b.id).enabled, true);
assert.equal(webhooksDb.updateWebhook("ghost", { enabled: false }), null);
});