@geekmidas/testkit 10.0.0-alpha.12 → 10.0.0-alpha.120

package/dist/transactions.cjs

@@ -0,0 +1,239 @@
+Object.defineProperty(exports, Symbol.toStringTag, { value: "Module" });
+const require_rolldown_runtime = require("./rolldown-runtime-VH7oDXx4.cjs");
+let pg = require("pg");
+pg = require_rolldown_runtime.__toESM(pg, 1);
+let kysely = require("kysely");
+//#region src/transactions.ts
+/**
+* A transaction per database, per test — as each runs deployed.
+*
+* An app's database and a schema tenant such as the auth server's are separate
+* databases to the code using them: separate connections, credentials, roles
+* and search paths. A test keeps them that way. Each gets its own connection,
+* opened into a transaction the first time the test touches it, and every one
+* is rolled back when the test ends.
+*
+* Code under test must be free to use transactions itself, and better-auth
+* does — it also calls `db.connection()`, which Kysely refuses on a
+* `Transaction`. So what a test hands over is not a `Transaction` but a Kysely
+* over the one connection already inside the test's transaction, on which a
+* `BEGIN` becomes a savepoint and a `COMMIT` releases it. Nothing the code does
+* can end the transaction the test will roll back.
+*
+* A statement the code runs outside a transaction of its own gets a savepoint
+* too, so a failure — a duplicate key two concurrent requests raced to insert —
+* fails that statement alone, as it would deployed, rather than every request
+* the test makes after it.
+*
+* A savepoint the test or the code opens by name — `savepoint refused`, a
+* statement expected to fail, `rollback to savepoint refused` — is its own
+* transaction control, and passes through untouched. While one is open,
+* statements run in it as they would inside a `begin`: unwrapped, so a failure
+* aborts the transaction until the matching `rollback to savepoint`.
+*/
+/**
+* The connection settings for a database URL.
+*
+* `?search_path=` is not a libpq parameter: a URL carrying it connects and then
+* resolves every name against `public`. Postgres takes it as a startup option,
+* which is what a schema tenant's URL needs to reach its own tables.
+*/
+function connectionConfig(url) {
+	const parsed = new URL(url);
+	const searchPath = parsed.searchParams.get("search_path");
+	if (!searchPath) return { connectionString: url };
+	parsed.searchParams.delete("search_path");
+	return {
+		connectionString: parsed.toString(),
+		options: `-c search_path=${searchPath}`
+	};
+}
+/**
+* Open a connection to `url` and begin the transaction a test will roll back.
+*
+* `config` is what the database's own clients are built with — its plugins
+* above all. The code under test is handed this Kysely in place of its own, so
+* one built without them runs different queries than production does.
+*/
+async function openBoundTransaction(url, config = {}) {
+	const client = new pg.default.Client(connectionConfig(url));
+	await client.connect();
+	await client.query("BEGIN");
+	const inTurn = turns();
+	const named = [];
+	return {
+		db: new kysely.Kysely({
+			...config,
+			dialect: new kysely.PostgresDialect({ pool: {
+				connect: async () => savepointing(client, inTurn, named),
+				end: async () => {}
+			} })
+		}),
+		async rollback() {
+			try {
+				await client.query("ROLLBACK");
+			} finally {
+				await client.end();
+			}
+		}
+	};
+}
+/**
+* The client, with transaction control turned into savepoints.
+*
+* Kysely issues `begin`/`commit`/`rollback` as plain statements; this is the
+* one place they can be caught. A depth counter names the savepoints, so
+* nested transactions nest.
+*/
+function savepointing(client, inTurn, named) {
+	let depth = 0;
+	const query = client.query.bind(client);
+	const wrapped = Object.create(client);
+	const closeTransaction = (level) => {
+		for (let i = named.length - 1; i >= 0; i--) {
+			const savepoint = named[i];
+			if (savepoint.checkout === wrapped && savepoint.depth >= level) named.splice(i, 1);
+		}
+	};
+	Object.assign(wrapped, {
+		query: (text, ...rest) => {
+			if (typeof text !== "string" || typeof rest.at(-1) === "function") return query(text, ...rest);
+			const statement = text.trim().toLowerCase();
+			if (statement === "begin" || statement.startsWith("start transaction")) {
+				const savepoint = `test_sp_${++depth}`;
+				return inTurn(() => query(`SAVEPOINT ${savepoint}`));
+			}
+			if (statement === "commit") {
+				const level = depth--;
+				closeTransaction(level);
+				return inTurn(() => query(`RELEASE SAVEPOINT test_sp_${level}`));
+			}
+			if (statement === "rollback") {
+				const level = depth--;
+				closeTransaction(level);
+				return inTurn(() => query(`ROLLBACK TO SAVEPOINT test_sp_${level}`));
+			}
+			const control = savepointControl(text);
+			if (control) return inTurn(async () => {
+				const result = await query(text, ...rest);
+				track(named, control, {
+					checkout: wrapped,
+					depth
+				});
+				return result;
+			});
+			if (depth > 0 || named.length > 0) return inTurn(() => query(text, ...rest));
+			return inTurn(() => autocommitted(query, text, rest));
+		},
+		release: () => {}
+	});
+	return wrapped;
+}
+const IDENTIFIER = String.raw`("(?:[^"]|"")+"|[^\s;"]+)`;
+const SAVEPOINT_CONTROL = [
+	["savepoint", new RegExp(String.raw`^savepoint\s+${IDENTIFIER}\s*;?$`, "i")],
+	["release", new RegExp(String.raw`^release\s+(?:savepoint\s+)?${IDENTIFIER}\s*;?$`, "i")],
+	["rollback to", new RegExp(String.raw`^rollback\s+(?:(?:work|transaction)\s+)?to\s+(?:savepoint\s+)?${IDENTIFIER}\s*;?$`, "i")]
+];
+/**
+* The savepoint a statement opens, releases or rolls back to, named as
+* Postgres names it: an unquoted name folded to lower case, a quoted one as
+* written.
+*/
+function savepointControl(text) {
+	const statement = text.trim();
+	for (const [action, pattern] of SAVEPOINT_CONTROL) {
+		const match = pattern.exec(statement);
+		if (!match) continue;
+		const identifier = match[1];
+		return {
+			action,
+			name: identifier.startsWith("\"") ? identifier.slice(1, -1).replaceAll("\"\"", "\"") : identifier.toLowerCase()
+		};
+	}
+}
+/**
+* Keep the named savepoints as Postgres does: `RELEASE x` ends x and every
+* savepoint after it; `ROLLBACK TO x` ends those after it and keeps x open.
+*/
+function track(named, control, at) {
+	if (control.action === "savepoint") {
+		named.push({
+			name: control.name,
+			...at
+		});
+		return;
+	}
+	const index = named.findLastIndex((s) => s.name === control.name);
+	if (index === -1) return;
+	named.splice(control.action === "release" ? index : index + 1);
+}
+/**
+* One statement at a time on a connection, in the order they were asked for.
+*
+* The connection runs them one at a time anyway; this keeps a statement and
+* the savepoint around it together, so nothing another request sends lands
+* between them — a `RELEASE` also ends every savepoint opened after its own.
+*/
+function turns() {
+	let queue = Promise.resolve();
+	return (work) => {
+		const turn = queue.then(work, work);
+		queue = turn.catch(() => {});
+		return turn;
+	};
+}
+/**
+* A statement run outside any transaction the code opened, failing the way
+* it would deployed: on its own.
+*
+* Deployed, such a statement is its own transaction, so one that fails — a
+* duplicate key two concurrent requests both tried to insert — fails alone,
+* and code written for that (Better Auth's rate limiter reads the row the
+* other request inserted, and carries on) does carry on. Inside the test's
+* transaction a failed statement would abort everything after it, every
+* request of the test included. A savepoint around it puts back what
+* deployed would have.
+*/
+async function autocommitted(query, text, rest) {
+	await query("SAVEPOINT test_statement");
+	try {
+		const result = await query(text, ...rest);
+		await query("RELEASE SAVEPOINT test_statement");
+		return result;
+	} catch (error) {
+		await query("ROLLBACK TO SAVEPOINT test_statement");
+		await query("RELEASE SAVEPOINT test_statement");
+		throw error;
+	}
+}
+/**
+* The transactions one test has opened, by database.
+*
+* `get` opens a database's transaction the first time it is asked for and
+* hands back the same one after that — including to a caller that asks while
+* it is still opening. `rollbackAll` ends every one of them.
+*/
+var TransactionRegistry = class {
+	open = /* @__PURE__ */ new Map();
+	get(key, url, config) {
+		let transaction = this.open.get(key);
+		if (!transaction) {
+			transaction = openBoundTransaction(url, config);
+			this.open.set(key, transaction);
+		}
+		return transaction.then(({ db }) => db);
+	}
+	async rollbackAll() {
+		const transactions = [...this.open.values()];
+		this.open.clear();
+		const failed = (await Promise.allSettled(transactions.map(async (transaction) => (await transaction).rollback()))).find((result) => result.status === "rejected");
+		if (failed) throw failed.reason;
+	}
+};
+//#endregion
+exports.TransactionRegistry = TransactionRegistry;
+exports.connectionConfig = connectionConfig;
+exports.openBoundTransaction = openBoundTransaction;
+
+//# sourceMappingURL=transactions.cjs.map

