@geekmidas/testkit 10.0.0-alpha.86 → 10.0.0-alpha.87

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
@@ -19,6 +19,11 @@ let kysely = require("kysely");
19
19
  * over the one connection already inside the test's transaction, on which a
20
20
  * `BEGIN` becomes a savepoint and a `COMMIT` releases it. Nothing the code does
21
21
  * can end the transaction the test will roll back.
22
+ *
23
+ * A statement the code runs outside a transaction of its own gets a savepoint
24
+ * too, so a failure — a duplicate key two concurrent requests raced to insert —
25
+ * fails that statement alone, as it would deployed, rather than every request
26
+ * the test makes after it.
22
27
  */
23
28
  /**
24
29
  * The connection settings for a database URL.
@@ -48,11 +53,12 @@ async function openBoundTransaction(url, config = {}) {
48
53
  const client = new pg.default.Client(connectionConfig(url));
49
54
  await client.connect();
50
55
  await client.query("BEGIN");
56
+ const inTurn = turns();
51
57
  return {
52
58
  db: new kysely.Kysely({
53
59
  ...config,
54
60
  dialect: new kysely.PostgresDialect({ pool: {
55
- connect: async () => savepointing(client),
61
+ connect: async () => savepointing(client, inTurn),
56
62
  end: async () => {}
57
63
  } })
58
64
  }),
@@ -72,26 +78,73 @@ async function openBoundTransaction(url, config = {}) {
72
78
  * one place they can be caught. A depth counter names the savepoints, so
73
79
  * nested transactions nest.
74
80
  */
