neon-testing 2.2.0 → 2.2.1-beta.0

package/dist/examples/minimal.test.js

@@ -0,0 +1,19 @@
+// minimal.test.ts
+import { expect, test } from "vitest";
+import { makeNeonTesting } from "neon-testing";
+import { Pool } from "@neondatabase/serverless";
+// Enable Neon test branch for this test file
+makeNeonTesting({
+    apiKey: process.env.NEON_API_KEY,
+    projectId: process.env.NEON_PROJECT_ID,
+    // Recommended for Neon WebSocket drivers to automatically close connections
+    autoCloseWebSockets: true,
+})();
+test("database operations", async () => {
+    const pool = new Pool({ connectionString: process.env.DATABASE_URL });
+    await pool.query(`CREATE TABLE users (id SERIAL PRIMARY KEY, name TEXT)`);
+    await pool.query(`INSERT INTO users (name) VALUES ('Ellen Ripley')`);
+    const users = await pool.query(`SELECT * FROM users`);
+    expect(users.rows).toStrictEqual([{ id: 1, name: "Ellen Ripley" }]);
+});
+//# sourceMappingURL=minimal.test.js.map

package/dist/examples/minimal.test.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"minimal.test.js","sourceRoot":"","sources":["../../examples/minimal.test.ts"],"names":[],"mappings":"AAAA,kBAAkB;AAClB,OAAO,EAAE,MAAM,EAAE,IAAI,EAAE,MAAM,QAAQ,CAAC;AACtC,OAAO,EAAE,eAAe,EAAE,MAAM,cAAc,CAAC;AAC/C,OAAO,EAAE,IAAI,EAAE,MAAM,0BAA0B,CAAC;AAEhD,6CAA6C;AAC7C,eAAe,CAAC;IACd,MAAM,EAAE,OAAO,CAAC,GAAG,CAAC,YAAa;IACjC,SAAS,EAAE,OAAO,CAAC,GAAG,CAAC,eAAgB;IACvC,4EAA4E;IAC5E,mBAAmB,EAAE,IAAI;CAC1B,CAAC,EAAE,CAAC;AAEL,IAAI,CAAC,qBAAqB,EAAE,KAAK,IAAI,EAAE;IACrC,MAAM,IAAI,GAAG,IAAI,IAAI,CAAC,EAAE,gBAAgB,EAAE,OAAO,CAAC,GAAG,CAAC,YAAY,EAAE,CAAC,CAAC;IAEtE,MAAM,IAAI,CAAC,KAAK,CAAC,uDAAuD,CAAC,CAAC;IAC1E,MAAM,IAAI,CAAC,KAAK,CAAC,kDAAkD,CAAC,CAAC;IAErE,MAAM,KAAK,GAAG,MAAM,IAAI,CAAC,KAAK,CAAC,qBAAqB,CAAC,CAAC;IACtD,MAAM,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,aAAa,CAAC,CAAC,EAAE,EAAE,EAAE,CAAC,EAAE,IAAI,EAAE,cAAc,EAAE,CAAC,CAAC,CAAC;AACtE,CAAC,CAAC,CAAC"}

package/dist/examples/neon-testing.d.ts

@@ -0,0 +1,6 @@
+export declare const neonTesting: {
+    (overrides?: import("neon-testing").NeonTestingOverrides): () => import("@neondatabase/api-client").Branch | undefined;
+    deleteAllTestBranches: () => Promise<void>;
+    api: import("@neondatabase/api-client").Api<unknown>;
+};
+//# sourceMappingURL=neon-testing.d.ts.map

package/dist/examples/neon-testing.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"neon-testing.d.ts","sourceRoot":"","sources":["../../examples/neon-testing.ts"],"names":[],"mappings":"AAIA,eAAO,MAAM,WAAW;;;;CAGtB,CAAC"}

package/dist/examples/neon-testing.js

@@ -0,0 +1,8 @@
+// neon-testing.ts
+import { makeNeonTesting } from "neon-testing";
+// Export a configured lifecycle function to use in test files
+export const neonTesting = makeNeonTesting({
+    apiKey: process.env.NEON_API_KEY,
+    projectId: process.env.NEON_PROJECT_ID,
+});
+//# sourceMappingURL=neon-testing.js.map

package/dist/examples/neon-testing.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"neon-testing.js","sourceRoot":"","sources":["../../examples/neon-testing.ts"],"names":[],"mappings":"AAAA,kBAAkB;AAClB,OAAO,EAAE,eAAe,EAAE,MAAM,cAAc,CAAC;AAE/C,8DAA8D;AAC9D,MAAM,CAAC,MAAM,WAAW,GAAG,eAAe,CAAC;IACzC,MAAM,EAAE,OAAO,CAAC,GAAG,CAAC,YAAa;IACjC,SAAS,EAAE,OAAO,CAAC,GAAG,CAAC,eAAgB;CACxC,CAAC,CAAC"}

package/dist/examples/options-validation.test.d.ts

@@ -0,0 +1,2 @@
+export {};
+//# sourceMappingURL=options-validation.test.d.ts.map

package/dist/examples/options-validation.test.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"options-validation.test.d.ts","sourceRoot":"","sources":["../../examples/options-validation.test.ts"],"names":[],"mappings":""}

package/dist/examples/options-validation.test.js

@@ -0,0 +1,99 @@
+import { describe, expect, test } from "vitest";
+import { makeNeonTesting } from "neon-testing";
+const projectId = process.env.NEON_PROJECT_ID;
+describe("expiresIn", () => {
+    /**
+     * Test options validation when setting the factory-level options
+     */
+    describe("factory-level (makeNeonTesting)", () => {
+        test("non-integer throws", () => {
+            expect(() => {
+                makeNeonTesting({
+                    apiKey: process.env.NEON_API_KEY,
+                    projectId,
+                    expiresIn: 600.5,
+                });
+            }).toThrow("expiresIn must be an integer");
+        });
+        test("zero throws", () => {
+            expect(() => {
+                makeNeonTesting({
+                    apiKey: process.env.NEON_API_KEY,
+                    projectId,
+                    expiresIn: 0,
+                });
+            }).toThrow("expiresIn must be a positive integer");
+        });
+        test("negative throws", () => {
+            expect(() => {
+                makeNeonTesting({
+                    apiKey: process.env.NEON_API_KEY,
+                    projectId,
+                    expiresIn: -10,
+                });
+            }).toThrow("expiresIn must be a positive integer");
+        });
+        test("> 30 days throws", () => {
+            expect(() => {
+                makeNeonTesting({
+                    apiKey: process.env.NEON_API_KEY,
+                    projectId,
+                    expiresIn: 2592001,
+                });
+            }).toThrow("expiresIn must not exceed 30 days");
+        });
+        test("valid value does not throw", () => {
+            expect(() => {
+                makeNeonTesting({
+                    apiKey: process.env.NEON_API_KEY,
+                    projectId,
+                    expiresIn: 600,
+                });
+            }).not.toThrow();
+        });
+        test("null does not throw", () => {
+            expect(() => {
+                makeNeonTesting({
+                    apiKey: process.env.NEON_API_KEY,
+                    projectId,
+                    expiresIn: null,
+                });
+            }).not.toThrow();
+        });
+    });
+    /**
+     * Test options validation when overriding the factory-level options
+     */
+    describe("override-level (setup function)", () => {
+        // Create a base neonTesting without expiresIn specified at factory level
+        const baseTesting = makeNeonTesting({
+            apiKey: process.env.NEON_API_KEY,
+            projectId,
+        });
+        test("non-integer in override throws", () => {
+            expect(() => {
+                baseTesting({ expiresIn: 600.5 });
+            }).toThrow("expiresIn must be an integer");
+        });
+        test("zero in override throws", () => {
+            expect(() => {
+                baseTesting({ expiresIn: 0 });
+            }).toThrow("expiresIn must be a positive integer");
+        });
+        test("negative in override throws", () => {
+            expect(() => {
+                baseTesting({ expiresIn: -10 });
+            }).toThrow("expiresIn must be a positive integer");
+        });
+        test("> 30 days in override throws", () => {
+            expect(() => {
+                baseTesting({ expiresIn: 2592001 });
+            }).toThrow("expiresIn must not exceed 30 days");
+        });
+        // Note: Valid override values (600, null) are tested in branch-expiration.test.ts
+        // where describe blocks actually create branches. We can't test them here since
+        // calling baseTesting() inside a test would register beforeAll() hooks which is
+        // not allowed.
+    });
+});
+//# sourceMappingURL=options-validation.test.js.map