package/dist/transactions.cjs.map

@@ -0,0 +1 @@
+{"version":3,"file":"transactions.cjs","names":["Kysely","PostgresDialect"],"sources":["../src/transactions.ts"],"sourcesContent":["/**\n * A transaction per database, per test — as each runs deployed.\n *\n * An app's database and a schema tenant such as the auth server's are separate\n * databases to the code using them: separate connections, credentials, roles\n * and search paths. A test keeps them that way. Each gets its own connection,\n * opened into a transaction the first time the test touches it, and every one\n * is rolled back when the test ends.\n *\n * Code under test must be free to use transactions itself, and better-auth\n * does — it also calls `db.connection()`, which Kysely refuses on a\n * `Transaction`. So what a test hands over is not a `Transaction` but a Kysely\n * over the one connection already inside the test's transaction, on which a\n * `BEGIN` becomes a savepoint and a `COMMIT` releases it. Nothing the code does\n * can end the transaction the test will roll back.\n *\n * A statement the code runs outside a transaction of its own gets a savepoint\n * too, so a failure — a duplicate key two concurrent requests raced to insert —\n * fails that statement alone, as it would deployed, rather than every request\n * the test makes after it.\n *\n * A savepoint the test or the code opens by name — `savepoint refused`, a\n * statement expected to fail, `rollback to savepoint refused` — is its own\n * transaction control, and passes through untouched. While one is open,\n * statements run in it as they would inside a `begin`: unwrapped, so a failure\n * aborts the transaction until the matching `rollback to savepoint`.\n */\n\nimport { Kysely, type KyselyConfig, PostgresDialect } from 'kysely';\nimport pg from 'pg';\n\n/** How to reach one database: a URL, and its search path as a startup option. */\nexport interface ConnectionConfig {\n\tconnectionString: string;\n\toptions?: string;\n}\n\n/**\n * The connection settings for a database URL.\n *\n * `?search_path=` is not a libpq parameter: a URL carrying it connects and then\n * resolves every name against `public`. Postgres takes it as a startup option,\n * which is what a schema tenant's URL needs to reach its own tables.\n */\nexport function connectionConfig(url: string): ConnectionConfig {\n\tconst parsed = new URL(url);\n\tconst searchPath = parsed.searchParams.get('search_path');\n\tif (!searchPath) return { connectionString: url };\n\n\tparsed.searchParams.delete('search_path');\n\treturn {\n\t\tconnectionString: parsed.toString(),\n\t\toptions: `-c search_path=${searchPath}`,\n\t};\n}\n\n/** One database's connection, inside a transaction the test will roll back. */\nexport interface BoundTransaction {\n\t/** Everything queried through this runs inside the test's transaction. */\n\treadonly db: Kysely<any>;\n\t/** Undo everything, and let the connection go. */\n\trollback(): Promise<void>;\n}\n\n/**\n * Open a connection to `url` and begin the transaction a test will roll back.\n *\n * `config` is what the database's own clients are built with — its plugins\n * above all. The code under test is handed this Kysely in place of its own, so\n * one built without them runs different queries than production does.\n */\nexport async function openBoundTransaction(\n\turl: string,\n\tconfig: Omit<KyselyConfig, 'dialect'> = {},\n): Promise<BoundTransaction> {\n\tconst client = new pg.Client(connectionConfig(url));\n\tawait client.connect();\n\tawait client.query('BEGIN');\n\t// Shared by every checkout of the connection: concurrent requests in one\n\t// test each check it out, and all of them take turns on it.\n\tconst inTurn = turns();\n\t// The savepoints opened by name, shared by every checkout as the\n\t// connection's transaction is.\n\tconst named: NamedSavepoint[] = [];\n\n\treturn {\n\t\tdb: new Kysely({\n\t\t\t...config,\n\t\t\tdialect: new PostgresDialect({\n\t\t\t\t// A \"pool\" of the one connection. Released never, ended never: the\n\t\t\t\t// test owns it, and it closes in `rollback`.\n\t\t\t\tpool: {\n\t\t\t\t\tconnect: async () => savepointing(client, inTurn, named),\n\t\t\t\t\tend: async () => {},\n\t\t\t\t} as unknown as pg.Pool,\n\t\t\t}),\n\t\t}),\n\t\tasync rollback() {\n\t\t\ttry {\n\t\t\t\tawait client.query('ROLLBACK');\n\t\t\t} finally {\n\t\t\t\tawait client.end();\n\t\t\t}\n\t\t},\n\t};\n}\n\n/**\n * The client, with transaction control turned into savepoints.\n *\n * Kysely issues `begin`/`commit`/`rollback` as plain statements; this is the\n * one place they can be caught. A depth counter names the savepoints, so\n * nested transactions nest.\n */\nfunction savepointing(\n\tclient: pg.Client,\n\tinTurn: Turns,\n\tnamed: NamedSavepoint[],\n): pg.PoolClient {\n\tlet depth = 0;\n\tconst query = client.query.bind(client) as (\n\t\t...args: unknown[]\n\t) => Promise<unknown>;\n\tconst wrapped = Object.create(client) as pg.PoolClient;\n\n\t// Ending one of the code's transactions ends every savepoint opened in it.\n\tconst closeTransaction = (level: number) => {\n\t\tfor (let i = named.length - 1; i >= 0; i--) {\n\t\t\tconst savepoint = named[i]!;\n\t\t\tif (savepoint.checkout === wrapped && savepoint.depth >= level) {\n\t\t\t\tnamed.splice(i, 1);\n\t\t\t}\n\t\t}\n\t};\n\n\tObject.assign(wrapped, {\n\t\tquery: (text: unknown, ...rest: unknown[]) => {\n\t\t\t// A cursor, a query config, a callback: not what Kysely sends for a\n\t\t\t// statement, and nothing to wrap.\n\t\t\tif (typeof text !== 'string' || typeof rest.at(-1) === 'function') {\n\t\t\t\treturn query(text, ...rest);\n\t\t\t}\n\t\t\tconst statement = text.trim().toLowerCase();\n\n\t\t\tif (statement === 'begin' || statement.startsWith('start transaction')) {\n\t\t\t\tconst savepoint = `test_sp_${++depth}`;\n\t\t\t\treturn inTurn(() => query(`SAVEPOINT ${savepoint}`));\n\t\t\t}\n\t\t\tif (statement === 'commit') {\n\t\t\t\tconst level = depth--;\n\t\t\t\tcloseTransaction(level);\n\t\t\t\treturn inTurn(() => query(`RELEASE SAVEPOINT test_sp_${level}`));\n\t\t\t}\n\t\t\tif (statement === 'rollback') {\n\t\t\t\tconst level = depth--;\n\t\t\t\tcloseTransaction(level);\n\t\t\t\treturn inTurn(() => query(`ROLLBACK TO SAVEPOINT test_sp_${level}`));\n\t\t\t}\n\n\t\t\tconst control = savepointControl(text);\n\t\t\tif (control) {\n\t\t\t\treturn inTurn(async () => {\n\t\t\t\t\tconst result = await query(text, ...rest);\n\t\t\t\t\ttrack(named, control, { checkout: wrapped, depth });\n\t\t\t\t\treturn result;\n\t\t\t\t});\n\t\t\t}\n\t\t\tif (depth > 0 || named.length > 0) {\n\t\t\t\treturn inTurn(() => query(text, ...rest));\n\t\t\t}\n\t\t\treturn inTurn(() => autocommitted(query, text, rest));\n\t\t},\n\t\trelease: () => {},\n\t});\n\treturn wrapped;\n}\n\n/** A savepoint the test or the code opened by name, and where it opened it. */\ninterface NamedSavepoint {\n\tname: string;\n\t/** The checkout, and its transaction depth, the savepoint was opened at. */\n\tcheckout: pg.PoolClient;\n\tdepth: number;\n}\n\n/** `SAVEPOINT x`, `RELEASE [SAVEPOINT] x`, `ROLLBACK [WORK] TO [SAVEPOINT] x`. */\ninterface SavepointControl {\n\taction: 'savepoint' | 'release' | 'rollback to';\n\tname: string;\n}\n\nconst IDENTIFIER = String.raw`(\"(?:[^\"]|\"\")+\"|[^\\s;\"]+)`;\nconst SAVEPOINT_CONTROL: [SavepointControl['action'], RegExp][] = [\n\t['savepoint', new RegExp(String.raw`^savepoint\\s+${IDENTIFIER}\\s*;?$`, 'i')],\n\t[\n\t\t'release',\n\t\tnew RegExp(\n\t\t\tString.raw`^release\\s+(?:savepoint\\s+)?${IDENTIFIER}\\s*;?$`,\n\t\t\t'i',\n\t\t),\n\t],\n\t[\n\t\t'rollback to',\n\t\tnew RegExp(\n\t\t\tString.raw`^rollback\\s+(?:(?:work|transaction)\\s+)?to\\s+(?:savepoint\\s+)?${IDENTIFIER}\\s*;?$`,\n\t\t\t'i',\n\t\t),\n\t],\n];\n\n/**\n * The savepoint a statement opens, releases or rolls back to, named as\n * Postgres names it: an unquoted name folded to lower case, a quoted one as\n * written.\n */\nfunction savepointControl(text: string): SavepointControl | undefined {\n\tconst statement = text.trim();\n\tfor (const [action, pattern] of SAVEPOINT_CONTROL) {\n\t\tconst match = pattern.exec(statement);\n\t\tif (!match) continue;\n\t\tconst identifier = match[1]!;\n\t\tconst name = identifier.startsWith('\"')\n\t\t\t? identifier.slice(1, -1).replaceAll('\"\"', '\"')\n\t\t\t: identifier.toLowerCase();\n\t\treturn { action, name };\n\t}\n\treturn undefined;\n}\n\n/**\n * Keep the named savepoints as Postgres does: `RELEASE x` ends x and every\n * savepoint after it; `ROLLBACK TO x` ends those after it and keeps x open.\n */\nfunction track(\n\tnamed: NamedSavepoint[],\n\tcontrol: SavepointControl,\n\tat: Omit<NamedSavepoint, 'name'>,\n): void {\n\tif (control.action === 'savepoint') {\n\t\tnamed.push({ name: control.name, ...at });\n\t\treturn;\n\t}\n\tconst index = named.findLastIndex((s) => s.name === control.name);\n\t// One opened before this connection was handed over, or by a name the\n\t// wrapper uses itself: nothing tracked to end.\n\tif (index === -1) return;\n\tnamed.splice(control.action === 'release' ? index : index + 1);\n}\n\n/** Runs work in the order it was asked for, each after the last has settled. */\ntype Turns = <T>(work: () => Promise<T>) => Promise<T>;\n\n/**\n * One statement at a time on a connection, in the order they were asked for.\n *\n * The connection runs them one at a time anyway; this keeps a statement and\n * the savepoint around it together, so nothing another request sends lands\n * between them — a `RELEASE` also ends every savepoint opened after its own.\n */\nfunction turns(): Turns {\n\tlet queue: Promise<unknown> = Promise.resolve();\n\treturn (work) => {\n\t\tconst turn = queue.then(work, work);\n\t\tqueue = turn.catch(() => {});\n\t\treturn turn;\n\t};\n}\n\n/**\n * A statement run outside any transaction the code opened, failing the way\n * it would deployed: on its own.\n *\n * Deployed, such a statement is its own transaction, so one that fails — a\n * duplicate key two concurrent requests both tried to insert — fails alone,\n * and code written for that (Better Auth's rate limiter reads the row the\n * other request inserted, and carries on) does carry on. Inside the test's\n * transaction a failed statement would abort everything after it, every\n * request of the test included. A savepoint around it puts back what\n * deployed would have.\n */\nasync function autocommitted(\n\tquery: (...args: unknown[]) => Promise<unknown>,\n\ttext: string,\n\trest: unknown[],\n): Promise<unknown> {\n\tawait query('SAVEPOINT test_statement');\n\ttry {\n\t\tconst result = await query(text, ...rest);\n\t\tawait query('RELEASE SAVEPOINT test_statement');\n\t\treturn result;\n\t} catch (error) {\n\t\tawait query('ROLLBACK TO SAVEPOINT test_statement');\n\t\tawait query('RELEASE SAVEPOINT test_statement');\n\t\tthrow error;\n\t}\n}\n\n/**\n * The transactions one test has opened, by database.\n *\n * `get` opens a database's transaction the first time it is asked for and\n * hands back the same one after that — including to a caller that asks while\n * it is still opening. `rollbackAll` ends every one of them.\n */\nexport class TransactionRegistry<TKey = string> {\n\tprivate readonly open = new Map<TKey, Promise<BoundTransaction>>();\n\n\tget(\n\t\tkey: TKey,\n\t\turl: string,\n\t\tconfig?: Omit<KyselyConfig, 'dialect'>,\n\t): Promise<Kysely<any>> {\n\t\tlet transaction = this.open.get(key);\n\t\tif (!transaction) {\n\t\t\ttransaction = openBoundTransaction(url, config);\n\t\t\tthis.open.set(key, transaction);\n\t\t}\n\t\treturn transaction.then(({ db }) => db);\n\t}\n\n\tasync rollbackAll(): Promise<void> {\n\t\tconst transactions = [...this.open.values()];\n\t\tthis.open.clear();\n\n\t\tconst results = await Promise.allSettled(\n\t\t\ttransactions.map(async (transaction) => (await transaction).rollback()),\n\t\t);\n\t\tconst failed = results.find(\n\t\t\t(result): result is PromiseRejectedResult => result.status === 'rejected',\n\t\t);\n\t\tif (failed) throw failed.reason;\n\t}\n}\n"],"mappings":";;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;AA4CA,SAAgB,iBAAiB,KAA+B;CAC/D,MAAM,SAAS,IAAI,IAAI,GAAG;CAC1B,MAAM,aAAa,OAAO,aAAa,IAAI,aAAa;CACxD,IAAI,CAAC,YAAY,OAAO,EAAE,kBAAkB,IAAI;CAEhD,OAAO,aAAa,OAAO,aAAa;CACxC,OAAO;EACN,kBAAkB,OAAO,SAAS;EAClC,SAAS,kBAAkB;CAC5B;AACD;;;;;;;;AAiBA,eAAsB,qBACrB,KACA,SAAwC,CAAC,GACb;CAC5B,MAAM,SAAS,IAAI,GAAA,QAAG,OAAO,iBAAiB,GAAG,CAAC;CAClD,MAAM,OAAO,QAAQ;CACrB,MAAM,OAAO,MAAM,OAAO;CAG1B,MAAM,SAAS,MAAM;CAGrB,MAAM,QAA0B,CAAC;CAEjC,OAAO;EACN,IAAI,IAAIA,OAAAA,OAAO;GACd,GAAG;GACH,SAAS,IAAIC,OAAAA,gBAAgB,EAG5B,MAAM;IACL,SAAS,YAAY,aAAa,QAAQ,QAAQ,KAAK;IACvD,KAAK,YAAY,CAAC;GACnB,EACD,CAAC;EACF,CAAC;EACD,MAAM,WAAW;GAChB,IAAI;IACH,MAAM,OAAO,MAAM,UAAU;GAC9B,UAAU;IACT,MAAM,OAAO,IAAI;GAClB;EACD;CACD;AACD;;;;;;;;AASA,SAAS,aACR,QACA,QACA,OACgB;CAChB,IAAI,QAAQ;CACZ,MAAM,QAAQ,OAAO,MAAM,KAAK,MAAM;CAGtC,MAAM,UAAU,OAAO,OAAO,MAAM;CAGpC,MAAM,oBAAoB,UAAkB;EAC3C,KAAK,IAAI,IAAI,MAAM,SAAS,GAAG,KAAK,GAAG,KAAK;GAC3C,MAAM,YAAY,MAAM;GACxB,IAAI,UAAU,aAAa,WAAW,UAAU,SAAS,OACxD,MAAM,OAAO,GAAG,CAAC;EAEnB;CACD;CAEA,OAAO,OAAO,SAAS;EACtB,QAAQ,MAAe,GAAG,SAAoB;GAG7C,IAAI,OAAO,SAAS,YAAY,OAAO,KAAK,GAAG,EAAE,MAAM,YACtD,OAAO,MAAM,MAAM,GAAG,IAAI;GAE3B,MAAM,YAAY,KAAK,KAAK,CAAC,CAAC,YAAY;GAE1C,IAAI,cAAc,WAAW,UAAU,WAAW,mBAAmB,GAAG;IACvE,MAAM,YAAY,WAAW,EAAE;IAC/B,OAAO,aAAa,MAAM,aAAa,WAAW,CAAC;GACpD;GACA,IAAI,cAAc,UAAU;IAC3B,MAAM,QAAQ;IACd,iBAAiB,KAAK;IACtB,OAAO,aAAa,MAAM,6BAA6B,OAAO,CAAC;GAChE;GACA,IAAI,cAAc,YAAY;IAC7B,MAAM,QAAQ;IACd,iBAAiB,KAAK;IACtB,OAAO,aAAa,MAAM,iCAAiC,OAAO,CAAC;GACpE;GAEA,MAAM,UAAU,iBAAiB,IAAI;GACrC,IAAI,SACH,OAAO,OAAO,YAAY;IACzB,MAAM,SAAS,MAAM,MAAM,MAAM,GAAG,IAAI;IACxC,MAAM,OAAO,SAAS;KAAE,UAAU;KAAS;IAAM,CAAC;IAClD,OAAO;GACR,CAAC;GAEF,IAAI,QAAQ,KAAK,MAAM,SAAS,GAC/B,OAAO,aAAa,MAAM,MAAM,GAAG,IAAI,CAAC;GAEzC,OAAO,aAAa,cAAc,OAAO,MAAM,IAAI,CAAC;EACrD;EACA,eAAe,CAAC;CACjB,CAAC;CACD,OAAO;AACR;AAgBA,MAAM,aAAa,OAAO,GAAG;AAC7B,MAAM,oBAA4D;CACjE,CAAC,aAAa,IAAI,OAAO,OAAO,GAAG,gBAAgB,WAAW,SAAS,GAAG,CAAC;CAC3E,CACC,WACA,IAAI,OACH,OAAO,GAAG,+BAA+B,WAAW,SACpD,GACD,CACD;CACA,CACC,eACA,IAAI,OACH,OAAO,GAAG,iEAAiE,WAAW,SACtF,GACD,CACD;AACD;;;;;;AAOA,SAAS,iBAAiB,MAA4C;CACrE,MAAM,YAAY,KAAK,KAAK;CAC5B,KAAK,MAAM,CAAC,QAAQ,YAAY,mBAAmB;EAClD,MAAM,QAAQ,QAAQ,KAAK,SAAS;EACpC,IAAI,CAAC,OAAO;EACZ,MAAM,aAAa,MAAM;EAIzB,OAAO;GAAE;GAAQ,MAHJ,WAAW,WAAW,IAAG,IACnC,WAAW,MAAM,GAAG,EAAE,CAAC,CAAC,WAAW,QAAM,IAAG,IAC5C,WAAW,YAAY;EACJ;CACvB;AAED;;;;;AAMA,SAAS,MACR,OACA,SACA,IACO;CACP,IAAI,QAAQ,WAAW,aAAa;EACnC,MAAM,KAAK;GAAE,MAAM,QAAQ;GAAM,GAAG;EAAG,CAAC;EACxC;CACD;CACA,MAAM,QAAQ,MAAM,eAAe,MAAM,EAAE,SAAS,QAAQ,IAAI;CAGhE,IAAI,UAAU,IAAI;CAClB,MAAM,OAAO,QAAQ,WAAW,YAAY,QAAQ,QAAQ,CAAC;AAC9D;;;;;;;;AAYA,SAAS,QAAe;CACvB,IAAI,QAA0B,QAAQ,QAAQ;CAC9C,QAAQ,SAAS;EAChB,MAAM,OAAO,MAAM,KAAK,MAAM,IAAI;EAClC,QAAQ,KAAK,YAAY,CAAC,CAAC;EAC3B,OAAO;CACR;AACD;;;;;;;;;;;;;AAcA,eAAe,cACd,OACA,MACA,MACmB;CACnB,MAAM,MAAM,0BAA0B;CACtC,IAAI;EACH,MAAM,SAAS,MAAM,MAAM,MAAM,GAAG,IAAI;EACxC,MAAM,MAAM,kCAAkC;EAC9C,OAAO;CACR,SAAS,OAAO;EACf,MAAM,MAAM,sCAAsC;EAClD,MAAM,MAAM,kCAAkC;EAC9C,MAAM;CACP;AACD;;;;;;;;AASA,IAAa,sBAAb,MAAgD;CAC/C,uBAAwB,IAAI,IAAqC;CAEjE,IACC,KACA,KACA,QACuB;EACvB,IAAI,cAAc,KAAK,KAAK,IAAI,GAAG;EACnC,IAAI,CAAC,aAAa;GACjB,cAAc,qBAAqB,KAAK,MAAM;GAC9C,KAAK,KAAK,IAAI,KAAK,WAAW;EAC/B;EACA,OAAO,YAAY,MAAM,EAAE,SAAS,EAAE;CACvC;CAEA,MAAM,cAA6B;EAClC,MAAM,eAAe,CAAC,GAAG,KAAK,KAAK,OAAO,CAAC;EAC3C,KAAK,KAAK,MAAM;EAKhB,MAAM,UAAS,MAHO,QAAQ,WAC7B,aAAa,IAAI,OAAO,iBAAiB,MAAM,YAAA,CAAa,SAAS,CAAC,CACvE,EAAA,CACuB,MACrB,WAA4C,OAAO,WAAW,UAChE;EACA,IAAI,QAAQ,MAAM,OAAO;CAC1B;AACD"}

