1 files changed, 11 insertions, 0 deletions
diff --git a/packages/db/drizzle.ts b/packages/db/drizzle.ts
index def1fc0a..adfe4884 100644
--- a/packages/db/drizzle.ts
+++ b/packages/db/drizzle.ts
@@ -2,6 +2,17 @@ import "dotenv/config";
 import { drizzle } from "drizzle-orm/better-sqlite3";
 import Database from "better-sqlite3";
 import * as schema from "./schema";
+import { migrate } from "drizzle-orm/better-sqlite3/migrator";
+import path from "path";
 
 const sqlite = new Database(process.env.DATABASE_URL);
 export const db = drizzle(sqlite, { schema, logger: true });
+
+export function getInMemoryDB(runMigrations: boolean) {
+  const mem = new Database(":memory:");
+  const db = drizzle(mem, { schema, logger: true });
+  if (runMigrations) {
+    migrate(db, { migrationsFolder: path.resolve(__dirname, "./drizzle") });
+  }
+  return db;
+}