package/dist/examples/options-validation.test.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"options-validation.test.js","sourceRoot":"","sources":["../../examples/options-validation.test.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,QAAQ,EAAE,MAAM,EAAE,IAAI,EAAE,MAAM,QAAQ,CAAC;AAChD,OAAO,EAAE,eAAe,EAAE,MAAM,cAAc,CAAC;AAE/C,MAAM,SAAS,GAAG,OAAO,CAAC,GAAG,CAAC,eAAgB,CAAC;AAE/C,QAAQ,CAAC,WAAW,EAAE,GAAG,EAAE;IACzB;;OAEG;IACH,QAAQ,CAAC,iCAAiC,EAAE,GAAG,EAAE;QAC/C,IAAI,CAAC,oBAAoB,EAAE,GAAG,EAAE;YAC9B,MAAM,CAAC,GAAG,EAAE;gBACV,eAAe,CAAC;oBACd,MAAM,EAAE,OAAO,CAAC,GAAG,CAAC,YAAa;oBACjC,SAAS;oBACT,SAAS,EAAE,KAAK;iBACjB,CAAC,CAAC;YACL,CAAC,CAAC,CAAC,OAAO,CAAC,8BAA8B,CAAC,CAAC;QAC7C,CAAC,CAAC,CAAC;QAEH,IAAI,CAAC,aAAa,EAAE,GAAG,EAAE;YACvB,MAAM,CAAC,GAAG,EAAE;gBACV,eAAe,CAAC;oBACd,MAAM,EAAE,OAAO,CAAC,GAAG,CAAC,YAAa;oBACjC,SAAS;oBACT,SAAS,EAAE,CAAC;iBACb,CAAC,CAAC;YACL,CAAC,CAAC,CAAC,OAAO,CAAC,sCAAsC,CAAC,CAAC;QACrD,CAAC,CAAC,CAAC;QAEH,IAAI,CAAC,iBAAiB,EAAE,GAAG,EAAE;YAC3B,MAAM,CAAC,GAAG,EAAE;gBACV,eAAe,CAAC;oBACd,MAAM,EAAE,OAAO,CAAC,GAAG,CAAC,YAAa;oBACjC,SAAS;oBACT,SAAS,EAAE,CAAC,EAAE;iBACf,CAAC,CAAC;YACL,CAAC,CAAC,CAAC,OAAO,CAAC,sCAAsC,CAAC,CAAC;QACrD,CAAC,CAAC,CAAC;QAEH,IAAI,CAAC,kBAAkB,EAAE,GAAG,EAAE;YAC5B,MAAM,CAAC,GAAG,EAAE;gBACV,eAAe,CAAC;oBACd,MAAM,EAAE,OAAO,CAAC,GAAG,CAAC,YAAa;oBACjC,SAAS;oBACT,SAAS,EAAE,OAAO;iBACnB,CAAC,CAAC;YACL,CAAC,CAAC,CAAC,OAAO,CAAC,mCAAmC,CAAC,CAAC;QAClD,CAAC,CAAC,CAAC;QAEH,IAAI,CAAC,4BAA4B,EAAE,GAAG,EAAE;YACtC,MAAM,CAAC,GAAG,EAAE;gBACV,eAAe,CAAC;oBACd,MAAM,EAAE,OAAO,CAAC,GAAG,CAAC,YAAa;oBACjC,SAAS;oBACT,SAAS,EAAE,GAAG;iBACf,CAAC,CAAC;YACL,CAAC,CAAC,CAAC,GAAG,CAAC,OAAO,EAAE,CAAC;QACnB,CAAC,CAAC,CAAC;QAEH,IAAI,CAAC,qBAAqB,EAAE,GAAG,EAAE;YAC/B,MAAM,CAAC,GAAG,EAAE;gBACV,eAAe,CAAC;oBACd,MAAM,EAAE,OAAO,CAAC,GAAG,CAAC,YAAa;oBACjC,SAAS;oBACT,SAAS,EAAE,IAAI;iBAChB,CAAC,CAAC;YACL,CAAC,CAAC,CAAC,GAAG,CAAC,OAAO,EAAE,CAAC;QACnB,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH;;OAEG;IACH,QAAQ,CAAC,iCAAiC,EAAE,GAAG,EAAE;QAC/C,yEAAyE;QACzE,MAAM,WAAW,GAAG,eAAe,CAAC;YAClC,MAAM,EAAE,OAAO,CAAC,GAAG,CAAC,YAAa;YACjC,SAAS;SACV,CAAC,CAAC;QAEH,IAAI,CAAC,gCAAgC,EAAE,GAAG,EAAE;YAC1C,MAAM,CAAC,GAAG,EAAE;gBACV,WAAW,CAAC,EAAE,SAAS,EAAE,KAAK,EAAE,CAAC,CAAC;YACpC,CAAC,CAAC,CAAC,OAAO,CAAC,8BAA8B,CAAC,CAAC;QAC7C,CAAC,CAAC,CAAC;QAEH,IAAI,CAAC,yBAAyB,EAAE,GAAG,EAAE;YACnC,MAAM,CAAC,GAAG,EAAE;gBACV,WAAW,CAAC,EAAE,SAAS,EAAE,CAAC,EAAE,CAAC,CAAC;YAChC,CAAC,CAAC,CAAC,OAAO,CAAC,sCAAsC,CAAC,CAAC;QACrD,CAAC,CAAC,CAAC;QAEH,IAAI,CAAC,6BAA6B,EAAE,GAAG,EAAE;YACvC,MAAM,CAAC,GAAG,EAAE;gBACV,WAAW,CAAC,EAAE,SAAS,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC;YAClC,CAAC,CAAC,CAAC,OAAO,CAAC,sCAAsC,CAAC,CAAC;QACrD,CAAC,CAAC,CAAC;QAEH,IAAI,CAAC,8BAA8B,EAAE,GAAG,EAAE;YACxC,MAAM,CAAC,GAAG,EAAE;gBACV,WAAW,CAAC,EAAE,SAAS,EAAE,OAAO,EAAE,CAAC,CAAC;YACtC,CAAC,CAAC,CAAC,OAAO,CAAC,mCAAmC,CAAC,CAAC;QAClD,CAAC,CAAC,CAAC;QAEH,kFAAkF;QAClF,gFAAgF;QAChF,gFAAgF;QAChF,eAAe;IACjB,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}

package/dist/examples/recommended.test.d.ts

@@ -0,0 +1,2 @@
+export {};
+//# sourceMappingURL=recommended.test.d.ts.map

package/dist/examples/recommended.test.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"recommended.test.d.ts","sourceRoot":"","sources":["../../examples/recommended.test.ts"],"names":[],"mappings":""}