package/dist/transactions.d.cts

@@ -0,0 +1,44 @@
+import { Kysely, KyselyConfig } from "kysely";
+//#region src/transactions.d.ts
+/** How to reach one database: a URL, and its search path as a startup option. */
+export interface ConnectionConfig {
+  connectionString: string;
+  options?: string;
+}
+/**
+ * The connection settings for a database URL.
+ *
+ * `?search_path=` is not a libpq parameter: a URL carrying it connects and then
+ * resolves every name against `public`. Postgres takes it as a startup option,
+ * which is what a schema tenant's URL needs to reach its own tables.
+ */
+export declare function connectionConfig(url: string): ConnectionConfig;
+/** One database's connection, inside a transaction the test will roll back. */
+export interface BoundTransaction {
+  /** Everything queried through this runs inside the test's transaction. */
+  readonly db: Kysely<any>;
+  /** Undo everything, and let the connection go. */
+  rollback(): Promise<void>;
+}
+/**
+ * Open a connection to `url` and begin the transaction a test will roll back.
+ *
+ * `config` is what the database's own clients are built with — its plugins
+ * above all. The code under test is handed this Kysely in place of its own, so
+ * one built without them runs different queries than production does.
+ */
+export declare function openBoundTransaction(url: string, config?: Omit<KyselyConfig, 'dialect'>): Promise<BoundTransaction>;
+/**
+ * The transactions one test has opened, by database.
+ *
+ * `get` opens a database's transaction the first time it is asked for and
+ * hands back the same one after that — including to a caller that asks while
+ * it is still opening. `rollbackAll` ends every one of them.
+ */
+export declare class TransactionRegistry<TKey = string> {
+  private readonly open;
+  get(key: TKey, url: string, config?: Omit<KyselyConfig, 'dialect'>): Promise<Kysely<any>>;
+  rollbackAll(): Promise<void>;
+}
+//#endregion
+//# sourceMappingURL=transactions.d.cts.map

