feat: migrate send logs and discord links to D1

This commit is contained in:
RhenCloud 2026-08-03 03:57:58 +08:00
parent 5a98e45544
commit 3a6a2cbbc5
No known key found for this signature in database
GPG key ID: A574A617378C4E0B
16 changed files with 139 additions and 79 deletions

View file

@ -12,6 +12,7 @@ function createEnv(overrides: Partial<Env> = {}): Env {
return {
GITHUB_WEBHOOK_SECRET: "secret",
KV: {} as KVNamespace,
DB: {} as D1Database,
...overrides,
};
}
@ -91,6 +92,17 @@ describe("dispatchEvent fallback routing", () => {
} as unknown as KVNamespace;
}
function createMockDB(): D1Database {
return {
prepare: () => ({
bind: () => ({
run: async () => ({ success: true }),
all: async () => ({ results: [] }),
}),
}),
} as unknown as D1Database;
}
const baseConfig = {
baseUrl: "https://example.com",
github: {
@ -110,7 +122,7 @@ describe("dispatchEvent fallback routing", () => {
sent.push(url);
return new Response("{}", { status: 200 });
});
const env = createEnv({ KV: createMockKV() });
const env = createEnv({ KV: createMockKV(), DB: createMockDB() });
const routes: Route[] = [
{
id: "regular-push",

View file

@ -1,39 +1,46 @@
import { describe, it, expect } from "bun:test";
import { recordSend, getSendLog } from "../lib/send-log";
function createMockKV(): KVNamespace {
const store = new Map<string, { value: string; expiration?: number }>();
function createMockDB(): D1Database {
const rows: Array<Record<string, unknown>> = [];
return {
get: async (key: string, type?: string) => {
const entry = store.get(key);
if (!entry) return null;
if (entry.expiration && Date.now() / 1000 > entry.expiration) {
store.delete(key);
return null;
}
if (type === "json") return JSON.parse(entry.value);
return entry.value;
},
put: async (key: string, value: string, opts?: { expirationTtl?: number }) => {
const expiration = opts?.expirationTtl ? Date.now() / 1000 + opts.expirationTtl : undefined;
store.set(key, { value, expiration });
},
delete: async (key: string) => {
store.delete(key);
},
list: async () => ({
keys: [...store.keys()].map((k) => ({ name: k })),
list_complete: true,
cacheStatus: null,
prepare: (sql: string) => ({
bind: (..._args: unknown[]) => ({
run: async (): Promise<{ success: boolean }> => {
if (sql.startsWith("INSERT")) {
const args = _args as unknown[];
rows.push({
ts: args[0],
route_id: args[1],
event: args[2],
repo: args[3],
target: args[4],
ok: args[5],
error: args[6],
});
}
return { success: true };
},
all: async (): Promise<{ results: Array<Record<string, unknown>> }> => {
const args = _args as unknown[];
const limit = (args[0] as number) ?? 50;
return {
results: rows
.slice()
.sort((a, b) => (b.ts as number) - (a.ts as number))
.slice(0, limit),
};
},
}),
}),
} as unknown as KVNamespace;
} as unknown as D1Database;
}
describe("send-log", () => {
it("records and returns logs sorted newest first", async () => {
const kv = createMockKV();
await recordSend(kv, { ts: 1000, routeId: "a", event: "push", target: "111", ok: true });
await recordSend(kv, {
const db = createMockDB();
await recordSend(db, { ts: 1000, routeId: "a", event: "push", target: "111", ok: true });
await recordSend(db, {
ts: 2000,
routeId: "b",
event: "issues",
@ -41,7 +48,7 @@ describe("send-log", () => {
ok: false,
error: "Missing Permissions",
});
const logs = await getSendLog(kv);
const logs = await getSendLog(db);
expect(logs).toHaveLength(2);
expect(logs[0]!.routeId).toBe("b");
expect(logs[0]!.ok).toBe(false);
@ -50,7 +57,7 @@ describe("send-log", () => {
});
it("returns empty when no logs", async () => {
const kv = createMockKV();
expect(await getSendLog(kv)).toEqual([]);
const db = createMockDB();
expect(await getSendLog(db)).toEqual([]);
});
});