mirror of
https://github.com/diegosouzapw/OmniRoute.git
synced 2026-07-31 04:12:10 +03:00
b7bd41942d
Move chat pipeline validation, circuit breaker execution, proxy resolution, logging, and session header handling into dedicated helpers to keep the SSE handler smaller and easier to verify. Also fix shared API option precedence, rebuild skill version caches after deletions, ignore api.trycloudflare.com false positives, and add rate-limit manager test flush/reset hooks for deterministic coverage. Expand integration and unit coverage across chat routing, auth, cloud sync, skills, executors, streaming, DB helpers, proxy handling, and provider/model utilities.
91 lines
2.9 KiB
JavaScript
91 lines
2.9 KiB
JavaScript
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);
|
|
});
|