package/dist/transactions.d.cts.map

@@ -0,0 +1 @@
+{"version":3,"file":"transactions.d.cts","names":[],"sources":["../src/transactions.ts"],"mappings":";;;iBAgCiB;EAChB;EACA;;;;;;;;;wBAUe,iBAAiB,cAAc;;iBAa9B;;WAEP,IAAI;;EAEb,YAAY;;;;;;;;;wBAUS,qBACrB,aACA,SAAQ,KAAK,2BACX,QAAQ;;;;;;;;qBAsOE,oBAAoB;mBACf;EAEjB,IACC,KAAK,MACL,aACA,SAAS,KAAK,2BACZ,QAAQ;EASL,eAAe"}

package/dist/transactions.d.mts

@@ -0,0 +1,44 @@
+import { Kysely, KyselyConfig } from "kysely";
+//#region src/transactions.d.ts
+/** How to reach one database: a URL, and its search path as a startup option. */
+export interface ConnectionConfig {
+  connectionString: string;
+  options?: string;
+}
+/**
+ * The connection settings for a database URL.
+ *
+ * `?search_path=` is not a libpq parameter: a URL carrying it connects and then
+ * resolves every name against `public`. Postgres takes it as a startup option,
+ * which is what a schema tenant's URL needs to reach its own tables.
+ */
+export declare function connectionConfig(url: string): ConnectionConfig;
+/** One database's connection, inside a transaction the test will roll back. */
+export interface BoundTransaction {
+  /** Everything queried through this runs inside the test's transaction. */
+  readonly db: Kysely<any>;
+  /** Undo everything, and let the connection go. */
+  rollback(): Promise<void>;
+}
+/**
+ * Open a connection to `url` and begin the transaction a test will roll back.
+ *
+ * `config` is what the database's own clients are built with — its plugins
+ * above all. The code under test is handed this Kysely in place of its own, so
+ * one built without them runs different queries than production does.
+ */
+export declare function openBoundTransaction(url: string, config?: Omit<KyselyConfig, 'dialect'>): Promise<BoundTransaction>;
+/**
+ * The transactions one test has opened, by database.
+ *
+ * `get` opens a database's transaction the first time it is asked for and
+ * hands back the same one after that — including to a caller that asks while
+ * it is still opening. `rollbackAll` ends every one of them.
+ */
+export declare class TransactionRegistry<TKey = string> {
+  private readonly open;
+  get(key: TKey, url: string, config?: Omit<KyselyConfig, 'dialect'>): Promise<Kysely<any>>;
+  rollbackAll(): Promise<void>;
+}
+//#endregion
+//# sourceMappingURL=transactions.d.mts.map