package/dist/examples/recommended.test.js

@@ -0,0 +1,17 @@
+// recommended.test.ts
+import { expect, test } from "vitest";
+import { neonTesting } from "./neon-testing";
+import { Pool } from "@neondatabase/serverless";
+// Enable Neon test branch for this test file
+neonTesting({
+    // Recommended for Neon WebSocket drivers to automatically close connections
+    autoCloseWebSockets: true,
+});
+test("database operations", async () => {
+    const pool = new Pool({ connectionString: process.env.DATABASE_URL });
+    await pool.query(`CREATE TABLE users (id SERIAL PRIMARY KEY, name TEXT)`);
+    await pool.query(`INSERT INTO users (name) VALUES ('Ellen Ripley')`);
+    const users = await pool.query(`SELECT * FROM users`);
+    expect(users.rows).toStrictEqual([{ id: 1, name: "Ellen Ripley" }]);
+});
+//# sourceMappingURL=recommended.test.js.map

package/dist/examples/recommended.test.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"recommended.test.js","sourceRoot":"","sources":["../../examples/recommended.test.ts"],"names":[],"mappings":"AAAA,sBAAsB;AACtB,OAAO,EAAE,MAAM,EAAE,IAAI,EAAE,MAAM,QAAQ,CAAC;AACtC,OAAO,EAAE,WAAW,EAAE,MAAM,gBAAgB,CAAC;AAC7C,OAAO,EAAE,IAAI,EAAE,MAAM,0BAA0B,CAAC;AAEhD,6CAA6C;AAC7C,WAAW,CAAC;IACV,4EAA4E;IAC5E,mBAAmB,EAAE,IAAI;CAC1B,CAAC,CAAC;AAEH,IAAI,CAAC,qBAAqB,EAAE,KAAK,IAAI,EAAE;IACrC,MAAM,IAAI,GAAG,IAAI,IAAI,CAAC,EAAE,gBAAgB,EAAE,OAAO,CAAC,GAAG,CAAC,YAAY,EAAE,CAAC,CAAC;IAEtE,MAAM,IAAI,CAAC,KAAK,CAAC,uDAAuD,CAAC,CAAC;IAC1E,MAAM,IAAI,CAAC,KAAK,CAAC,kDAAkD,CAAC,CAAC;IAErE,MAAM,KAAK,GAAG,MAAM,IAAI,CAAC,KAAK,CAAC,qBAAqB,CAAC,CAAC;IACtD,MAAM,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,aAAa,CAAC,CAAC,EAAE,EAAE,EAAE,CAAC,EAAE,IAAI,EAAE,cAAc,EAAE,CAAC,CAAC,CAAC;AACtE,CAAC,CAAC,CAAC"}

package/dist/index.d.ts

@@ -1,3 +1,7 @@
+/**
+ * https://neon.com/docs/reference/typescript-sdk
+ */
+import { type Branch } from "@neondatabase/api-client";
 export interface NeonTestingOptions {
     /**
      * The Neon API key, this is used to create and teardown test branches
@@ -40,6 +44,18 @@ export interface NeonTestingOptions {
      * connections
      */
     autoCloseWebSockets?: boolean;
+    /**
+     * Time in seconds until the branch expires and is automatically deleted
+     * (default: 600 = 10 minutes)
+     *
+     * This provides automatic cleanup for dangling branches from interrupted or
+     * failed test runs. Set to `null` to disable automatic expiration.
+     *
+     * Must be a positive integer. Maximum 30 days (2,592,000 seconds).
+     *
+     * https://neon.com/docs/guides/branch-expiration
+     */
+    expiresIn?: number | null;
 }
 /** Options for overriding test database setup (excludes apiKey) */
 export type NeonTestingOverrides = Omit<Partial<NeonTestingOptions>, "apiKey">;
@@ -47,21 +63,38 @@ export type NeonTestingOverrides = Omit<Partial<NeonTestingOptions>, "apiKey">;
  * Factory function that creates a Neon test database setup/teardown function
  * for Vitest test suites.
  *
- * @param apiKey - The Neon API key, this is used to create and teardown test branches
- * @param projectId - The Neon project ID to operate on
- * @param parentBranchId - The parent branch ID for the new branch. If omitted or empty, the branch will be created from the project's default branch.
- * @param schemaOnly - Whether to create a schema-only branch (default: false)
- * @param endpoint - The type of connection to create (pooler is recommended)
- * @param deleteBranch - Delete the test branch in afterAll (default: true). Disabling this will leave each test branch in the Neon project after the test suite runs
- * @returns A setup/teardown function for Vitest test suites
+ * @param factoryOptions - Configuration options (see {@link NeonTestingOptions})
+ * @returns A setup/teardown function with attached utilities:
+ *  - `deleteAllTestBranches()` - Cleanup method to delete all test branches
+ *  - `api` - Direct access to the Neon API client
+ *
+ * @example
+ * ```ts
+ * // neon-testing.ts
+ * import { makeNeonTesting } from "neon-testing";
+ *
+ * export const neonTesting = makeNeonTesting({
+ *   apiKey: "apiKey",
+ *   projectId: "projectId",
+ * });
+ * ```
+ *
+ * @example
+ * ```ts
+ * // my-test.test.ts
+ * import { neonTesting } from "./neon-testing";
+ *
+ * const getBranch = neonTesting();
  *
- * Side effects:
- * - Sets the `DATABASE_URL` environment variable to the connection URI for the
- *   new branch
- * - Deletes the test branch after the test suite runs
+ * test("my test", () => {
+ *   const branch = getBranch();
+ *   console.log(branch.id);
+ * });
+ * ```
  */
 export declare function makeNeonTesting(factoryOptions: NeonTestingOptions): {
-    (overrides?: NeonTestingOverrides): void;
+    (overrides?: NeonTestingOverrides): () => Branch | undefined;
     deleteAllTestBranches: () => Promise<void>;
+    api: import("@neondatabase/api-client").Api<unknown>;
 };
 //# sourceMappingURL=index.d.ts.map

