All files / plainweb/src isolate.ts

65.38% Statements 17/26
25% Branches 1/4
100% Functions 1/1
65.38% Lines 17/26

Press n or j to go to the next uncovered block, b, p or k for the previous block.

1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 351x     2x 2x 2x 2x 2x       2x   2x   2x 2x 2x 2x     2x 2x 2x   2x               2x  
import { sql } from "drizzle-orm";
import type { BetterSQLite3Database } from "drizzle-orm/better-sqlite3";
 
export async function isolate<Schema extends Record<string, unknown>>(
  db: BetterSQLite3Database<Schema>,
  fn: (db: BetterSQLite3Database<Schema>) => Promise<void>,
) {
  if (process.env.NODE_ENV !== "test") {
    throw new Error("Make sure NODE_ENV=test is set when running tests");
  }
  // TODO check if pending migrations, and print warning if so
  let err: Error | null = null;
 
  try {
    // Begin the transaction
    db.run(sql.raw("BEGIN"));
    try {
      await fn(db);
    } catch (e) {
      err = e as Error;
    }
  } finally {
    db.run(sql.raw("ROLLBACK"));
  }
 
  if (err) {
    // rethrow the error with the original error attached
    const e = new Error(`Rethrowing error: "${err.message}"`);
    // @ts-ignore
    e.original_error = err;
    e.stack = err.stack;
    throw e;
  }
}