package/dist/transactions.d.mts.map

@@ -0,0 +1 @@
+{"version":3,"file":"transactions.d.mts","names":[],"sources":["../src/transactions.ts"],"mappings":";;;iBAgCiB;EAChB;EACA;;;;;;;;;wBAUe,iBAAiB,cAAc;;iBAa9B;;WAEP,IAAI;;EAEb,YAAY;;;;;;;;;wBAUS,qBACrB,aACA,SAAQ,KAAK,2BACX,QAAQ;;;;;;;;qBAsOE,oBAAoB;mBACf;EAEjB,IACC,KAAK,MACL,aACA,SAAS,KAAK,2BACZ,QAAQ;EASL,eAAe"}

package/dist/transactions.mjs

@@ -0,0 +1,234 @@
+import pg from "pg";
+import { Kysely, PostgresDialect } from "kysely";
+//#region src/transactions.ts
+/**
+* A transaction per database, per test — as each runs deployed.
+*
+* An app's database and a schema tenant such as the auth server's are separate
+* databases to the code using them: separate connections, credentials, roles
+* and search paths. A test keeps them that way. Each gets its own connection,
+* opened into a transaction the first time the test touches it, and every one
+* is rolled back when the test ends.
+*
+* Code under test must be free to use transactions itself, and better-auth
+* does — it also calls `db.connection()`, which Kysely refuses on a
+* `Transaction`. So what a test hands over is not a `Transaction` but a Kysely
+* over the one connection already inside the test's transaction, on which a
+* `BEGIN` becomes a savepoint and a `COMMIT` releases it. Nothing the code does
+* can end the transaction the test will roll back.
+*
+* A statement the code runs outside a transaction of its own gets a savepoint
+* too, so a failure — a duplicate key two concurrent requests raced to insert —
+* fails that statement alone, as it would deployed, rather than every request
+* the test makes after it.
+*
+* A savepoint the test or the code opens by name — `savepoint refused`, a
+* statement expected to fail, `rollback to savepoint refused` — is its own
+* transaction control, and passes through untouched. While one is open,
+* statements run in it as they would inside a `begin`: unwrapped, so a failure
+* aborts the transaction until the matching `rollback to savepoint`.
+*/
+/**
+* The connection settings for a database URL.
+*
+* `?search_path=` is not a libpq parameter: a URL carrying it connects and then
+* resolves every name against `public`. Postgres takes it as a startup option,
+* which is what a schema tenant's URL needs to reach its own tables.
+*/
+function connectionConfig(url) {
+	const parsed = new URL(url);
+	const searchPath = parsed.searchParams.get("search_path");
+	if (!searchPath) return { connectionString: url };
+	parsed.searchParams.delete("search_path");
+	return {
+		connectionString: parsed.toString(),
+		options: `-c search_path=${searchPath}`
+	};
+}
+/**
+* Open a connection to `url` and begin the transaction a test will roll back.
+*
+* `config` is what the database's own clients are built with — its plugins
+* above all. The code under test is handed this Kysely in place of its own, so
+* one built without them runs different queries than production does.
+*/
+async function openBoundTransaction(url, config = {}) {
+	const client = new pg.Client(connectionConfig(url));
+	await client.connect();
+	await client.query("BEGIN");
+	const inTurn = turns();
+	const named = [];
+	return {
+		db: new Kysely({
+			...config,
+			dialect: new PostgresDialect({ pool: {
+				connect: async () => savepointing(client, inTurn, named),
+				end: async () => {}
+			} })
+		}),
+		async rollback() {
+			try {
+				await client.query("ROLLBACK");
+			} finally {
+				await client.end();
+			}
+		}
+	};
+}
+/**
+* The client, with transaction control turned into savepoints.
+*
+* Kysely issues `begin`/`commit`/`rollback` as plain statements; this is the
+* one place they can be caught. A depth counter names the savepoints, so
+* nested transactions nest.
+*/
+function savepointing(client, inTurn, named) {
+	let depth = 0;
+	const query = client.query.bind(client);
+	const wrapped = Object.create(client);
+	const closeTransaction = (level) => {
+		for (let i = named.length - 1; i >= 0; i--) {
+			const savepoint = named[i];
+			if (savepoint.checkout === wrapped && savepoint.depth >= level) named.splice(i, 1);
+		}
+	};
+	Object.assign(wrapped, {
+		query: (text, ...rest) => {
+			if (typeof text !== "string" || typeof rest.at(-1) === "function") return query(text, ...rest);
+			const statement = text.trim().toLowerCase();
+			if (statement === "begin" || statement.startsWith("start transaction")) {
+				const savepoint = `test_sp_${++depth}`;
+				return inTurn(() => query(`SAVEPOINT ${savepoint}`));
+			}
+			if (statement === "commit") {
+				const level = depth--;
+				closeTransaction(level);
+				return inTurn(() => query(`RELEASE SAVEPOINT test_sp_${level}`));
+			}
+			if (statement === "rollback") {
+				const level = depth--;
+				closeTransaction(level);
+				return inTurn(() => query(`ROLLBACK TO SAVEPOINT test_sp_${level}`));
+			}
+			const control = savepointControl(text);
+			if (control) return inTurn(async () => {
+				const result = await query(text, ...rest);
+				track(named, control, {
+					checkout: wrapped,
+					depth
+				});
+				return result;
+			});
+			if (depth > 0 || named.length > 0) return inTurn(() => query(text, ...rest));
+			return inTurn(() => autocommitted(query, text, rest));
+		},
+		release: () => {}
+	});
+	return wrapped;
+}
+const IDENTIFIER = String.raw`("(?:[^"]|"")+"|[^\s;"]+)`;
+const SAVEPOINT_CONTROL = [
+	["savepoint", new RegExp(String.raw`^savepoint\s+${IDENTIFIER}\s*;?$`, "i")],
+	["release", new RegExp(String.raw`^release\s+(?:savepoint\s+)?${IDENTIFIER}\s*;?$`, "i")],
+	["rollback to", new RegExp(String.raw`^rollback\s+(?:(?:work|transaction)\s+)?to\s+(?:savepoint\s+)?${IDENTIFIER}\s*;?$`, "i")]
+];
+/**
+* The savepoint a statement opens, releases or rolls back to, named as
+* Postgres names it: an unquoted name folded to lower case, a quoted one as
+* written.
+*/
+function savepointControl(text) {
+	const statement = text.trim();
+	for (const [action, pattern] of SAVEPOINT_CONTROL) {
+		const match = pattern.exec(statement);
+		if (!match) continue;
+		const identifier = match[1];
+		return {
+			action,
+			name: identifier.startsWith("\"") ? identifier.slice(1, -1).replaceAll("\"\"", "\"") : identifier.toLowerCase()
+		};
+	}
+}
+/**
+* Keep the named savepoints as Postgres does: `RELEASE x` ends x and every
+* savepoint after it; `ROLLBACK TO x` ends those after it and keeps x open.
+*/
+function track(named, control, at) {
+	if (control.action === "savepoint") {
+		named.push({
+			name: control.name,
+			...at
+		});
+		return;
+	}
+	const index = named.findLastIndex((s) => s.name === control.name);
+	if (index === -1) return;
+	named.splice(control.action === "release" ? index : index + 1);
+}
+/**
+* One statement at a time on a connection, in the order they were asked for.
+*
+* The connection runs them one at a time anyway; this keeps a statement and
+* the savepoint around it together, so nothing another request sends lands
+* between them — a `RELEASE` also ends every savepoint opened after its own.
+*/
+function turns() {
+	let queue = Promise.resolve();
+	return (work) => {
+		const turn = queue.then(work, work);
+		queue = turn.catch(() => {});
+		return turn;
+	};
+}
+/**
+* A statement run outside any transaction the code opened, failing the way
+* it would deployed: on its own.
+*
+* Deployed, such a statement is its own transaction, so one that fails — a
+* duplicate key two concurrent requests both tried to insert — fails alone,
+* and code written for that (Better Auth's rate limiter reads the row the
+* other request inserted, and carries on) does carry on. Inside the test's
+* transaction a failed statement would abort everything after it, every
+* request of the test included. A savepoint around it puts back what
+* deployed would have.
+*/
+async function autocommitted(query, text, rest) {
+	await query("SAVEPOINT test_statement");
+	try {
+		const result = await query(text, ...rest);
+		await query("RELEASE SAVEPOINT test_statement");
+		return result;
+	} catch (error) {
+		await query("ROLLBACK TO SAVEPOINT test_statement");
+		await query("RELEASE SAVEPOINT test_statement");
+		throw error;
+	}
+}
+/**
+* The transactions one test has opened, by database.
+*
+* `get` opens a database's transaction the first time it is asked for and
+* hands back the same one after that — including to a caller that asks while
+* it is still opening. `rollbackAll` ends every one of them.
+*/
+var TransactionRegistry = class {
+	open = /* @__PURE__ */ new Map();
+	get(key, url, config) {
+		let transaction = this.open.get(key);
+		if (!transaction) {
+			transaction = openBoundTransaction(url, config);
+			this.open.set(key, transaction);
+		}
+		return transaction.then(({ db }) => db);
+	}
+	async rollbackAll() {
+		const transactions = [...this.open.values()];
+		this.open.clear();
+		const failed = (await Promise.allSettled(transactions.map(async (transaction) => (await transaction).rollback()))).find((result) => result.status === "rejected");
+		if (failed) throw failed.reason;
+	}
+};
+//#endregion
+export { TransactionRegistry, connectionConfig, openBoundTransaction };
+
+//# sourceMappingURL=transactions.mjs.map