75
- function savepointing(client) {
81
+ function savepointing(client, inTurn) {
76
82
  let depth = 0;
77
83
  const query = client.query.bind(client);
78
84
  const wrapped = Object.create(client);
79
85
  Object.assign(wrapped, {
80
86
  query: (text, ...rest) => {
81
- const statement = typeof text === "string" ? text.trim().toLowerCase() : void 0;
82
- if (statement === "begin" || statement?.startsWith("start transaction")) {
83
- depth++;
84
- return query(`SAVEPOINT test_sp_${depth}`);
87
+ if (typeof text !== "string" || typeof rest.at(-1) === "function") return query(text, ...rest);
88
+ const statement = text.trim().toLowerCase();
89
+ if (statement === "begin" || statement.startsWith("start transaction")) {
90
+ const savepoint = `test_sp_${++depth}`;
91
+ return inTurn(() => query(`SAVEPOINT ${savepoint}`));
92
+ }
93
+ if (statement === "commit") {
94
+ const savepoint = `test_sp_${depth--}`;
95
+ return inTurn(() => query(`RELEASE SAVEPOINT ${savepoint}`));
85
96
  }
86
- if (statement === "commit") return query(`RELEASE SAVEPOINT test_sp_${depth--}`);
87
- if (statement === "rollback") return query(`ROLLBACK TO SAVEPOINT test_sp_${depth--}`);
88
- return query(text, ...rest);
97
+ if (statement === "rollback") {
98
+ const savepoint = `test_sp_${depth--}`;
99
+ return inTurn(() => query(`ROLLBACK TO SAVEPOINT ${savepoint}`));
100
+ }
101
+ if (depth > 0) return inTurn(() => query(text, ...rest));
102
+ return inTurn(() => autocommitted(query, text, rest));
89
103
  },
90
104
  release: () => {}
91
105
  });
92
106
  return wrapped;
93
107
  }
94
108
  /**
109
+ * One statement at a time on a connection, in the order they were asked for.
110
+ *
111
+ * The connection runs them one at a time anyway; this keeps a statement and
112
+ * the savepoint around it together, so nothing another request sends lands
113
+ * between them — a `RELEASE` also ends every savepoint opened after its own.
114
+ */
115
+ function turns() {
116
+ let queue = Promise.resolve();
117
+ return (work) => {
118
+ const turn = queue.then(work, work);
119
+ queue = turn.catch(() => {});
120
+ return turn;
121
+ };
122
+ }
123
+ /**
124
+ * A statement run outside any transaction the code opened, failing the way
125
+ * it would deployed: on its own.
126
+ *
127
+ * Deployed, such a statement is its own transaction, so one that fails — a
128
+ * duplicate key two concurrent requests both tried to insert — fails alone,
129
+ * and code written for that (Better Auth's rate limiter reads the row the
130
+ * other request inserted, and carries on) does carry on. Inside the test's
131
+ * transaction a failed statement would abort everything after it, every
132
+ * request of the test included. A savepoint around it puts back what
133
+ * deployed would have.
134
+ */
135
+ async function autocommitted(query, text, rest) {
136
+ await query("SAVEPOINT test_statement");
137
+ try {
138
+ const result = await query(text, ...rest);
139
+ await query("RELEASE SAVEPOINT test_statement");
140
+ return result;
141
+ } catch (error) {
142
+ await query("ROLLBACK TO SAVEPOINT test_statement");
143
+ await query("RELEASE SAVEPOINT test_statement");
144
+ throw error;
145
+ }
146
+ }
147
+ /**
95
148
  * The transactions one test has opened, by database.
96
149
  *
97
150
  * `get` opens a database's transaction the first time it is asked for and
@@ -1 +1 @@
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\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\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),\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(client: pg.Client): pg.PoolClient {\n\tlet depth = 0;\n\tconst query = client.query.bind(client) as (...args: unknown[]) => unknown;\n\n\tconst wrapped = Object.create(client) as pg.PoolClient;\n\tObject.assign(wrapped, {\n\t\tquery: (text: unknown, ...rest: unknown[]) => {\n\t\t\tconst statement =\n\t\t\t\ttypeof text === 'string' ? text.trim().toLowerCase() : undefined;\n\n\t\t\tif (statement === 'begin' || statement?.startsWith('start transaction')) {\n\t\t\t\tdepth++;\n\t\t\t\treturn query(`SAVEPOINT test_sp_${depth}`);\n\t\t\t}\n\t\t\tif (statement === 'commit') {\n\t\t\t\treturn query(`RELEASE SAVEPOINT test_sp_${depth--}`);\n\t\t\t}\n\t\t\tif (statement === 'rollback') {\n\t\t\t\treturn query(`ROLLBACK TO SAVEPOINT test_sp_${depth--}`);\n\t\t\t}\n\t\t\treturn query(text, ...rest);\n\t\t},\n\t\trelease: () => {},\n\t});\n\treturn wrapped;\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":";;;;;;;;;;;;;;;;;;;;;;;;;;;;;AAiCA,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;CAE1B,OAAO;EACN,IAAI,IAAIA,OAAAA,OAAO;GACd,GAAG;GACH,SAAS,IAAIC,OAAAA,gBAAgB,EAG5B,MAAM;IACL,SAAS,YAAY,aAAa,MAAM;IACxC,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,aAAa,QAAkC;CACvD,IAAI,QAAQ;CACZ,MAAM,QAAQ,OAAO,MAAM,KAAK,MAAM;CAEtC,MAAM,UAAU,OAAO,OAAO,MAAM;CACpC,OAAO,OAAO,SAAS;EACtB,QAAQ,MAAe,GAAG,SAAoB;GAC7C,MAAM,YACL,OAAO,SAAS,WAAW,KAAK,KAAK,CAAC,CAAC,YAAY,IAAI,KAAA;GAExD,IAAI,cAAc,WAAW,WAAW,WAAW,mBAAmB,GAAG;IACxE;IACA,OAAO,MAAM,qBAAqB,OAAO;GAC1C;GACA,IAAI,cAAc,UACjB,OAAO,MAAM,6BAA6B,SAAS;GAEpD,IAAI,cAAc,YACjB,OAAO,MAAM,iCAAiC,SAAS;GAExD,OAAO,MAAM,MAAM,GAAG,IAAI;EAC3B;EACA,eAAe,CAAC;CACjB,CAAC;CACD,OAAO;AACR;;;;;;;;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"}
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\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\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),\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(client: pg.Client, inTurn: Turns): pg.PoolClient {\n\tlet depth = 0;\n\tconst query = client.query.bind(client) as (\n\t\t...args: unknown[]\n\t) => Promise<unknown>;\n\n\tconst wrapped = Object.create(client) as pg.PoolClient;\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 savepoint = `test_sp_${depth--}`;\n\t\t\t\treturn inTurn(() => query(`RELEASE SAVEPOINT ${savepoint}`));\n\t\t\t}\n\t\t\tif (statement === 'rollback') {\n\t\t\t\tconst savepoint = `test_sp_${depth--}`;\n\t\t\t\treturn inTurn(() => query(`ROLLBACK TO SAVEPOINT ${savepoint}`));\n\t\t\t}\n\t\t\tif (depth > 0) return inTurn(() => query(text, ...rest));\n\t\t\treturn inTurn(() => autocommitted(query, text, rest));\n\t\t},\n\t\trelease: () => {},\n\t});\n\treturn wrapped;\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":";;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;AAsCA,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;CAErB,OAAO;EACN,IAAI,IAAIA,OAAAA,OAAO;GACd,GAAG;GACH,SAAS,IAAIC,OAAAA,gBAAgB,EAG5B,MAAM;IACL,SAAS,YAAY,aAAa,QAAQ,MAAM;IAChD,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,aAAa,QAAmB,QAA8B;CACtE,IAAI,QAAQ;CACZ,MAAM,QAAQ,OAAO,MAAM,KAAK,MAAM;CAItC,MAAM,UAAU,OAAO,OAAO,MAAM;CACpC,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,YAAY,WAAW;IAC7B,OAAO,aAAa,MAAM,qBAAqB,WAAW,CAAC;GAC5D;GACA,IAAI,cAAc,YAAY;IAC7B,MAAM,YAAY,WAAW;IAC7B,OAAO,aAAa,MAAM,yBAAyB,WAAW,CAAC;GAChE;GACA,IAAI,QAAQ,GAAG,OAAO,aAAa,MAAM,MAAM,GAAG,IAAI,CAAC;GACvD,OAAO,aAAa,cAAc,OAAO,MAAM,IAAI,CAAC;EACrD;EACA,eAAe,CAAC;CACjB,CAAC;CACD,OAAO;AACR;;;;;;;;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"}
@@ -1 +1 @@
1
- {"version":3,"file":"transactions.d.cts","names":[],"sources":["../src/transactions.ts"],"mappings":";;;iBAqBiB;EAChB;EACA;;;;;;;;;wBAUe,iBAAiB,cAAc;;iBAa9B;;WAEP,IAAI;;EAEb,YAAY;;;;;;;;;wBAUS,qBACrB,aACA,SAAQ,KAAK,2BACX,QAAQ;;;;;;;;qBAoEE,oBAAoB;mBACf;EAEjB,IACC,KAAK,MACL,aACA,SAAS,KAAK,2BACZ,QAAQ;EASL,eAAe"}
1
+ {"version":3,"file":"transactions.d.cts","names":[],"sources":["../src/transactions.ts"],"mappings":";;;iBA0BiB;EAChB;EACA;;;;;;;;;wBAUe,iBAAiB,cAAc;;iBAa9B;;WAEP,IAAI;;EAEb,YAAY;;;;;;;;;wBAUS,qBACrB,aACA,SAAQ,KAAK,2BACX,QAAQ;;;;;;;;qBAgIE,oBAAoB;mBACf;EAEjB,IACC,KAAK,MACL,aACA,SAAS,KAAK,2BACZ,QAAQ;EASL,eAAe"}
@@ -1 +1 @@
1
- {"version":3,"file":"transactions.d.mts","names":[],"sources":["../src/transactions.ts"],"mappings":";;;iBAqBiB;EAChB;EACA;;;;;;;;;wBAUe,iBAAiB,cAAc;;iBAa9B;;WAEP,IAAI;;EAEb,YAAY;;;;;;;;;wBAUS,qBACrB,aACA,SAAQ,KAAK,2BACX,QAAQ;;;;;;;;qBAoEE,oBAAoB;mBACf;EAEjB,IACC,KAAK,MACL,aACA,SAAS,KAAK,2BACZ,QAAQ;EASL,eAAe"}
1
+ {"version":3,"file":"transactions.d.mts","names":[],"sources":["../src/transactions.ts"],"mappings":";;;iBA0BiB;EAChB;EACA;;;;;;;;;wBAUe,iBAAiB,cAAc;;iBAa9B;;WAEP,IAAI;;EAEb,YAAY;;;;;;;;;wBAUS,qBACrB,aACA,SAAQ,KAAK,2BACX,QAAQ;;;;;;;;qBAgIE,oBAAoB;mBACf;EAEjB,IACC,KAAK,MACL,aACA,SAAS,KAAK,2BACZ,QAAQ;EASL,eAAe"}
@@ -16,6 +16,11 @@ import { Kysely, PostgresDialect } from "kysely";
16
16
  * over the one connection already inside the test's transaction, on which a
17
17
  * `BEGIN` becomes a savepoint and a `COMMIT` releases it. Nothing the code does
18
18
  * can end the transaction the test will roll back.
19
+ *
20
+ * A statement the code runs outside a transaction of its own gets a savepoint
21
+ * too, so a failure — a duplicate key two concurrent requests raced to insert —
22
+ * fails that statement alone, as it would deployed, rather than every request
23
+ * the test makes after it.
19
24
  */
20
25
  /**
21
26
  * The connection settings for a database URL.
@@ -45,11 +50,12 @@ async function openBoundTransaction(url, config = {}) {
45
50
  const client = new pg.Client(connectionConfig(url));
46
51
  await client.connect();
47
52
  await client.query("BEGIN");
53
+ const inTurn = turns();
48
54
  return {
49
55
  db: new Kysely({
50
56
  ...config,
51
57
  dialect: new PostgresDialect({ pool: {
52
- connect: async () => savepointing(client),
58
+ connect: async () => savepointing(client, inTurn),
53
59
  end: async () => {}
54
60
  } })
55
61
  }),
@@ -69,26 +75,73 @@ async function openBoundTransaction(url, config = {}) {
69
75
  * one place they can be caught. A depth counter names the savepoints, so
70
76
  * nested transactions nest.
71
77
  */
72
- function savepointing(client) {
78
+ function savepointing(client, inTurn) {
73
79
  let depth = 0;
74
80
  const query = client.query.bind(client);
75
81
  const wrapped = Object.create(client);
76
82
  Object.assign(wrapped, {
77
83
  query: (text, ...rest) => {
78
- const statement = typeof text === "string" ? text.trim().toLowerCase() : void 0;
79
- if (statement === "begin" || statement?.startsWith("start transaction")) {
80
- depth++;
81
- return query(`SAVEPOINT test_sp_${depth}`);
84
+ if (typeof text !== "string" || typeof rest.at(-1) === "function") return query(text, ...rest);
85
+ const statement = text.trim().toLowerCase();
86
+ if (statement === "begin" || statement.startsWith("start transaction")) {
87
+ const savepoint = `test_sp_${++depth}`;
88
+ return inTurn(() => query(`SAVEPOINT ${savepoint}`));
89
+ }
90
+ if (statement === "commit") {
91
+ const savepoint = `test_sp_${depth--}`;
92
+ return inTurn(() => query(`RELEASE SAVEPOINT ${savepoint}`));
82
93
  }
83
- if (statement === "commit") return query(`RELEASE SAVEPOINT test_sp_${depth--}`);
84
- if (statement === "rollback") return query(`ROLLBACK TO SAVEPOINT test_sp_${depth--}`);
85
- return query(text, ...rest);
94
+ if (statement === "rollback") {
95
+ const savepoint = `test_sp_${depth--}`;
96
+ return inTurn(() => query(`ROLLBACK TO SAVEPOINT ${savepoint}`));
97
+ }
98
+ if (depth > 0) return inTurn(() => query(text, ...rest));
99
+ return inTurn(() => autocommitted(query, text, rest));
86
100
  },
87
101
  release: () => {}
88
102
  });
89
103
  return wrapped;
90
104
  }
91
105
  /**
106
+ * One statement at a time on a connection, in the order they were asked for.
107
+ *
108
+ * The connection runs them one at a time anyway; this keeps a statement and
109
+ * the savepoint around it together, so nothing another request sends lands
110
+ * between them — a `RELEASE` also ends every savepoint opened after its own.
111
+ */
112
+ function turns() {
113
+ let queue = Promise.resolve();
114
+ return (work) => {
115
+ const turn = queue.then(work, work);
116
+ queue = turn.catch(() => {});
117
+ return turn;
118
+ };
119
+ }
120
+ /**
121
+ * A statement run outside any transaction the code opened, failing the way
122
+ * it would deployed: on its own.
123
+ *
124
+ * Deployed, such a statement is its own transaction, so one that fails — a
125
+ * duplicate key two concurrent requests both tried to insert — fails alone,
126
+ * and code written for that (Better Auth's rate limiter reads the row the
127
+ * other request inserted, and carries on) does carry on. Inside the test's
128
+ * transaction a failed statement would abort everything after it, every
129
+ * request of the test included. A savepoint around it puts back what
130
+ * deployed would have.
131
+ */
132
+ async function autocommitted(query, text, rest) {
133
+ await query("SAVEPOINT test_statement");
134
+ try {
135
+ const result = await query(text, ...rest);
136
+ await query("RELEASE SAVEPOINT test_statement");
137
+ return result;
138
+ } catch (error) {
139
+ await query("ROLLBACK TO SAVEPOINT test_statement");
140
+ await query("RELEASE SAVEPOINT test_statement");
141
+ throw error;
142
+ }
143
+ }
144
+ /**
92
145
  * The transactions one test has opened, by database.
93
146
  *
94
147
  * `get` opens a database's transaction the first time it is asked for and
@@ -1 +1 @@
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\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\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),\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(client: pg.Client): pg.PoolClient {\n\tlet depth = 0;\n\tconst query = client.query.bind(client) as (...args: unknown[]) => unknown;\n\n\tconst wrapped = Object.create(client) as pg.PoolClient;\n\tObject.assign(wrapped, {\n\t\tquery: (text: unknown, ...rest: unknown[]) => {\n\t\t\tconst statement =\n\t\t\t\ttypeof text === 'string' ? text.trim().toLowerCase() : undefined;\n\n\t\t\tif (statement === 'begin' || statement?.startsWith('start transaction')) {\n\t\t\t\tdepth++;\n\t\t\t\treturn query(`SAVEPOINT test_sp_${depth}`);\n\t\t\t}\n\t\t\tif (statement === 'commit') {\n\t\t\t\treturn query(`RELEASE SAVEPOINT test_sp_${depth--}`);\n\t\t\t}\n\t\t\tif (statement === 'rollback') {\n\t\t\t\treturn query(`ROLLBACK TO SAVEPOINT test_sp_${depth--}`);\n\t\t\t}\n\t\t\treturn query(text, ...rest);\n\t\t},\n\t\trelease: () => {},\n\t});\n\treturn wrapped;\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":";;;;;;;;;;;;;;;;;;;;;;;;;;AAiCA,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;CAE1B,OAAO;EACN,IAAI,IAAI,OAAO;GACd,GAAG;GACH,SAAS,IAAI,gBAAgB,EAG5B,MAAM;IACL,SAAS,YAAY,aAAa,MAAM;IACxC,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,aAAa,QAAkC;CACvD,IAAI,QAAQ;CACZ,MAAM,QAAQ,OAAO,MAAM,KAAK,MAAM;CAEtC,MAAM,UAAU,OAAO,OAAO,MAAM;CACpC,OAAO,OAAO,SAAS;EACtB,QAAQ,MAAe,GAAG,SAAoB;GAC7C,MAAM,YACL,OAAO,SAAS,WAAW,KAAK,KAAK,CAAC,CAAC,YAAY,IAAI,KAAA;GAExD,IAAI,cAAc,WAAW,WAAW,WAAW,mBAAmB,GAAG;IACxE;IACA,OAAO,MAAM,qBAAqB,OAAO;GAC1C;GACA,IAAI,cAAc,UACjB,OAAO,MAAM,6BAA6B,SAAS;GAEpD,IAAI,cAAc,YACjB,OAAO,MAAM,iCAAiC,SAAS;GAExD,OAAO,MAAM,MAAM,GAAG,IAAI;EAC3B;EACA,eAAe,CAAC;CACjB,CAAC;CACD,OAAO;AACR;;;;;;;;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"}
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\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\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),\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(client: pg.Client, inTurn: Turns): pg.PoolClient {\n\tlet depth = 0;\n\tconst query = client.query.bind(client) as (\n\t\t...args: unknown[]\n\t) => Promise<unknown>;\n\n\tconst wrapped = Object.create(client) as pg.PoolClient;\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 savepoint = `test_sp_${depth--}`;\n\t\t\t\treturn inTurn(() => query(`RELEASE SAVEPOINT ${savepoint}`));\n\t\t\t}\n\t\t\tif (statement === 'rollback') {\n\t\t\t\tconst savepoint = `test_sp_${depth--}`;\n\t\t\t\treturn inTurn(() => query(`ROLLBACK TO SAVEPOINT ${savepoint}`));\n\t\t\t}\n\t\t\tif (depth > 0) return inTurn(() => query(text, ...rest));\n\t\t\treturn inTurn(() => autocommitted(query, text, rest));\n\t\t},\n\t\trelease: () => {},\n\t});\n\treturn wrapped;\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":";;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;AAsCA,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;CAErB,OAAO;EACN,IAAI,IAAI,OAAO;GACd,GAAG;GACH,SAAS,IAAI,gBAAgB,EAG5B,MAAM;IACL,SAAS,YAAY,aAAa,QAAQ,MAAM;IAChD,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,aAAa,QAAmB,QAA8B;CACtE,IAAI,QAAQ;CACZ,MAAM,QAAQ,OAAO,MAAM,KAAK,MAAM;CAItC,MAAM,UAAU,OAAO,OAAO,MAAM;CACpC,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,YAAY,WAAW;IAC7B,OAAO,aAAa,MAAM,qBAAqB,WAAW,CAAC;GAC5D;GACA,IAAI,cAAc,YAAY;IAC7B,MAAM,YAAY,WAAW;IAC7B,OAAO,aAAa,MAAM,yBAAyB,WAAW,CAAC;GAChE;GACA,IAAI,QAAQ,GAAG,OAAO,aAAa,MAAM,MAAM,GAAG,IAAI,CAAC;GACvD,OAAO,aAAa,cAAc,OAAO,MAAM,IAAI,CAAC;EACrD;EACA,eAAe,CAAC;CACjB,CAAC;CACD,OAAO;AACR;;;;;;;;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/package.json CHANGED
@@ -1,6 +1,6 @@
1
1
  {
2
2
  "name": "@geekmidas/testkit",
3
- "version": "10.0.0-alpha.86",
3
+ "version": "10.0.0-alpha.87",
4
4
  "private": false,
5
5
  "type": "module",
6
6
  "exports": {
@@ -161,17 +161,17 @@
161
161
  "dependencies": {
162
162
  "@faker-js/faker": "~10.6.0",
163
163
  "tough-cookie": "~6.0.2",
164
- "@geekmidas/envkit": "^10.0.0-alpha.86",
165
- "@geekmidas/logger": "^10.0.0-alpha.86",
166
- "@geekmidas/services": "^10.0.0-alpha.86"
164
+ "@geekmidas/envkit": "^10.0.0-alpha.87",
165
+ "@geekmidas/logger": "^10.0.0-alpha.87",
166
+ "@geekmidas/services": "^10.0.0-alpha.87"
167
167
  },
168
168
  "devDependencies": {
169
169
  "@better-auth/test-utils": "~1.7.6",
170
170
  "@types/pg": "~8.23.1",
171
171
  "better-auth": "~1.7.6",
172
- "@geekmidas/envkit": "^10.0.0-alpha.86",
173
- "@geekmidas/logger": "^10.0.0-alpha.86",
174
- "@geekmidas/services": "^10.0.0-alpha.86"
172
+ "@geekmidas/logger": "^10.0.0-alpha.87",
173
+ "@geekmidas/envkit": "^10.0.0-alpha.87",
174
+ "@geekmidas/services": "^10.0.0-alpha.87"
175
175
  },
176
176
  "repository": {
177
177
  "type": "git",