package/dist/index.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../index.ts"],"names":[],"mappings":"AA8BA,MAAM,WAAW,kBAAkB;IACjC;;;;OAIG;IACH,MAAM,EAAE,MAAM,CAAC;IACf;;;;OAIG;IACH,SAAS,EAAE,MAAM,CAAC;IAClB;;;OAGG;IACH,cAAc,CAAC,EAAE,MAAM,CAAC;IACxB;;OAEG;IACH,UAAU,CAAC,EAAE,OAAO,CAAC;IACrB;;OAEG;IACH,QAAQ,CAAC,EAAE,QAAQ,GAAG,QAAQ,CAAC;IAC/B;;;;;OAKG;IACH,YAAY,CAAC,EAAE,OAAO,CAAC;IACvB;;;;;;;OAOG;IACH,mBAAmB,CAAC,EAAE,OAAO,CAAC;CAC/B;AAED,mEAAmE;AACnE,MAAM,MAAM,oBAAoB,GAAG,IAAI,CAAC,OAAO,CAAC,kBAAkB,CAAC,EAAE,QAAQ,CAAC,CAAC;AAE/E;;;;;;;;;;;;;;;;GAgBG;AACH,wBAAgB,eAAe,CAAC,cAAc,EAAE,kBAAkB;iBA0BlD,oBAAoB;;EAmGnC"}
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../index.ts"],"names":[],"mappings":"AAAA;;GAEG;AACH,OAAO,EAIL,KAAK,MAAM,EACZ,MAAM,0BAA0B,CAAC;AA0ClC,MAAM,WAAW,kBAAkB;IACjC;;;;OAIG;IACH,MAAM,EAAE,MAAM,CAAC;IACf;;;;OAIG;IACH,SAAS,EAAE,MAAM,CAAC;IAClB;;;OAGG;IACH,cAAc,CAAC,EAAE,MAAM,CAAC;IACxB;;OAEG;IACH,UAAU,CAAC,EAAE,OAAO,CAAC;IACrB;;OAEG;IACH,QAAQ,CAAC,EAAE,QAAQ,GAAG,QAAQ,CAAC;IAC/B;;;;;OAKG;IACH,YAAY,CAAC,EAAE,OAAO,CAAC;IACvB;;;;;;;OAOG;IACH,mBAAmB,CAAC,EAAE,OAAO,CAAC;IAC9B;;;;;;;;;;OAUG;IACH,SAAS,CAAC,EAAE,MAAM,GAAG,IAAI,CAAC;CAC3B;AAED,mEAAmE;AACnE,MAAM,MAAM,oBAAoB,GAAG,IAAI,CAAC,OAAO,CAAC,kBAAkB,CAAC,EAAE,QAAQ,CAAC,CAAC;AAE/E;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;GAgCG;AACH,wBAAgB,eAAe,CAAC,cAAc,EAAE,kBAAkB;iBAyClD,oBAAoB;;;EA0HnC"}

package/dist/index.js

@@ -16,27 +16,61 @@ function createConnectionUri(connectionParameters, type) {
     const hostname = type === "pooler" ? pooler_host : host;
     return `postgresql://${role}:${password}@${hostname}/${database}?sslmode=require`;
 }