package/dist/transactions.mjs.map

@@ -0,0 +1 @@
+{"version":3,"file":"transactions.mjs","names":[],"sources":["../src/transactions.ts"],"sourcesContent":["/**\n * A transaction per database, per test — as each runs deployed.\n *\n * An app's database and a schema tenant such as the auth server's are separate\n * databases to the code using them: separate connections, credentials, roles\n * and search paths. A test keeps them that way. Each gets its own connection,\n * opened into a transaction the first time the test touches it, and every one\n * is rolled back when the test ends.\n *\n * Code under test must be free to use transactions itself, and better-auth\n * does — it also calls `db.connection()`, which Kysely refuses on a\n * `Transaction`. So what a test hands over is not a `Transaction` but a Kysely\n * over the one connection already inside the test's transaction, on which a\n * `BEGIN` becomes a savepoint and a `COMMIT` releases it. Nothing the code does\n * can end the transaction the test will roll back.\n *\n * A statement the code runs outside a transaction of its own gets a savepoint\n * too, so a failure — a duplicate key two concurrent requests raced to insert —\n * fails that statement alone, as it would deployed, rather than every request\n * the test makes after it.\n *\n * A savepoint the test or the code opens by name — `savepoint refused`, a\n * statement expected to fail, `rollback to savepoint refused` — is its own\n * transaction control, and passes through untouched. While one is open,\n * statements run in it as they would inside a `begin`: unwrapped, so a failure\n * aborts the transaction until the matching `rollback to savepoint`.\n */\n\nimport { Kysely, type KyselyConfig, PostgresDialect } from 'kysely';\nimport pg from 'pg';\n\n/** How to reach one database: a URL, and its search path as a startup option. */\nexport interface ConnectionConfig {\n\tconnectionString: string;\n\toptions?: string;\n}\n\n/**\n * The connection settings for a database URL.\n *\n * `?search_path=` is not a libpq parameter: a URL carrying it connects and then\n * resolves every name against `public`. Postgres takes it as a startup option,\n * which is what a schema tenant's URL needs to reach its own tables.\n */\nexport function connectionConfig(url: string): ConnectionConfig {\n\tconst parsed = new URL(url);\n\tconst searchPath = parsed.searchParams.get('search_path');\n\tif (!searchPath) return { connectionString: url };\n\n\tparsed.searchParams.delete('search_path');\n\treturn {\n\t\tconnectionString: parsed.toString(),\n\t\toptions: `-c search_path=${searchPath}`,\n\t};\n}\n\n/** One database's connection, inside a transaction the test will roll back. */\nexport interface BoundTransaction {\n\t/** Everything queried through this runs inside the test's transaction. */\n\treadonly db: Kysely<any>;\n\t/** Undo everything, and let the connection go. */\n\trollback(): Promise<void>;\n}\n\n/**\n * Open a connection to `url` and begin the transaction a test will roll back.\n *\n * `config` is what the database's own clients are built with — its plugins\n * above all. The code under test is handed this Kysely in place of its own, so\n * one built without them runs different queries than production does.\n */\nexport async function openBoundTransaction(\n\turl: string,\n\tconfig: Omit<KyselyConfig, 'dialect'> = {},\n): Promise<BoundTransaction> {\n\tconst client = new pg.Client(connectionConfig(url));\n\tawait client.connect();\n\tawait client.query('BEGIN');\n\t// Shared by every checkout of the connection: concurrent requests in one\n\t// test each check it out, and all of them take turns on it.\n\tconst inTurn = turns();\n\t// The savepoints opened by name, shared by every checkout as the\n\t// connection's transaction is.\n\tconst named: NamedSavepoint[] = [];\n\n\treturn {\n\t\tdb: new Kysely({\n\t\t\t...config,\n\t\t\tdialect: new PostgresDialect({\n\t\t\t\t// A \"pool\" of the one connection. Released never, ended never: the\n\t\t\t\t// test owns it, and it closes in `rollback`.\n\t\t\t\tpool: {\n\t\t\t\t\tconnect: async () => savepointing(client, inTurn, named),\n\t\t\t\t\tend: async () => {},\n\t\t\t\t} as unknown as pg.Pool,\n\t\t\t}),\n\t\t}),\n\t\tasync rollback() {\n\t\t\ttry {\n\t\t\t\tawait client.query('ROLLBACK');\n\t\t\t} finally {\n\t\t\t\tawait client.end();\n\t\t\t}\n\t\t},\n\t};\n}\n\n/**\n * The client, with transaction control turned into savepoints.\n *\n * Kysely issues `begin`/`commit`/`rollback` as plain statements; this is the\n * one place they can be caught. A depth counter names the savepoints, so\n * nested transactions nest.\n */\nfunction savepointing(\n\tclient: pg.Client,\n\tinTurn: Turns,\n\tnamed: NamedSavepoint[],\n): pg.PoolClient {\n\tlet depth = 0;\n\tconst query = client.query.bind(client) as (\n\t\t...args: unknown[]\n\t) => Promise<unknown>;\n\tconst wrapped = Object.create(client) as pg.PoolClient;\n\n\t// Ending one of the code's transactions ends every savepoint opened in it.\n\tconst closeTransaction = (level: number) => {\n\t\tfor (let i = named.length - 1; i >= 0; i--) {\n\t\t\tconst savepoint = named[i]!;\n\t\t\tif (savepoint.checkout === wrapped && savepoint.depth >= level) {\n\t\t\t\tnamed.splice(i, 1);\n\t\t\t}\n\t\t}\n\t};\n\n\tObject.assign(wrapped, {\n\t\tquery: (text: unknown, ...rest: unknown[]) => {\n\t\t\t// A cursor, a query config, a callback: not what Kysely sends for a\n\t\t\t// statement, and nothing to wrap.\n\t\t\tif (typeof text !== 'string' || typeof rest.at(-1) === 'function') {\n\t\t\t\treturn query(text, ...rest);\n\t\t\t}\n\t\t\tconst statement = text.trim().toLowerCase();\n\n\t\t\tif (statement === 'begin' || statement.startsWith('start transaction')) {\n\t\t\t\tconst savepoint = `test_sp_${++depth}`;\n\t\t\t\treturn inTurn(() => query(`SAVEPOINT ${savepoint}`));\n\t\t\t}\n\t\t\tif (statement === 'commit') {\n\t\t\t\tconst level = depth--;\n\t\t\t\tcloseTransaction(level);\n\t\t\t\treturn inTurn(() => query(`RELEASE SAVEPOINT test_sp_${level}`));\n\t\t\t}\n\t\t\tif (statement === 'rollback') {\n\t\t\t\tconst level = depth--;\n\t\t\t\tcloseTransaction(level);\n\t\t\t\treturn inTurn(() => query(`ROLLBACK TO SAVEPOINT test_sp_${level}`));\n\t\t\t}\n\n\t\t\tconst control = savepointControl(text);\n\t\t\tif (control) {\n\t\t\t\treturn inTurn(async () => {\n\t\t\t\t\tconst result = await query(text, ...rest);\n\t\t\t\t\ttrack(named, control, { checkout: wrapped, depth });\n\t\t\t\t\treturn result;\n\t\t\t\t});\n\t\t\t}\n\t\t\tif (depth > 0 || named.length > 0) {\n\t\t\t\treturn inTurn(() => query(text, ...rest));\n\t\t\t}\n\t\t\treturn inTurn(() => autocommitted(query, text, rest));\n\t\t},\n\t\trelease: () => {},\n\t});\n\treturn wrapped;\n}\n\n/** A savepoint the test or the code opened by name, and where it opened it. */\ninterface NamedSavepoint {\n\tname: string;\n\t/** The checkout, and its transaction depth, the savepoint was opened at. */\n\tcheckout: pg.PoolClient;\n\tdepth: number;\n}\n\n/** `SAVEPOINT x`, `RELEASE [SAVEPOINT] x`, `ROLLBACK [WORK] TO [SAVEPOINT] x`. */\ninterface SavepointControl {\n\taction: 'savepoint' | 'release' | 'rollback to';\n\tname: string;\n}\n\nconst IDENTIFIER = String.raw`(\"(?:[^\"]|\"\")+\"|[^\\s;\"]+)`;\nconst SAVEPOINT_CONTROL: [SavepointControl['action'], RegExp][] = [\n\t['savepoint', new RegExp(String.raw`^savepoint\\s+${IDENTIFIER}\\s*;?$`, 'i')],\n\t[\n\t\t'release',\n\t\tnew RegExp(\n\t\t\tString.raw`^release\\s+(?:savepoint\\s+)?${IDENTIFIER}\\s*;?$`,\n\t\t\t'i',\n\t\t),\n\t],\n\t[\n\t\t'rollback to',\n\t\tnew RegExp(\n\t\t\tString.raw`^rollback\\s+(?:(?:work|transaction)\\s+)?to\\s+(?:savepoint\\s+)?${IDENTIFIER}\\s*;?$`,\n\t\t\t'i',\n\t\t),\n\t],\n];\n\n/**\n * The savepoint a statement opens, releases or rolls back to, named as\n * Postgres names it: an unquoted name folded to lower case, a quoted one as\n * written.\n */\nfunction savepointControl(text: string): SavepointControl | undefined {\n\tconst statement = text.trim();\n\tfor (const [action, pattern] of SAVEPOINT_CONTROL) {\n\t\tconst match = pattern.exec(statement);\n\t\tif (!match) continue;\n\t\tconst identifier = match[1]!;\n\t\tconst name = identifier.startsWith('\"')\n\t\t\t? identifier.slice(1, -1).replaceAll('\"\"', '\"')\n\t\t\t: identifier.toLowerCase();\n\t\treturn { action, name };\n\t}\n\treturn undefined;\n}\n\n/**\n * Keep the named savepoints as Postgres does: `RELEASE x` ends x and every\n * savepoint after it; `ROLLBACK TO x` ends those after it and keeps x open.\n */\nfunction track(\n\tnamed: NamedSavepoint[],\n\tcontrol: SavepointControl,\n\tat: Omit<NamedSavepoint, 'name'>,\n): void {\n\tif (control.action === 'savepoint') {\n\t\tnamed.push({ name: control.name, ...at });\n\t\treturn;\n\t}\n\tconst index = named.findLastIndex((s) => s.name === control.name);\n\t// One opened before this connection was handed over, or by a name the\n\t// wrapper uses itself: nothing tracked to end.\n\tif (index === -1) return;\n\tnamed.splice(control.action === 'release' ? index : index + 1);\n}\n\n/** Runs work in the order it was asked for, each after the last has settled. */\ntype Turns = <T>(work: () => Promise<T>) => Promise<T>;\n\n/**\n * One statement at a time on a connection, in the order they were asked for.\n *\n * The connection runs them one at a time anyway; this keeps a statement and\n * the savepoint around it together, so nothing another request sends lands\n * between them — a `RELEASE` also ends every savepoint opened after its own.\n */\nfunction turns(): Turns {\n\tlet queue: Promise<unknown> = Promise.resolve();\n\treturn (work) => {\n\t\tconst turn = queue.then(work, work);\n\t\tqueue = turn.catch(() => {});\n\t\treturn turn;\n\t};\n}\n\n/**\n * A statement run outside any transaction the code opened, failing the way\n * it would deployed: on its own.\n *\n * Deployed, such a statement is its own transaction, so one that fails — a\n * duplicate key two concurrent requests both tried to insert — fails alone,\n * and code written for that (Better Auth's rate limiter reads the row the\n * other request inserted, and carries on) does carry on. Inside the test's\n * transaction a failed statement would abort everything after it, every\n * request of the test included. A savepoint around it puts back what\n * deployed would have.\n */\nasync function autocommitted(\n\tquery: (...args: unknown[]) => Promise<unknown>,\n\ttext: string,\n\trest: unknown[],\n): Promise<unknown> {\n\tawait query('SAVEPOINT test_statement');\n\ttry {\n\t\tconst result = await query(text, ...rest);\n\t\tawait query('RELEASE SAVEPOINT test_statement');\n\t\treturn result;\n\t} catch (error) {\n\t\tawait query('ROLLBACK TO SAVEPOINT test_statement');\n\t\tawait query('RELEASE SAVEPOINT test_statement');\n\t\tthrow error;\n\t}\n}\n\n/**\n * The transactions one test has opened, by database.\n *\n * `get` opens a database's transaction the first time it is asked for and\n * hands back the same one after that — including to a caller that asks while\n * it is still opening. `rollbackAll` ends every one of them.\n */\nexport class TransactionRegistry<TKey = string> {\n\tprivate readonly open = new Map<TKey, Promise<BoundTransaction>>();\n\n\tget(\n\t\tkey: TKey,\n\t\turl: string,\n\t\tconfig?: Omit<KyselyConfig, 'dialect'>,\n\t): Promise<Kysely<any>> {\n\t\tlet transaction = this.open.get(key);\n\t\tif (!transaction) {\n\t\t\ttransaction = openBoundTransaction(url, config);\n\t\t\tthis.open.set(key, transaction);\n\t\t}\n\t\treturn transaction.then(({ db }) => db);\n\t}\n\n\tasync rollbackAll(): Promise<void> {\n\t\tconst transactions = [...this.open.values()];\n\t\tthis.open.clear();\n\n\t\tconst results = await Promise.allSettled(\n\t\t\ttransactions.map(async (transaction) => (await transaction).rollback()),\n\t\t);\n\t\tconst failed = results.find(\n\t\t\t(result): result is PromiseRejectedResult => result.status === 'rejected',\n\t\t);\n\t\tif (failed) throw failed.reason;\n\t}\n}\n"],"mappings":";;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;AA4CA,SAAgB,iBAAiB,KAA+B;CAC/D,MAAM,SAAS,IAAI,IAAI,GAAG;CAC1B,MAAM,aAAa,OAAO,aAAa,IAAI,aAAa;CACxD,IAAI,CAAC,YAAY,OAAO,EAAE,kBAAkB,IAAI;CAEhD,OAAO,aAAa,OAAO,aAAa;CACxC,OAAO;EACN,kBAAkB,OAAO,SAAS;EAClC,SAAS,kBAAkB;CAC5B;AACD;;;;;;;;AAiBA,eAAsB,qBACrB,KACA,SAAwC,CAAC,GACb;CAC5B,MAAM,SAAS,IAAI,GAAG,OAAO,iBAAiB,GAAG,CAAC;CAClD,MAAM,OAAO,QAAQ;CACrB,MAAM,OAAO,MAAM,OAAO;CAG1B,MAAM,SAAS,MAAM;CAGrB,MAAM,QAA0B,CAAC;CAEjC,OAAO;EACN,IAAI,IAAI,OAAO;GACd,GAAG;GACH,SAAS,IAAI,gBAAgB,EAG5B,MAAM;IACL,SAAS,YAAY,aAAa,QAAQ,QAAQ,KAAK;IACvD,KAAK,YAAY,CAAC;GACnB,EACD,CAAC;EACF,CAAC;EACD,MAAM,WAAW;GAChB,IAAI;IACH,MAAM,OAAO,MAAM,UAAU;GAC9B,UAAU;IACT,MAAM,OAAO,IAAI;GAClB;EACD;CACD;AACD;;;;;;;;AASA,SAAS,aACR,QACA,QACA,OACgB;CAChB,IAAI,QAAQ;CACZ,MAAM,QAAQ,OAAO,MAAM,KAAK,MAAM;CAGtC,MAAM,UAAU,OAAO,OAAO,MAAM;CAGpC,MAAM,oBAAoB,UAAkB;EAC3C,KAAK,IAAI,IAAI,MAAM,SAAS,GAAG,KAAK,GAAG,KAAK;GAC3C,MAAM,YAAY,MAAM;GACxB,IAAI,UAAU,aAAa,WAAW,UAAU,SAAS,OACxD,MAAM,OAAO,GAAG,CAAC;EAEnB;CACD;CAEA,OAAO,OAAO,SAAS;EACtB,QAAQ,MAAe,GAAG,SAAoB;GAG7C,IAAI,OAAO,SAAS,YAAY,OAAO,KAAK,GAAG,EAAE,MAAM,YACtD,OAAO,MAAM,MAAM,GAAG,IAAI;GAE3B,MAAM,YAAY,KAAK,KAAK,CAAC,CAAC,YAAY;GAE1C,IAAI,cAAc,WAAW,UAAU,WAAW,mBAAmB,GAAG;IACvE,MAAM,YAAY,WAAW,EAAE;IAC/B,OAAO,aAAa,MAAM,aAAa,WAAW,CAAC;GACpD;GACA,IAAI,cAAc,UAAU;IAC3B,MAAM,QAAQ;IACd,iBAAiB,KAAK;IACtB,OAAO,aAAa,MAAM,6BAA6B,OAAO,CAAC;GAChE;GACA,IAAI,cAAc,YAAY;IAC7B,MAAM,QAAQ;IACd,iBAAiB,KAAK;IACtB,OAAO,aAAa,MAAM,iCAAiC,OAAO,CAAC;GACpE;GAEA,MAAM,UAAU,iBAAiB,IAAI;GACrC,IAAI,SACH,OAAO,OAAO,YAAY;IACzB,MAAM,SAAS,MAAM,MAAM,MAAM,GAAG,IAAI;IACxC,MAAM,OAAO,SAAS;KAAE,UAAU;KAAS;IAAM,CAAC;IAClD,OAAO;GACR,CAAC;GAEF,IAAI,QAAQ,KAAK,MAAM,SAAS,GAC/B,OAAO,aAAa,MAAM,MAAM,GAAG,IAAI,CAAC;GAEzC,OAAO,aAAa,cAAc,OAAO,MAAM,IAAI,CAAC;EACrD;EACA,eAAe,CAAC;CACjB,CAAC;CACD,OAAO;AACR;AAgBA,MAAM,aAAa,OAAO,GAAG;AAC7B,MAAM,oBAA4D;CACjE,CAAC,aAAa,IAAI,OAAO,OAAO,GAAG,gBAAgB,WAAW,SAAS,GAAG,CAAC;CAC3E,CACC,WACA,IAAI,OACH,OAAO,GAAG,+BAA+B,WAAW,SACpD,GACD,CACD;CACA,CACC,eACA,IAAI,OACH,OAAO,GAAG,iEAAiE,WAAW,SACtF,GACD,CACD;AACD;;;;;;AAOA,SAAS,iBAAiB,MAA4C;CACrE,MAAM,YAAY,KAAK,KAAK;CAC5B,KAAK,MAAM,CAAC,QAAQ,YAAY,mBAAmB;EAClD,MAAM,QAAQ,QAAQ,KAAK,SAAS;EACpC,IAAI,CAAC,OAAO;EACZ,MAAM,aAAa,MAAM;EAIzB,OAAO;GAAE;GAAQ,MAHJ,WAAW,WAAW,IAAG,IACnC,WAAW,MAAM,GAAG,EAAE,CAAC,CAAC,WAAW,QAAM,IAAG,IAC5C,WAAW,YAAY;EACJ;CACvB;AAED;;;;;AAMA,SAAS,MACR,OACA,SACA,IACO;CACP,IAAI,QAAQ,WAAW,aAAa;EACnC,MAAM,KAAK;GAAE,MAAM,QAAQ;GAAM,GAAG;EAAG,CAAC;EACxC;CACD;CACA,MAAM,QAAQ,MAAM,eAAe,MAAM,EAAE,SAAS,QAAQ,IAAI;CAGhE,IAAI,UAAU,IAAI;CAClB,MAAM,OAAO,QAAQ,WAAW,YAAY,QAAQ,QAAQ,CAAC;AAC9D;;;;;;;;AAYA,SAAS,QAAe;CACvB,IAAI,QAA0B,QAAQ,QAAQ;CAC9C,QAAQ,SAAS;EAChB,MAAM,OAAO,MAAM,KAAK,MAAM,IAAI;EAClC,QAAQ,KAAK,YAAY,CAAC,CAAC;EAC3B,OAAO;CACR;AACD;;;;;;;;;;;;;AAcA,eAAe,cACd,OACA,MACA,MACmB;CACnB,MAAM,MAAM,0BAA0B;CACtC,IAAI;EACH,MAAM,SAAS,MAAM,MAAM,MAAM,GAAG,IAAI;EACxC,MAAM,MAAM,kCAAkC;EAC9C,OAAO;CACR,SAAS,OAAO;EACf,MAAM,MAAM,sCAAsC;EAClD,MAAM,MAAM,kCAAkC;EAC9C,MAAM;CACP;AACD;;;;;;;;AASA,IAAa,sBAAb,MAAgD;CAC/C,uBAAwB,IAAI,IAAqC;CAEjE,IACC,KACA,KACA,QACuB;EACvB,IAAI,cAAc,KAAK,KAAK,IAAI,GAAG;EACnC,IAAI,CAAC,aAAa;GACjB,cAAc,qBAAqB,KAAK,MAAM;GAC9C,KAAK,KAAK,IAAI,KAAK,WAAW;EAC/B;EACA,OAAO,YAAY,MAAM,EAAE,SAAS,EAAE;CACvC;CAEA,MAAM,cAA6B;EAClC,MAAM,eAAe,CAAC,GAAG,KAAK,KAAK,OAAO,CAAC;EAC3C,KAAK,KAAK,MAAM;EAKhB,MAAM,UAAS,MAHO,QAAQ,WAC7B,aAAa,IAAI,OAAO,iBAAiB,MAAM,YAAA,CAAa,SAAS,CAAC,CACvE,EAAA,CACuB,MACrB,WAA4C,OAAO,WAAW,UAChE;EACA,IAAI,QAAQ,MAAM,OAAO;CAC1B;AACD"}