43 lines
1.1 KiB
TypeScript
43 lines
1.1 KiB
TypeScript
import { vi } from "vitest";
|
|
import { SystemRole } from "@capakraken/shared";
|
|
import type { ToolContext } from "../router/assistant-tools.js";
|
|
|
|
export function createToolContext(
|
|
db: Record<string, unknown>,
|
|
userRole: SystemRole = SystemRole.USER,
|
|
): ToolContext {
|
|
const dbWithTransaction = "$transaction" in db
|
|
? db
|
|
: {
|
|
...db,
|
|
$transaction: async <T>(callback: (tx: ToolContext["db"]) => Promise<T>) =>
|
|
callback(db as ToolContext["db"]),
|
|
};
|
|
|
|
return {
|
|
db: dbWithTransaction as ToolContext["db"],
|
|
userId: "user_1",
|
|
userRole,
|
|
permissions: new Set(),
|
|
session: {
|
|
user: { email: "user@example.com", name: "Assistant User", image: null },
|
|
expires: "2026-03-29T00:00:00.000Z",
|
|
},
|
|
dbUser: {
|
|
id: "user_1",
|
|
systemRole: userRole,
|
|
permissionOverrides: null,
|
|
},
|
|
roleDefaults: null,
|
|
};
|
|
}
|
|
|
|
export function withUserLookup(db: Record<string, unknown>, userId = "user_1") {
|
|
return {
|
|
user: {
|
|
findUnique: vi.fn().mockResolvedValue({ id: userId }),
|
|
},
|
|
...db,
|
|
};
|
|
}
|