+/**
+ * Validates the expiresIn option
+ */
+function validateExpiresIn(expiresIn) {
+    if (expiresIn !== null && expiresIn !== undefined) {
+        if (!Number.isInteger(expiresIn)) {
+            throw new Error("expiresIn must be an integer");
+        }
+        if (expiresIn <= 0) {
+            throw new Error("expiresIn must be a positive integer");
+        }
+        if (expiresIn > 2592000) {
+            throw new Error("expiresIn must not exceed 30 days (2,592,000 seconds)");
+        }
+    }
+}
 /**
  * Factory function that creates a Neon test database setup/teardown function
  * for Vitest test suites.
  *
- * @param apiKey - The Neon API key, this is used to create and teardown test branches
- * @param projectId - The Neon project ID to operate on
- * @param parentBranchId - The parent branch ID for the new branch. If omitted or empty, the branch will be created from the project's default branch.
- * @param schemaOnly - Whether to create a schema-only branch (default: false)
- * @param endpoint - The type of connection to create (pooler is recommended)
- * @param deleteBranch - Delete the test branch in afterAll (default: true). Disabling this will leave each test branch in the Neon project after the test suite runs
- * @returns A setup/teardown function for Vitest test suites
+ * @param factoryOptions - Configuration options (see {@link NeonTestingOptions})
+ * @returns A setup/teardown function with attached utilities:
+ *  - `deleteAllTestBranches()` - Cleanup method to delete all test branches
+ *  - `api` - Direct access to the Neon API client
+ *
+ * @example
+ * ```ts
+ * // neon-testing.ts
+ * import { makeNeonTesting } from "neon-testing";
+ *
+ * export const neonTesting = makeNeonTesting({
+ *   apiKey: "apiKey",
+ *   projectId: "projectId",
+ * });
+ * ```
  *
- * Side effects:
- * - Sets the `DATABASE_URL` environment variable to the connection URI for the
- *   new branch
- * - Deletes the test branch after the test suite runs
+ * @example
+ * ```ts
+ * // my-test.test.ts
+ * import { neonTesting } from "./neon-testing";
+ *
+ * const getBranch = neonTesting();
+ *
+ * test("my test", () => {
+ *   const branch = getBranch();
+ *   console.log(branch.id);
+ * });
+ * ```
  */
 export function makeNeonTesting(factoryOptions) {
+    // Validate factory options
+    validateExpiresIn(factoryOptions.expiresIn);
     const apiClient = createApiClient({ apiKey: factoryOptions.apiKey });
     /**
-     * Delete all test branches
+     * Delete all test branches (branches with the "integration-test: true" annotation)
      */
     async function deleteAllTestBranches() {
         const { data } = await apiClient.listProjectBranches({
@@ -49,13 +83,29 @@ export function makeNeonTesting(factoryOptions) {
             }
         }
     }
-    const testDbSetup = (
+    /**
+     * Setup/teardown function for Vitest test suites
+     *
+     * Registers Vitest lifecycle hooks that:
+     * - Create an isolated test branch from your parent branch
+     * - Set `DATABASE_URL` environment variable to the test branch connection URI
+     * - Automatically delete the test branch after tests complete (unless `deleteBranch: false`)
+     * - Automatically expire branches after 10 minutes for cleanup (unless `expiresIn: null`)
+     *
+     * @param overrides - Optional overrides for the factory options
+     * @returns A function that provides access to the current Neon branch object
+     */
+    const neonTesting = (
     /** Override any factory options except apiKey */
     overrides) => {
+        // Validate overrides
+        if (overrides?.expiresIn !== undefined) {
+            validateExpiresIn(overrides.expiresIn);
+        }
         // Merge factory options with overrides
         const options = { ...factoryOptions, ...overrides };
-        // Each test file gets its own branch ID and database client
-        let branchId;
+        // Each test file gets its own branch and database client
+        let branch;
         // List of tracked Neon WebSocket connections
         const neonSockets = new Set();
         // Custom WebSocket constructor that tracks Neon WebSocket connections
@@ -74,18 +124,24 @@ export function makeNeonTesting(factoryOptions) {
          * @returns The connection URI for the new branch
          */
         async function createBranch() {
+            // Calculate expiration timestamp if expiresIn is set
+            const expiresIn = options.expiresIn === undefined ? 600 : options.expiresIn; // Default: 10 minutes
+            const expiresAt = expiresIn !== null
+                ? new Date(Date.now() + expiresIn * 1000).toISOString()
+                : undefined;
             const { data } = await apiClient.createProjectBranch(options.projectId, {
                 branch: {
                     name: `test/${crypto.randomUUID()}`,
                     parent_id: options.parentBranchId,
                     init_source: options.schemaOnly ? "schema-only" : undefined,
+                    expires_at: expiresAt,
                 },
                 endpoints: [{ type: EndpointType.ReadWrite }],
                 annotation_value: {
                     "integration-test": "true",
                 },
             });
-            branchId = data.branch.id;
+            branch = data.branch;
             const [connectionUri] = data.connection_uris ?? [];
             if (!connectionUri) {
                 throw new Error("No connection URI found");
@@ -96,11 +152,11 @@ export function makeNeonTesting(factoryOptions) {
          * Delete the test branch
          */
         async function deleteBranch() {
-            if (!branchId) {
+            if (!branch?.id) {
                 throw new Error("No branch to delete");
             }
-            await apiClient.deleteProjectBranch(options.projectId, branchId);
-            branchId = undefined;
+            await apiClient.deleteProjectBranch(options.projectId, branch.id);
+            branch = undefined;
         }
         beforeAll(async () => {
             process.env.DATABASE_URL = await withRetry(createBranch, {
@@ -126,10 +182,17 @@ export function makeNeonTesting(factoryOptions) {
                 await deleteBranch();
             }
         });
+        /**
+         * Return the Neon branch object
+         *
+         * @returns The Neon branch object
+         */
+        return () => branch;
     };
-    // Attach the utility
-    testDbSetup.deleteAllTestBranches = deleteAllTestBranches;
-    return testDbSetup;
+    // Attach utilities
+    neonTesting.deleteAllTestBranches = deleteAllTestBranches;
+    neonTesting.api = apiClient;
+    return neonTesting;
 }
 /**
  * Error handler: Suppress Neon WebSocket "Connection terminated unexpectedly"

package/dist/index.js.map

@@ -1 +1 @@
-{"version":3,"file":"index.js","sourceRoot":"","sources":["../index.ts"],"names":[],"mappings":"AAAA;;GAEG;AACH,OAAO,EACL,eAAe,EACf,YAAY,GAEb,MAAM,0BAA0B,CAAC;AAClC,OAAO,EAAE,QAAQ,EAAE,SAAS,EAAE,MAAM,QAAQ,CAAC;AAC7C,OAAO,EAAE,UAAU,EAAE,MAAM,0BAA0B,CAAC;AAEtD;;;;;;GAMG;AACH,SAAS,mBAAmB,CAC1B,oBAAuC,EACvC,IAAyB;IAEzB,MAAM,EAAE,IAAI,EAAE,QAAQ,EAAE,IAAI,EAAE,WAAW,EAAE,QAAQ,EAAE,GACnD,oBAAoB,CAAC,qBAAqB,CAAC;IAE7C,MAAM,QAAQ,GAAG,IAAI,KAAK,QAAQ,CAAC,CAAC,CAAC,WAAW,CAAC,CAAC,CAAC,IAAI,CAAC;IAExD,OAAO,gBAAgB,IAAI,IAAI,QAAQ,IAAI,QAAQ,IAAI,QAAQ,kBAAkB,CAAC;AACpF,CAAC;AAiDD;;;;;;;;;;;;;;;;GAgBG;AACH,MAAM,UAAU,eAAe,CAAC,cAAkC;IAChE,MAAM,SAAS,GAAG,eAAe,CAAC,EAAE,MAAM,EAAE,cAAc,CAAC,MAAM,EAAE,CAAC,CAAC;IAErE;;OAEG;IACH,KAAK,UAAU,qBAAqB;QAClC,MAAM,EAAE,IAAI,EAAE,GAAG,MAAM,SAAS,CAAC,mBAAmB,CAAC;YACnD,SAAS,EAAE,cAAc,CAAC,SAAS;SACpC,CAAC,CAAC;QAEH,KAAK,MAAM,MAAM,IAAI,IAAI,CAAC,QAAQ,EAAE,CAAC;YACnC,MAAM,YAAY,GAChB,IAAI,CAAC,WAAW,CAAC,MAAM,CAAC,EAAE,CAAC,EAAE,KAAK,CAAC,kBAAkB,CAAC,KAAK,MAAM,CAAC;YAEpE,IAAI,YAAY,EAAE,CAAC;gBACjB,MAAM,SAAS,CAAC,mBAAmB,CACjC,cAAc,CAAC,SAAS,EACxB,MAAM,CAAC,EAAE,CACV,CAAC;YACJ,CAAC;QACH,CAAC;IACH,CAAC;IAED,MAAM,WAAW,GAAG;IAClB,iDAAiD;IACjD,SAAgC,EAChC,EAAE;QACF,uCAAuC;QACvC,MAAM,OAAO,GAAG,EAAE,GAAG,cAAc,EAAE,GAAG,SAAS,EAAE,CAAC;QAEpD,4DAA4D;QAC5D,IAAI,QAA4B,CAAC;QAEjC,6CAA6C;QAC7C,MAAM,WAAW,GAAG,IAAI,GAAG,EAAa,CAAC;QAEzC,sEAAsE;QACtE,MAAM,iBAAkB,SAAQ,SAAS;YACvC,YAAY,GAAW;gBACrB,KAAK,CAAC,GAAG,CAAC,CAAC;gBAEX,wCAAwC;gBACxC,IAAI,CAAC,GAAG,CAAC,QAAQ,CAAC,aAAa,CAAC;oBAAE,OAAO;gBAEzC,WAAW,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC;YACxB,CAAC;SACF;QAED;;;;WAIG;QACH,KAAK,UAAU,YAAY;YACzB,MAAM,EAAE,IAAI,EAAE,GAAG,MAAM,SAAS,CAAC,mBAAmB,CAAC,OAAO,CAAC,SAAS,EAAE;gBACtE,MAAM,EAAE;oBACN,IAAI,EAAE,QAAQ,MAAM,CAAC,UAAU,EAAE,EAAE;oBACnC,SAAS,EAAE,OAAO,CAAC,cAAc;oBACjC,WAAW,EAAE,OAAO,CAAC,UAAU,CAAC,CAAC,CAAC,aAAa,CAAC,CAAC,CAAC,SAAS;iBAC5D;gBACD,SAAS,EAAE,CAAC,EAAE,IAAI,EAAE,YAAY,CAAC,SAAS,EAAE,CAAC;gBAC7C,gBAAgB,EAAE;oBAChB,kBAAkB,EAAE,MAAM;iBAC3B;aACF,CAAC,CAAC;YAEH,QAAQ,GAAG,IAAI,CAAC,MAAM,CAAC,EAAE,CAAC;YAE1B,MAAM,CAAC,aAAa,CAAC,GAAG,IAAI,CAAC,eAAe,IAAI,EAAE,CAAC;YAEnD,IAAI,CAAC,aAAa,EAAE,CAAC;gBACnB,MAAM,IAAI,KAAK,CAAC,yBAAyB,CAAC,CAAC;YAC7C,CAAC;YAED,OAAO,mBAAmB,CAAC,aAAa,EAAE,OAAO,CAAC,QAAQ,IAAI,QAAQ,CAAC,CAAC;QAC1E,CAAC;QAED;;WAEG;QACH,KAAK,UAAU,YAAY;YACzB,IAAI,CAAC,QAAQ,EAAE,CAAC;gBACd,MAAM,IAAI,KAAK,CAAC,qBAAqB,CAAC,CAAC;YACzC,CAAC;YAED,MAAM,SAAS,CAAC,mBAAmB,CAAC,OAAO,CAAC,SAAS,EAAE,QAAQ,CAAC,CAAC;YACjE,QAAQ,GAAG,SAAS,CAAC;QACvB,CAAC;QAED,SAAS,CAAC,KAAK,IAAI,EAAE;YACnB,OAAO,CAAC,GAAG,CAAC,YAAY,GAAG,MAAM,SAAS,CAAC,YAAY,EAAE;gBACvD,UAAU,EAAE,CAAC;gBACb,WAAW,EAAE,IAAI;aAClB,CAAC,CAAC;YAEH,IAAI,OAAO,CAAC,mBAAmB,EAAE,CAAC;gBAChC,oEAAoE;gBACpE,yDAAyD;gBACzD,UAAU,CAAC,oBAAoB,GAAG,iBAAiB,CAAC;YACtD,CAAC;QACH,CAAC,CAAC,CAAC;QAEH,QAAQ,CAAC,KAAK,IAAI,EAAE;YAClB,OAAO,CAAC,GAAG,CAAC,YAAY,GAAG,SAAS,CAAC;YAErC,0EAA0E;YAC1E,IAAI,OAAO,CAAC,mBAAmB,EAAE,CAAC;gBAChC,qEAAqE;gBACrE,OAAO,CAAC,eAAe,CAAC,mBAAmB,EAAE,kBAAkB,CAAC,CAAC;gBAEjE,sEAAsE;gBACtE,WAAW,CAAC,OAAO,CAAC,CAAC,EAAE,EAAE,EAAE,CAAC,EAAE,CAAC,KAAK,EAAE,CAAC,CAAC;YAC1C,CAAC;YAED,IAAI,OAAO,CAAC,YAAY,KAAK,KAAK,EAAE,CAAC;gBACnC,MAAM,YAAY,EAAE,CAAC;YACvB,CAAC;QACH,CAAC,CAAC,CAAC;IACL,CAAC,CAAC;IAEF,qBAAqB;IACrB,WAAW,CAAC,qBAAqB,GAAG,qBAAqB,CAAC;IAE1D,OAAO,WAAW,CAAC;AACrB,CAAC;AAED;;;GAGG;AACH,MAAM,kBAAkB,GAAG,CAAC,KAAY,EAAE,EAAE;IAC1C,MAAM,aAAa,GACjB,KAAK,CAAC,OAAO,CAAC,QAAQ,CAAC,oCAAoC,CAAC;QAC5D,KAAK,CAAC,KAAK,EAAE,QAAQ,CAAC,0BAA0B,CAAC,CAAC;IAEpD,IAAI,aAAa,EAAE,CAAC;QAClB,kDAAkD;QAClD,OAAO;IACT,CAAC;IAED,0CAA0C;IAC1C,MAAM,KAAK,CAAC;AACd,CAAC,CAAC;AAEF;;;;;;GAMG;AACH,KAAK,UAAU,SAAS,CACtB,EAAoB,EACpB,OAGC;IAED,IAAI,CAAC,MAAM,CAAC,SAAS,CAAC,OAAO,CAAC,UAAU,CAAC,IAAI,OAAO,CAAC,UAAU,IAAI,CAAC,EAAE,CAAC;QACrE,MAAM,IAAI,KAAK,CAAC,uCAAuC,CAAC,CAAC;IAC3D,CAAC;IAED,IAAI,CAAC,MAAM,CAAC,SAAS,CAAC,OAAO,CAAC,WAAW,CAAC,IAAI,OAAO,CAAC,WAAW,IAAI,CAAC,EAAE,CAAC;QACvE,MAAM,IAAI,KAAK,CAAC,wCAAwC,CAAC,CAAC;IAC5D,CAAC;IAED,KAAK,IAAI,OAAO,GAAG,CAAC,EAAE,OAAO,IAAI,OAAO,CAAC,UAAU,EAAE,OAAO,EAAE,EAAE,CAAC;QAC/D,IAAI,CAAC;YACH,OAAO,MAAM,EAAE,EAAE,CAAC;QACpB,CAAC;QAAC,OAAO,KAAU,EAAE,CAAC;YACpB,MAAM,MAAM,GAAG,KAAK,EAAE,QAAQ,EAAE,MAAM,CAAC;YAEvC,IAAI,MAAM,KAAK,GAAG,IAAI,OAAO,GAAG,OAAO,CAAC,UAAU,EAAE,CAAC;gBACnD,MAAM,KAAK,GAAG,OAAO,CAAC,WAAW,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,OAAO,GAAG,CAAC,CAAC,CAAC;gBAE7D,OAAO,CAAC,GAAG,CACT,yCAAyC,KAAK,eAAe,OAAO,IAAI,OAAO,CAAC,UAAU,GAAG,CAC9F,CAAC;gBACF,MAAM,IAAI,OAAO,CAAC,CAAC,OAAO,EAAE,EAAE,CAAC,UAAU,CAAC,OAAO,EAAE,KAAK,CAAC,CAAC,CAAC;gBAE3D,SAAS;YACX,CAAC;YAED,MAAM,KAAK,CAAC;QACd,CAAC;IACH,CAAC;IACD,MAAM,IAAI,KAAK,CAAC,yCAAyC,CAAC,CAAC;AAC7D,CAAC"}
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../index.ts"],"names":[],"mappings":"AAAA;;GAEG;AACH,OAAO,EACL,eAAe,EACf,YAAY,GAGb,MAAM,0BAA0B,CAAC;AAClC,OAAO,EAAE,QAAQ,EAAE,SAAS,EAAE,MAAM,QAAQ,CAAC;AAC7C,OAAO,EAAE,UAAU,EAAE,MAAM,0BAA0B,CAAC;AAEtD;;;;;;GAMG;AACH,SAAS,mBAAmB,CAC1B,oBAAuC,EACvC,IAAyB;IAEzB,MAAM,EAAE,IAAI,EAAE,QAAQ,EAAE,IAAI,EAAE,WAAW,EAAE,QAAQ,EAAE,GACnD,oBAAoB,CAAC,qBAAqB,CAAC;IAE7C,MAAM,QAAQ,GAAG,IAAI,KAAK,QAAQ,CAAC,CAAC,CAAC,WAAW,CAAC,CAAC,CAAC,IAAI,CAAC;IAExD,OAAO,gBAAgB,IAAI,IAAI,QAAQ,IAAI,QAAQ,IAAI,QAAQ,kBAAkB,CAAC;AACpF,CAAC;AAED;;GAEG;AACH,SAAS,iBAAiB,CAAC,SAAoC;IAC7D,IAAI,SAAS,KAAK,IAAI,IAAI,SAAS,KAAK,SAAS,EAAE,CAAC;QAClD,IAAI,CAAC,MAAM,CAAC,SAAS,CAAC,SAAS,CAAC,EAAE,CAAC;YACjC,MAAM,IAAI,KAAK,CAAC,8BAA8B,CAAC,CAAC;QAClD,CAAC;QAED,IAAI,SAAS,IAAI,CAAC,EAAE,CAAC;YACnB,MAAM,IAAI,KAAK,CAAC,sCAAsC,CAAC,CAAC;QAC1D,CAAC;QAED,IAAI,SAAS,GAAG,OAAO,EAAE,CAAC;YACxB,MAAM,IAAI,KAAK,CAAC,uDAAuD,CAAC,CAAC;QAC3E,CAAC;IACH,CAAC;AACH,CAAC;AA6DD;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;GAgCG;AACH,MAAM,UAAU,eAAe,CAAC,cAAkC;IAChE,2BAA2B;IAC3B,iBAAiB,CAAC,cAAc,CAAC,SAAS,CAAC,CAAC;IAE5C,MAAM,SAAS,GAAG,eAAe,CAAC,EAAE,MAAM,EAAE,cAAc,CAAC,MAAM,EAAE,CAAC,CAAC;IAErE;;OAEG;IACH,KAAK,UAAU,qBAAqB;QAClC,MAAM,EAAE,IAAI,EAAE,GAAG,MAAM,SAAS,CAAC,mBAAmB,CAAC;YACnD,SAAS,EAAE,cAAc,CAAC,SAAS;SACpC,CAAC,CAAC;QAEH,KAAK,MAAM,MAAM,IAAI,IAAI,CAAC,QAAQ,EAAE,CAAC;YACnC,MAAM,YAAY,GAChB,IAAI,CAAC,WAAW,CAAC,MAAM,CAAC,EAAE,CAAC,EAAE,KAAK,CAAC,kBAAkB,CAAC,KAAK,MAAM,CAAC;YAEpE,IAAI,YAAY,EAAE,CAAC;gBACjB,MAAM,SAAS,CAAC,mBAAmB,CACjC,cAAc,CAAC,SAAS,EACxB,MAAM,CAAC,EAAE,CACV,CAAC;YACJ,CAAC;QACH,CAAC;IACH,CAAC;IAED;;;;;;;;;;;OAWG;IACH,MAAM,WAAW,GAAG;IAClB,iDAAiD;IACjD,SAAgC,EAChC,EAAE;QACF,qBAAqB;QACrB,IAAI,SAAS,EAAE,SAAS,KAAK,SAAS,EAAE,CAAC;YACvC,iBAAiB,CAAC,SAAS,CAAC,SAAS,CAAC,CAAC;QACzC,CAAC;QAED,uCAAuC;QACvC,MAAM,OAAO,GAAG,EAAE,GAAG,cAAc,EAAE,GAAG,SAAS,EAAE,CAAC;QAEpD,yDAAyD;QACzD,IAAI,MAA0B,CAAC;QAE/B,6CAA6C;QAC7C,MAAM,WAAW,GAAG,IAAI,GAAG,EAAa,CAAC;QAEzC,sEAAsE;QACtE,MAAM,iBAAkB,SAAQ,SAAS;YACvC,YAAY,GAAW;gBACrB,KAAK,CAAC,GAAG,CAAC,CAAC;gBAEX,wCAAwC;gBACxC,IAAI,CAAC,GAAG,CAAC,QAAQ,CAAC,aAAa,CAAC;oBAAE,OAAO;gBAEzC,WAAW,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC;YACxB,CAAC;SACF;QAED;;;;WAIG;QACH,KAAK,UAAU,YAAY;YACzB,qDAAqD;YACrD,MAAM,SAAS,GACb,OAAO,CAAC,SAAS,KAAK,SAAS,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,OAAO,CAAC,SAAS,CAAC,CAAC,sBAAsB;YAEnF,MAAM,SAAS,GACb,SAAS,KAAK,IAAI;gBAChB,CAAC,CAAC,IAAI,IAAI,CAAC,IAAI,CAAC,GAAG,EAAE,GAAG,SAAS,GAAG,IAAI,CAAC,CAAC,WAAW,EAAE;gBACvD,CAAC,CAAC,SAAS,CAAC;YAEhB,MAAM,EAAE,IAAI,EAAE,GAAG,MAAM,SAAS,CAAC,mBAAmB,CAAC,OAAO,CAAC,SAAS,EAAE;gBACtE,MAAM,EAAE;oBACN,IAAI,EAAE,QAAQ,MAAM,CAAC,UAAU,EAAE,EAAE;oBACnC,SAAS,EAAE,OAAO,CAAC,cAAc;oBACjC,WAAW,EAAE,OAAO,CAAC,UAAU,CAAC,CAAC,CAAC,aAAa,CAAC,CAAC,CAAC,SAAS;oBAC3D,UAAU,EAAE,SAAS;iBACtB;gBACD,SAAS,EAAE,CAAC,EAAE,IAAI,EAAE,YAAY,CAAC,SAAS,EAAE,CAAC;gBAC7C,gBAAgB,EAAE;oBAChB,kBAAkB,EAAE,MAAM;iBAC3B;aACF,CAAC,CAAC;YAEH,MAAM,GAAG,IAAI,CAAC,MAAM,CAAC;YAErB,MAAM,CAAC,aAAa,CAAC,GAAG,IAAI,CAAC,eAAe,IAAI,EAAE,CAAC;YAEnD,IAAI,CAAC,aAAa,EAAE,CAAC;gBACnB,MAAM,IAAI,KAAK,CAAC,yBAAyB,CAAC,CAAC;YAC7C,CAAC;YAED,OAAO,mBAAmB,CAAC,aAAa,EAAE,OAAO,CAAC,QAAQ,IAAI,QAAQ,CAAC,CAAC;QAC1E,CAAC;QAED;;WAEG;QACH,KAAK,UAAU,YAAY;YACzB,IAAI,CAAC,MAAM,EAAE,EAAE,EAAE,CAAC;gBAChB,MAAM,IAAI,KAAK,CAAC,qBAAqB,CAAC,CAAC;YACzC,CAAC;YAED,MAAM,SAAS,CAAC,mBAAmB,CAAC,OAAO,CAAC,SAAS,EAAE,MAAM,CAAC,EAAE,CAAC,CAAC;YAClE,MAAM,GAAG,SAAS,CAAC;QACrB,CAAC;QAED,SAAS,CAAC,KAAK,IAAI,EAAE;YACnB,OAAO,CAAC,GAAG,CAAC,YAAY,GAAG,MAAM,SAAS,CAAC,YAAY,EAAE;gBACvD,UAAU,EAAE,CAAC;gBACb,WAAW,EAAE,IAAI;aAClB,CAAC,CAAC;YAEH,IAAI,OAAO,CAAC,mBAAmB,EAAE,CAAC;gBAChC,oEAAoE;gBACpE,yDAAyD;gBACzD,UAAU,CAAC,oBAAoB,GAAG,iBAAiB,CAAC;YACtD,CAAC;QACH,CAAC,CAAC,CAAC;QAEH,QAAQ,CAAC,KAAK,IAAI,EAAE;YAClB,OAAO,CAAC,GAAG,CAAC,YAAY,GAAG,SAAS,CAAC;YAErC,0EAA0E;YAC1E,IAAI,OAAO,CAAC,mBAAmB,EAAE,CAAC;gBAChC,qEAAqE;gBACrE,OAAO,CAAC,eAAe,CAAC,mBAAmB,EAAE,kBAAkB,CAAC,CAAC;gBAEjE,sEAAsE;gBACtE,WAAW,CAAC,OAAO,CAAC,CAAC,EAAE,EAAE,EAAE,CAAC,EAAE,CAAC,KAAK,EAAE,CAAC,CAAC;YAC1C,CAAC;YAED,IAAI,OAAO,CAAC,YAAY,KAAK,KAAK,EAAE,CAAC;gBACnC,MAAM,YAAY,EAAE,CAAC;YACvB,CAAC;QACH,CAAC,CAAC,CAAC;QAEH;;;;WAIG;QACH,OAAO,GAAG,EAAE,CAAC,MAAM,CAAC;IACtB,CAAC,CAAC;IAEF,mBAAmB;IACnB,WAAW,CAAC,qBAAqB,GAAG,qBAAqB,CAAC;IAC1D,WAAW,CAAC,GAAG,GAAG,SAAS,CAAC;IAE5B,OAAO,WAAW,CAAC;AACrB,CAAC;AAED;;;GAGG;AACH,MAAM,kBAAkB,GAAG,CAAC,KAAY,EAAE,EAAE;IAC1C,MAAM,aAAa,GACjB,KAAK,CAAC,OAAO,CAAC,QAAQ,CAAC,oCAAoC,CAAC;QAC5D,KAAK,CAAC,KAAK,EAAE,QAAQ,CAAC,0BAA0B,CAAC,CAAC;IAEpD,IAAI,aAAa,EAAE,CAAC;QAClB,kDAAkD;QAClD,OAAO;IACT,CAAC;IAED,0CAA0C;IAC1C,MAAM,KAAK,CAAC;AACd,CAAC,CAAC;AAEF;;;;;;GAMG;AACH,KAAK,UAAU,SAAS,CACtB,EAAoB,EACpB,OAGC;IAED,IAAI,CAAC,MAAM,CAAC,SAAS,CAAC,OAAO,CAAC,UAAU,CAAC,IAAI,OAAO,CAAC,UAAU,IAAI,CAAC,EAAE,CAAC;QACrE,MAAM,IAAI,KAAK,CAAC,uCAAuC,CAAC,CAAC;IAC3D,CAAC;IAED,IAAI,CAAC,MAAM,CAAC,SAAS,CAAC,OAAO,CAAC,WAAW,CAAC,IAAI,OAAO,CAAC,WAAW,IAAI,CAAC,EAAE,CAAC;QACvE,MAAM,IAAI,KAAK,CAAC,wCAAwC,CAAC,CAAC;IAC5D,CAAC;IAED,KAAK,IAAI,OAAO,GAAG,CAAC,EAAE,OAAO,IAAI,OAAO,CAAC,UAAU,EAAE,OAAO,EAAE,EAAE,CAAC;QAC/D,IAAI,CAAC;YACH,OAAO,MAAM,EAAE,EAAE,CAAC;QACpB,CAAC;QAAC,OAAO,KAAU,EAAE,CAAC;YACpB,MAAM,MAAM,GAAG,KAAK,EAAE,QAAQ,EAAE,MAAM,CAAC;YAEvC,IAAI,MAAM,KAAK,GAAG,IAAI,OAAO,GAAG,OAAO,CAAC,UAAU,EAAE,CAAC;gBACnD,MAAM,KAAK,GAAG,OAAO,CAAC,WAAW,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,OAAO,GAAG,CAAC,CAAC,CAAC;gBAE7D,OAAO,CAAC,GAAG,CACT,yCAAyC,KAAK,eAAe,OAAO,IAAI,OAAO,CAAC,UAAU,GAAG,CAC9F,CAAC;gBACF,MAAM,IAAI,OAAO,CAAC,CAAC,OAAO,EAAE,EAAE,CAAC,UAAU,CAAC,OAAO,EAAE,KAAK,CAAC,CAAC,CAAC;gBAE3D,SAAS;YACX,CAAC;YAED,MAAM,KAAK,CAAC;QACd,CAAC;IACH,CAAC;IACD,MAAM,IAAI,KAAK,CAAC,yCAAyC,CAAC,CAAC;AAC7D,CAAC"}

package/dist/vitest.config.d.ts

@@ -0,0 +1,3 @@
+declare const _default: import("vite").UserConfig;
+export default _default;
+//# sourceMappingURL=vitest.config.d.ts.map

package/dist/vitest.config.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"vitest.config.d.ts","sourceRoot":"","sources":["../vitest.config.ts"],"names":[],"mappings":";AAOA,wBASG"}

package/dist/vitest.config.js

@@ -0,0 +1,16 @@
+import { defineConfig, configDefaults } from "vitest/config";
+import dotenv from "dotenv";
+import tsconfigPaths from "vite-tsconfig-paths";
+import { neonTesting } from "./vite-plugin";
+dotenv.config();
+export default defineConfig({
+    test: {
+        testTimeout: 10000,
+        hookTimeout: 20000,
+        exclude: [...configDefaults.exclude, "**/dist/**"],
+        // Limit to concurrent test files to avoid Neon API rate limiting
+        maxConcurrency: 2,
+    },
+    plugins: [tsconfigPaths(), neonTesting({ debug: false })],
+});
+//# sourceMappingURL=vitest.config.js.map

package/dist/vitest.config.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"vitest.config.js","sourceRoot":"","sources":["../vitest.config.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,YAAY,EAAE,cAAc,EAAE,MAAM,eAAe,CAAC;AAC7D,OAAO,MAAM,MAAM,QAAQ,CAAC;AAC5B,OAAO,aAAa,MAAM,qBAAqB,CAAC;AAChD,OAAO,EAAE,WAAW,EAAE,MAAM,eAAe,CAAC;AAE5C,MAAM,CAAC,MAAM,EAAE,CAAC;AAEhB,eAAe,YAAY,CAAC;IAC1B,IAAI,EAAE;QACJ,WAAW,EAAE,KAAK;QAClB,WAAW,EAAE,KAAK;QAClB,OAAO,EAAE,CAAC,GAAG,cAAc,CAAC,OAAO,EAAE,YAAY,CAAC;QAClD,iEAAiE;QACjE,cAAc,EAAE,CAAC;KAClB;IACD,OAAO,EAAE,CAAC,aAAa,EAAE,EAAE,WAAW,CAAC,EAAE,KAAK,EAAE,KAAK,EAAE,CAAC,CAAC;CAC1D,CAAC,CAAC"}

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "neon-testing",
-  "version": "2.2.0",
+  "version": "2.2.1-beta.0",
   "description": "A Vitest utility for seamless integration tests with Neon Postgres",
   "keywords": [
     "neon",
@@ -54,15 +54,17 @@
   },
   "devDependencies": {
     "@neondatabase/serverless": "^1.0.2",
+    "@types/pg": "^8.15.6",
     "dotenv": "^17.2.3",
-    "drizzle-orm": "^0.44.6",
+    "drizzle-orm": "^0.44.7",
     "pg": "^8.16.3",
     "postgres": "^3.4.7",
     "prettier": "^3.6.2",
+    "tiny-invariant": "^1.3.3",
     "typescript": "^5.9.3",
-    "vite": "^7.1.12",
+    "vite": "^7.2.2",
     "vite-tsconfig-paths": "^5.1.4",
-    "vitest": "^4.0.1"
+    "vitest": "^4.0.8"
   },
   "browser": {
     "./dist/vite-plugin.js": "./dist/browser-empty.js"