@empiricalrun/playwright-utils 0.32.0 → 0.34.0

package/CHANGELOG.md

@@ -1,5 +1,24 @@
 # @empiricalrun/playwright-utils
 
+## 0.34.0
+
+### Minor Changes
+
+- 88dd018: feat: proxy endpoint for neon db on dash
+
+## 0.33.0
+
+### Minor Changes
+
+- 6c4f6de: feat: support html reporter for 1.57.0
+- a94a7f7: fix: resolve multipart upload retry failures
+
+### Patch Changes
+
+- Updated dependencies [a94a7f7]
+  - @empiricalrun/r2-uploader@0.5.0
+  - @empiricalrun/test-gen@0.78.3
+
 ## 0.32.0
 
 ### Minor Changes

package/README.md

@@ -13,6 +13,8 @@ Playwright utils for test code repos of our customers
   - Captcha
 - [Email automation](./docs/email.md)
 - [Authentication](./docs/auth.md)
+- [KV Store](./docs/kv.md)
+- [Postgres](./docs/postgres.md)
 
 ## Development
 

package/dist/index.d.ts

@@ -5,4 +5,5 @@ export { baseConfig, chromeStablePath, devices } from "./config";
 export * from "./email";
 export * from "./kv";
 export * from "./playwright-extensions";
+export * from "./postgres";
 //# sourceMappingURL=index.d.ts.map

package/dist/index.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AAAA,cAAc,QAAQ,CAAC;AACvB,cAAc,eAAe,CAAC;AAC9B,OAAO,EAAE,cAAc,EAAE,MAAM,WAAW,CAAC;AAC3C,OAAO,EAAE,UAAU,EAAE,gBAAgB,EAAE,OAAO,EAAE,MAAM,UAAU,CAAC;AACjE,cAAc,SAAS,CAAC;AACxB,cAAc,MAAM,CAAC;AACrB,cAAc,yBAAyB,CAAC"}
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AAAA,cAAc,QAAQ,CAAC;AACvB,cAAc,eAAe,CAAC;AAC9B,OAAO,EAAE,cAAc,EAAE,MAAM,WAAW,CAAC;AAC3C,OAAO,EAAE,UAAU,EAAE,gBAAgB,EAAE,OAAO,EAAE,MAAM,UAAU,CAAC;AACjE,cAAc,SAAS,CAAC;AACxB,cAAc,MAAM,CAAC;AACrB,cAAc,yBAAyB,CAAC;AACxC,cAAc,YAAY,CAAC"}

package/dist/index.js

@@ -26,3 +26,4 @@ Object.defineProperty(exports, "devices", { enumerable: true, get: function () {
 __exportStar(require("./email"), exports);
 __exportStar(require("./kv"), exports);
 __exportStar(require("./playwright-extensions"), exports);
+__exportStar(require("./postgres"), exports);

package/dist/postgres.d.ts

@@ -0,0 +1,24 @@
+type PostgresDatabase = {
+    projectId: string;
+    connectionUri: string;
+};
+type GetOptions = {
+    name?: string;
+    pgVersion?: number;
+    ttl?: number;
+};
+export declare class PostgresClient {
+    private client;
+    private kv;
+    constructor();
+    get(name: string, options?: GetOptions): Promise<PostgresDatabase>;
+    delete(name: string): Promise<void>;
+    query<T = Record<string, unknown>>(connectionUri: string, sql: string): Promise<T[]>;
+    execute(connectionUri: string, sql: string): Promise<{
+        statement: string;
+        rowCount: number;
+    }[]>;
+    private parseStatements;
+}
+export {};
+//# sourceMappingURL=postgres.d.ts.map

package/dist/postgres.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"postgres.d.ts","sourceRoot":"","sources":["../src/postgres.ts"],"names":[],"mappings":"AA6BA,KAAK,gBAAgB,GAAG;IACtB,SAAS,EAAE,MAAM,CAAC;IAClB,aAAa,EAAE,MAAM,CAAC;CACvB,CAAC;AAEF,KAAK,UAAU,GAAG;IAChB,IAAI,CAAC,EAAE,MAAM,CAAC;IACd,SAAS,CAAC,EAAE,MAAM,CAAC;IACnB,GAAG,CAAC,EAAE,MAAM,CAAC;CACd,CAAC;AAIF,qBAAa,cAAc;IACzB,OAAO,CAAC,MAAM,CAAqB;IACnC,OAAO,CAAC,EAAE,CAAW;;IAYf,GAAG,CAAC,IAAI,EAAE,MAAM,EAAE,OAAO,CAAC,EAAE,UAAU,GAAG,OAAO,CAAC,gBAAgB,CAAC;IAuClE,MAAM,CAAC,IAAI,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC;IAwBnC,KAAK,CAAC,CAAC,GAAG,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,EACrC,aAAa,EAAE,MAAM,EACrB,GAAG,EAAE,MAAM,GACV,OAAO,CAAC,CAAC,EAAE,CAAC;IAWT,OAAO,CACX,aAAa,EAAE,MAAM,EACrB,GAAG,EAAE,MAAM,GACV,OAAO,CAAC;QAAE,SAAS,EAAE,MAAM,CAAC;QAAC,QAAQ,EAAE,MAAM,CAAA;KAAE,EAAE,CAAC;IAsBrD,OAAO,CAAC,eAAe;CAsBxB"}

package/dist/postgres.js

@@ -0,0 +1,117 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.PostgresClient = void 0;
+const dashboard_1 = require("@empiricalrun/test-gen/dashboard");
+const pg_1 = require("pg");
+const kv_1 = require("./kv");
+const DEFAULT_TTL = 3600;
+class PostgresClient {
+    client;
+    kv;
+    constructor() {
+        if (!process.env.EMPIRICALRUN_API_KEY) {
+            throw new Error("EMPIRICALRUN_API_KEY environment variable is required");
+        }
+        this.client = new dashboard_1.DashboardAPIClient({
+            authType: "project-api-key",
+        });
+        this.kv = new kv_1.KVClient();
+    }
+    async get(name, options) {
+        const kvKey = `postgres:${name}`;
+        const existing = await this.kv.get(kvKey);
+        if (existing) {
+            return existing;
+        }
+        const response = await this.client.request("/api/neon/proxy", {
+            method: "POST",
+            body: {
+                action: "create",
+                payload: {
+                    name: options?.name ?? name,
+                    pg_version: options?.pgVersion,
+                },
+            },
+        });
+        if (response.error) {
+            throw new Error(`Postgres create failed: ${response.error}`);
+        }
+        const projectId = response.data.project.id;
+        const connectionUri = response.data.connection_uris?.[0]?.connection_uri;
+        if (!connectionUri) {
+            throw new Error("No connection URI returned from Neon API");
+        }
+        const db = { projectId, connectionUri };
+        await this.kv.set(kvKey, db, options?.ttl ?? DEFAULT_TTL);
+        return db;
+    }
+    async delete(name) {
+        const kvKey = `postgres:${name}`;
+        const existing = await this.kv.get(kvKey);
+        if (!existing) {
+            return;
+        }
+        const response = await this.client.request("/api/neon/proxy", {
+            method: "POST",
+            body: {
+                action: "delete",
+                projectId: existing.projectId,
+            },
+        });
+        if (response.error) {
+            throw new Error(`Postgres delete failed: ${response.error}`);
+        }
+    }
+    async query(connectionUri, sql) {
+        const client = new pg_1.Client({ connectionString: connectionUri });
+        try {
+            await client.connect();
+            const result = await client.query(sql);
+            return result.rows;
+        }
+        finally {
+            await client.end();
+        }
+    }
+    async execute(connectionUri, sql) {
+        const client = new pg_1.Client({ connectionString: connectionUri });
+        const results = [];
+        try {
+            await client.connect();
+            const statements = this.parseStatements(sql);
+            for (const statement of statements) {
+                const result = await client.query(statement);
+                results.push({
+                    statement: statement.substring(0, 80),
+                    rowCount: result.rowCount ?? 0,
+                });
+            }
+            return results;
+        }
+        finally {
+            await client.end();
+        }
+    }
+    parseStatements(sql) {
+        const lines = sql.split("\n");
+        const cleanedLines = [];
+        for (const line of lines) {
+            const trimmed = line.trim();
+            if (trimmed && !trimmed.startsWith("--")) {
+                const commentIndex = line.indexOf("--");
+                if (commentIndex > 0) {
+                    cleanedLines.push(line.substring(0, commentIndex));
+                }
+                else {
+                    cleanedLines.push(line);
+                }
+            }
+        }
+        return cleanedLines
+            .join("\n")
+            .split(";")
+            .map((q) => q.trim())
+            .filter((q) => q.length > 0);
+    }
+}
+exports.PostgresClient = PostgresClient;

package/dist/reporter/empirical-reporter.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"empirical-reporter.d.ts","sourceRoot":"","sources":["../../src/reporter/empirical-reporter.ts"],"names":[],"mappings":"AAMA,OAAO,KAAK,EACV,UAAU,EACV,QAAQ,EACR,QAAQ,EACR,UAAU,EACX,MAAM,2BAA2B,CAAC;AAanC,cAAM,iBAAkB,YAAW,QAAQ;IACzC,OAAO,CAAC,eAAe,CAAS;IAChC,OAAO,CAAC,aAAa,CAAyB;IAC9C,OAAO,CAAC,kBAAkB,CAAyB;IACnD,OAAO,CAAC,oBAAoB,CAAkB;IAC9C,OAAO,CAAC,oBAAoB,CAG1B;IACF,OAAO,CAAC,QAAQ,CAAmC;IACnD,OAAO,CAAC,UAAU,CACwE;IAE1F,OAAO,CAAC,YAAY;;IAyBpB,OAAO,CAAC,6BAA6B,CA4BnC;IAEF,SAAS,CAAC,IAAI,EAAE,QAAQ,EAAE,MAAM,EAAE,UAAU;IAsEtC,KAAK,CAAC,MAAM,EAAE,UAAU;CA+G/B;AAED,eAAe,iBAAiB,CAAC"}
+{"version":3,"file":"empirical-reporter.d.ts","sourceRoot":"","sources":["../../src/reporter/empirical-reporter.ts"],"names":[],"mappings":"AAMA,OAAO,KAAK,EACV,UAAU,EACV,QAAQ,EACR,QAAQ,EACR,UAAU,EACX,MAAM,2BAA2B,CAAC;AAanC,cAAM,iBAAkB,YAAW,QAAQ;IACzC,OAAO,CAAC,eAAe,CAAS;IAChC,OAAO,CAAC,aAAa,CAAyB;IAC9C,OAAO,CAAC,kBAAkB,CAAyB;IACnD,OAAO,CAAC,oBAAoB,CAAkB;IAC9C,OAAO,CAAC,oBAAoB,CAG1B;IACF,OAAO,CAAC,QAAQ,CAAmC;IACnD,OAAO,CAAC,UAAU,CACwE;IAE1F,OAAO,CAAC,YAAY;;IAyBpB,OAAO,CAAC,6BAA6B,CA4BnC;IAEF,SAAS,CAAC,IAAI,EAAE,QAAQ,EAAE,MAAM,EAAE,UAAU;IA8EtC,KAAK,CAAC,MAAM,EAAE,UAAU;CA+G/B;AAED,eAAe,iBAAiB,CAAC"}

package/dist/reporter/empirical-reporter.js

@@ -67,49 +67,54 @@ class EmpiricalReporter {
         }
         try {
             const attachmentPromises = result.attachments.map(async (attachment) => {
-                const fileMap = await this.enqueTestAttachmentUploadTask(attachment);
-                if (!fileMap) {
-                    logger_1.logger.error(`No upload result for ${attachment}`);
-                    return undefined;
-                }
-                // Since we are uploading a single attachmnet, the fileMap will have one entry
-                const url = Object.values(fileMap)[0];
-                if (!url) {
-                    logger_1.logger.error(`No url in file map for ${attachment}`);
-                    return undefined;
+                try {
+                    const fileMap = await this.enqueTestAttachmentUploadTask(attachment);
+                    if (!fileMap) {
+                        logger_1.logger.error(`No upload result for ${attachment.name}`);
+                        return undefined;
+                    }
+                    const url = Object.values(fileMap)[0];
+                    if (!url) {
+                        logger_1.logger.error(`No url in file map for ${attachment.name}`);
+                        return undefined;
+                    }
+                    if (!attachment.path) {
+                        logger_1.logger.error(`No path in attachment for ${attachment.name}`);
+                        return undefined;
+                    }
+                    return {
+                        name: attachment.name,
+                        contentType: attachment.contentType,
+                        path: attachment.path,
+                        url,
+                    };
                 }
-                if (!attachment.path) {
-                    // This should never happen because we only upload if attachment.path is there
-                    logger_1.logger.error(`No path in file map for ${attachment}`);
+                catch (err) {
+                    logger_1.logger.error(`[Empirical Reporter] Error uploading attachment ${attachment.name} for test ${test.title}:`, err);
                     return undefined;
                 }
-                return {
-                    name: attachment.name,
-                    contentType: attachment.contentType,
-                    path: attachment.path,
-                    url,
-                };
             });
             const testCaseRunEventTask = async () => {
-                return Promise.all(attachmentPromises)
-                    .then((uploadedAttachments) => {
-                    logger_1.logger.debug(`[Empirical Reporter] Attachments for test ${test.title} are uploaded:`, uploadedAttachments);
+                const uploadedAttachments = await Promise.all(attachmentPromises);
+                const successfulAttachments = uploadedAttachments.filter((a) => a !== undefined);
+                try {
+                    logger_1.logger.debug(`[Empirical Reporter] Attachments for test ${test.title} are uploaded:`, successfulAttachments);
                     const { suites, projectName } = (0, util_1.suitesAndProjectForTest)(test);
                     const params = {
                         test,
                         suites,
                         result,
                         projectName,
-                        attachments: uploadedAttachments.filter((a) => a !== undefined),
+                        attachments: successfulAttachments,
                         runId: process.env.TEST_RUN_GITHUB_ACTION_ID,
                     };
                     return (0, util_1.sendTestCaseUpdateToDashboard)(params);
-                })
-                    .catch((error) => {
+                }
+                catch (error) {
                     logger_1.logger.error(`[Empirical Reporter] Error sending test case event for: ${test.title}:`, error);
-                });
+                }
             };
-            void (0, r2_uploader_1.sendTaskToQueue)(testCaseRunEventTask);
+            void testCaseRunEventTask();
             return;
         }
         catch (error) {

package/dist/reporter/util.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"util.d.ts","sourceRoot":"","sources":["../../src/reporter/util.ts"],"names":[],"mappings":"AAAA,OAAO,EACL,2BAA2B,EAC3B,qBAAqB,EACtB,MAAM,6CAA6C,CAAC;AACrD,OAAO,EACL,cAAc,EAGd,UAAU,IAAI,oBAAoB,EAElC,QAAQ,EACT,MAAM,2BAA2B,CAAC;AAoBnC,wBAAgB,kBAAkB,CAAC,UAAU,EAAE,MAAM,GAAG,MAAM,CAmB7D;AAED,wBAAgB,yBAAyB,CACvC,YAAY,EAAE,MAAM,EACpB,SAAS,EAAE,MAAM,EACjB,WAAW,EAAE,MAAM,GAAG,SAAS,UAMhC;AAED,wBAAsB,0BAA0B,CAC9C,UAAU,EAAE,MAAM,EAClB,IAAI,EAAE,MAAM,EACZ,OAAO,GAAE;IAAE,IAAI,CAAC,EAAE,MAAM,CAAC;IAAC,IAAI,CAAC,EAAE,MAAM,GAAG,MAAM,CAAC;IAAC,WAAW,CAAC,EAAE,MAAM,CAAA;CAAO,GAC5E,OAAO,CAAC;IACT,IAAI,EAAE,MAAM,CAAC;IACb,IAAI,CAAC,EAAE,MAAM,GAAG,SAAS,CAAC;IAC1B,IAAI,CAAC,EAAE,MAAM,GAAG,SAAS,CAAC;IAC1B,WAAW,EAAE,MAAM,CAAC;CACrB,CAAC,CAyCD;AAED,wBAAgB,uBAAuB,CAAC,IAAI,EAAE,QAAQ;;;EAwBrD;AAED,wBAAsB,6BAA6B,CACjD,MAAM,EAAE,qBAAqB,GAC5B,OAAO,CAAC,IAAI,CAAC,CAyDf;AAED,wBAAgB,mBAAmB,CACjC,GAAG,EAAE,GAAG,EACR,SAAS,CAAC,EAAE,CAAC,GAAG,EAAE,MAAM,KAAK,OAAO,GACnC,GAAG,CAIL;AAED,wBAAsB,gCAAgC,CACpD,YAAY,EAAE,MAAM,EACpB,YAAY,EAAE,MAAM,EACpB,QAAQ,EAAE,MAAM,iBAuFjB;AAED;;;;;;;;;;GAUG;AACH,wBAAgB,iBAAiB,CAC/B,QAAQ,EAAE,GAAG,EACb,iBAAiB,EAAE,GAAG,CACpB,MAAM,EACN;IAAE,WAAW,EAAE,2BAA2B,EAAE,CAAA;CAAE,EAAE,CACjD,QA+BF;AAGD,wBAAgB,wBAAwB,CACtC,WAAW,EAAE,oBAAoB,EACjC,MAAM,EAAE,CAAC,IAAI,EAAE,cAAc,KAAK,IAAI,QAavC;AAED;;;;;GAKG;AACH,wBAAgB,iCAAiC,CAC/C,mBAAmB,EAAE,MAAM,EAC3B,QAAQ,EAAE,MAAM,EAChB,cAAc,GAAE,OAAe,GAC9B,GAAG,CAAC,MAAM,EAAE;IAAE,WAAW,EAAE,2BAA2B,EAAE,CAAA;CAAE,EAAE,CAAC,CA4C/D;AAED;;;;GAIG;AACH,wBAAsB,oBAAoB,CAAC,QAAQ,EAAE,MAAM,GAAG,OAAO,CAAC,OAAO,CAAC,CAO7E;AAED;;;;;;;;;;GAUG;AACH,wBAAsB,gCAAgC,CACpD,QAAQ,EAAE,MAAM,EAChB,QAAQ,EAAE,MAAM,EAChB,SAAS,EAAE,MAAM,6CAoClB"}
+{"version":3,"file":"util.d.ts","sourceRoot":"","sources":["../../src/reporter/util.ts"],"names":[],"mappings":"AAAA,OAAO,EACL,2BAA2B,EAC3B,qBAAqB,EACtB,MAAM,6CAA6C,CAAC;AACrD,OAAO,EACL,cAAc,EAGd,UAAU,IAAI,oBAAoB,EAElC,QAAQ,EACT,MAAM,2BAA2B,CAAC;AAoBnC,wBAAgB,kBAAkB,CAAC,UAAU,EAAE,MAAM,GAAG,MAAM,CAmB7D;AAED,wBAAgB,yBAAyB,CACvC,YAAY,EAAE,MAAM,EACpB,SAAS,EAAE,MAAM,EACjB,WAAW,EAAE,MAAM,GAAG,SAAS,UAMhC;AAED,wBAAsB,0BAA0B,CAC9C,UAAU,EAAE,MAAM,EAClB,IAAI,EAAE,MAAM,EACZ,OAAO,GAAE;IAAE,IAAI,CAAC,EAAE,MAAM,CAAC;IAAC,IAAI,CAAC,EAAE,MAAM,GAAG,MAAM,CAAC;IAAC,WAAW,CAAC,EAAE,MAAM,CAAA;CAAO,GAC5E,OAAO,CAAC;IACT,IAAI,EAAE,MAAM,CAAC;IACb,IAAI,CAAC,EAAE,MAAM,GAAG,SAAS,CAAC;IAC1B,IAAI,CAAC,EAAE,MAAM,GAAG,SAAS,CAAC;IAC1B,WAAW,EAAE,MAAM,CAAC;CACrB,CAAC,CAyCD;AAED,wBAAgB,uBAAuB,CAAC,IAAI,EAAE,QAAQ;;;EAwBrD;AAED,wBAAsB,6BAA6B,CACjD,MAAM,EAAE,qBAAqB,GAC5B,OAAO,CAAC,IAAI,CAAC,CAyDf;AAED,wBAAgB,mBAAmB,CACjC,GAAG,EAAE,GAAG,EACR,SAAS,CAAC,EAAE,CAAC,GAAG,EAAE,MAAM,KAAK,OAAO,GACnC,GAAG,CAIL;AAED,wBAAsB,gCAAgC,CACpD,YAAY,EAAE,MAAM,EACpB,YAAY,EAAE,MAAM,EACpB,QAAQ,EAAE,MAAM,iBAwGjB;AAED;;;;;;;;;;GAUG;AACH,wBAAgB,iBAAiB,CAC/B,QAAQ,EAAE,GAAG,EACb,iBAAiB,EAAE,GAAG,CACpB,MAAM,EACN;IAAE,WAAW,EAAE,2BAA2B,EAAE,CAAA;CAAE,EAAE,CACjD,QA+BF;AAGD,wBAAgB,wBAAwB,CACtC,WAAW,EAAE,oBAAoB,EACjC,MAAM,EAAE,CAAC,IAAI,EAAE,cAAc,KAAK,IAAI,QAavC;AAED;;;;;GAKG;AACH,wBAAgB,iCAAiC,CAC/C,mBAAmB,EAAE,MAAM,EAC3B,QAAQ,EAAE,MAAM,EAChB,cAAc,GAAE,OAAe,GAC9B,GAAG,CAAC,MAAM,EAAE;IAAE,WAAW,EAAE,2BAA2B,EAAE,CAAA;CAAE,EAAE,CAAC,CA4C/D;AAED;;;;GAIG;AACH,wBAAsB,oBAAoB,CAAC,QAAQ,EAAE,MAAM,GAAG,OAAO,CAAC,OAAO,CAAC,CAO7E;AAED;;;;;;;;;;GAUG;AACH,wBAAsB,gCAAgC,CACpD,QAAQ,EAAE,MAAM,EAChB,QAAQ,EAAE,MAAM,EAChB,SAAS,EAAE,MAAM,6CAoClB"}

package/dist/reporter/util.js

@@ -166,7 +166,12 @@ async function updateHtmlZipFileAttachmentPaths(jsonFilePath, htmlFilePath, repo
         console.error(`❌ Failed to read HTML file at ${htmlFilePath}:`, err);
         return;
     }
-    const base64 = htmlFile.match(/window\.playwrightReportBase64\s*=\s*"(?:data:application\/zip;base64,)?([^"]+)"/)?.[1];
+    // Support both old format (1.53.x) and new format (1.57.0+)
+    // Old: window.playwrightReportBase64 = "data:application/zip;base64,..."
+    // New: <script id="playwrightReportBase64" type="application/zip">data:application/zip;base64,...</script>
+    const oldFormatMatch = htmlFile.match(/window\.playwrightReportBase64\s*=\s*"(?:data:application\/zip;base64,)?([^"]+)"/);
+    const newFormatMatch = htmlFile.match(/<script\s+id="playwrightReportBase64"[^>]*>(?:data:application\/zip;base64,)?([^<]+)<\/script>/);
+    const base64 = oldFormatMatch?.[1] || newFormatMatch?.[1];
     if (!base64) {
         console.error("❌ Base64 zip data not found in HTML.");
         return;
@@ -203,7 +208,15 @@ async function updateHtmlZipFileAttachmentPaths(jsonFilePath, htmlFilePath, repo
         const newZip = new adm_zip_1.default();
         newZip.addLocalFolder(tempDir);
         const newBase64 = newZip.toBuffer().toString("base64");
-        const updatedHtml = htmlFile.replace(/(window\.playwrightReportBase64\s*=\s*")(?:data:application\/zip;base64,)?[^"]*(")/, `$1data:application/zip;base64,${newBase64}$2`);
+        let updatedHtml;
+        if (oldFormatMatch) {
+            // Old format (1.53.x): window.playwrightReportBase64 = "..."
+            updatedHtml = htmlFile.replace(/(window\.playwrightReportBase64\s*=\s*")(?:data:application\/zip;base64,)?[^"]*(")/, `$1data:application/zip;base64,${newBase64}$2`);
+        }
+        else {
+            // New format (1.57.0+): <script id="playwrightReportBase64" ...>...</script>
+            updatedHtml = htmlFile.replace(/(<script\s+id="playwrightReportBase64"[^>]*>)(?:data:application\/zip;base64,)?[^<]*(<\/script>)/, `$1data:application/zip;base64,${newBase64}$2`);
+        }
         await fs_1.default.promises.writeFile(htmlFilePath, updatedHtml, "utf8");
         logger_1.logger.debug("✅ HTML file updated with new base64 zip attachment.");
     }

package/dist/test/index.d.ts

@@ -1,5 +1,6 @@
 import type { BrowserContext, BrowserContextOptions, expect, Page, PlaywrightTestArgs, PlaywrightTestOptions, PlaywrightWorkerArgs, PlaywrightWorkerOptions, TestType } from "@playwright/test";
 import { KVClient } from "../kv";
+import { PostgresClient } from "../postgres";
 import { injectLocatorHighlightScripts } from "./scripts";
 import { HighlighterOpts } from "./types";
 import { setVideoLabel } from "./video-labels";
@@ -14,6 +15,7 @@ declare const extendExpect: (expectInstance: typeof expect) => import("@playwrig
 type TestOptions = {
     page: Page;
     kv: KVClient;
+    postgres: PostgresClient;
     customContextPageProvider: (options?: BrowserContextOptions) => Promise<{
         context: BrowserContext;
         page: Page;

package/dist/test/index.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../src/test/index.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EACV,cAAc,EACd,qBAAqB,EACrB,MAAM,EACN,IAAI,EACJ,kBAAkB,EAClB,qBAAqB,EACrB,oBAAoB,EACpB,uBAAuB,EACvB,QAAQ,EACT,MAAM,kBAAkB,CAAC;AAG1B,OAAO,EAAE,QAAQ,EAAE,MAAM,OAAO,CAAC;AAEjC,OAAO,EAAE,6BAA6B,EAAE,MAAM,WAAW,CAAC;AAC1D,OAAO,EAAE,eAAe,EAAE,MAAM,SAAS,CAAC;AAC1C,OAAO,EAA8B,aAAa,EAAE,MAAM,gBAAgB,CAAC;AAE3E,OAAO,CAAC,MAAM,CAAC;IACb,UAAU,cAAc,CAAC;QACvB,UAAU,QAAQ,CAAC,CAAC;YAClB,WAAW,CAAC,WAAW,EAAE,MAAM,GAAG,OAAO,CAAC,CAAC,CAAC,CAAC;SAC9C;KACF;CACF;AAED,QAAA,MAAM,YAAY,GAAa,gBAAgB,OAAO,MAAM,0CAG3D,CAAC;AAEF,KAAK,WAAW,GAAG;IACjB,IAAI,EAAE,IAAI,CAAC;IACX,EAAE,EAAE,QAAQ,CAAC;IACb,yBAAyB,EAAE,CACzB,OAAO,CAAC,EAAE,qBAAqB,KAC5B,OAAO,CAAC;QAAE,OAAO,EAAE,cAAc,CAAC;QAAC,IAAI,EAAE,IAAI,CAAA;KAAE,CAAC,CAAC;IACtD,UAAU,EAAE,IAAI,CAAC;CAClB,CAAC;AAYF,QAAA,MAAM,eAAe,GACnB,QAAQ,QAAQ,CACd,kBAAkB,GAAG,qBAAqB,EAC1C,oBAAoB,GAAG,uBAAuB,CAC/C,EACD,UAAS,eAAgC,uHAuG1C,CAAC;AAEF,OAAO,EACL,eAAe,EACf,YAAY,EACZ,6BAA6B,EAC7B,aAAa,GACd,CAAC"}
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../src/test/index.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EACV,cAAc,EACd,qBAAqB,EACrB,MAAM,EACN,IAAI,EACJ,kBAAkB,EAClB,qBAAqB,EACrB,oBAAoB,EACpB,uBAAuB,EACvB,QAAQ,EACT,MAAM,kBAAkB,CAAC;AAG1B,OAAO,EAAE,QAAQ,EAAE,MAAM,OAAO,CAAC;AACjC,OAAO,EAAE,cAAc,EAAE,MAAM,aAAa,CAAC;AAE7C,OAAO,EAAE,6BAA6B,EAAE,MAAM,WAAW,CAAC;AAC1D,OAAO,EAAE,eAAe,EAAE,MAAM,SAAS,CAAC;AAC1C,OAAO,EAA8B,aAAa,EAAE,MAAM,gBAAgB,CAAC;AAE3E,OAAO,CAAC,MAAM,CAAC;IACb,UAAU,cAAc,CAAC;QACvB,UAAU,QAAQ,CAAC,CAAC;YAClB,WAAW,CAAC,WAAW,EAAE,MAAM,GAAG,OAAO,CAAC,CAAC,CAAC,CAAC;SAC9C;KACF;CACF;AAED,QAAA,MAAM,YAAY,GAAa,gBAAgB,OAAO,MAAM,0CAG3D,CAAC;AAEF,KAAK,WAAW,GAAG;IACjB,IAAI,EAAE,IAAI,CAAC;IACX,EAAE,EAAE,QAAQ,CAAC;IACb,QAAQ,EAAE,cAAc,CAAC;IACzB,yBAAyB,EAAE,CACzB,OAAO,CAAC,EAAE,qBAAqB,KAC5B,OAAO,CAAC;QAAE,OAAO,EAAE,cAAc,CAAC;QAAC,IAAI,EAAE,IAAI,CAAA;KAAE,CAAC,CAAC;IACtD,UAAU,EAAE,IAAI,CAAC;CAClB,CAAC;AAYF,QAAA,MAAM,eAAe,GACnB,QAAQ,QAAQ,CACd,kBAAkB,GAAG,qBAAqB,EAC1C,oBAAoB,GAAG,uBAAuB,CAC/C,EACD,UAAS,eAAgC,uHA2G1C,CAAC;AAEF,OAAO,EACL,eAAe,EACf,YAAY,EACZ,6BAA6B,EAC7B,aAAa,GACd,CAAC"}

package/dist/test/index.js

@@ -6,6 +6,7 @@ Object.defineProperty(exports, "__esModule", { value: true });
 exports.setVideoLabel = exports.injectLocatorHighlightScripts = exports.extendExpect = exports.baseTestFixture = void 0;
 const path_1 = __importDefault(require("path"));
 const kv_1 = require("../kv");
+const postgres_1 = require("../postgres");
 const expect_1 = require("./expect");
 const scripts_1 = require("./scripts");
 Object.defineProperty(exports, "injectLocatorHighlightScripts", { enumerable: true, get: function () { return scripts_1.injectLocatorHighlightScripts; } });
@@ -53,6 +54,10 @@ const baseTestFixture = function (testFn, options = defaultOptions) {
             const client = new kv_1.KVClient();
             await use(client);
         },
+        postgres: async ({}, use) => {
+            const client = new postgres_1.PostgresClient();
+            await use(client);
+        },
         customContextPageProvider: async ({ browser }, use, testInfo) => {
             const contexts = [];
             async function createContext(contextOptions = {}) {

package/docs/postgres.md

@@ -0,0 +1,78 @@
+# Postgres
+
+A PostgreSQL client for creating and managing temporary Neon databases in tests.
+
+## Fixture usage
+
+```ts
+test("example", async ({ page, postgres }) => {
+  // Get or create a database (cached via KV for 1 hour by default)
+  const { connectionUri } = await postgres.get("my-test-db");
+
+  // Run queries
+  const users = await postgres.query<{ id: number; name: string }>(
+    connectionUri,
+    "SELECT * FROM users",
+  );
+
+  // Execute multiple SQL statements
+  await postgres.execute(
+    connectionUri,
+    `
+    CREATE TABLE users (id SERIAL PRIMARY KEY, name TEXT);
+    INSERT INTO users (name) VALUES ('Alice');
+  `,
+  );
+
+  // Delete when done (optional - TTL will auto-expire)
+  await postgres.delete("my-test-db");
+});
+```
+
+## Standalone usage
+
+```ts
+import { PostgresClient } from "@empiricalrun/playwright-utils";
+
+const postgres = new PostgresClient();
+
+// Get or create a database
+const { projectId, connectionUri } = await postgres.get("my-db", {
+  pgVersion: 15,
+  ttl: 7200, // 2 hours
+});
+
+// Run a query
+const rows = await postgres.query(connectionUri, "SELECT NOW()");
+
+// Delete when done
+await postgres.delete("my-db");
+```
+
+## API
+
+### `postgres.get(name, options?)`
+
+Gets an existing database or creates a new one. Uses KV store to cache the connection.
+
+- `name` - Unique identifier for the database
+- `options.pgVersion` - PostgreSQL version (default: 15)
+- `options.ttl` - Cache TTL in seconds (default: 3600)
+
+Returns `{ projectId, connectionUri }`.
+
+### `postgres.delete(name)`
+
+Deletes the database and removes it from cache.
+
+### `postgres.query(connectionUri, sql)`
+
+Runs a SQL query and returns rows.
+
+### `postgres.execute(connectionUri, sql)`
+
+Executes multiple SQL statements separated by `;`.
+
+## Requirements
+
+- `EMPIRICALRUN_API_KEY` environment variable must be set

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@empiricalrun/playwright-utils",
-  "version": "0.32.0",
+  "version": "0.34.0",
   "publishConfig": {
     "registry": "https://registry.npmjs.org/",
     "access": "public"
@@ -30,21 +30,23 @@
     "@types/node": "^20.14.9",
     "@types/serve-handler": "^6.1.4",
     "@types/adm-zip": "^0.5.7",
+    "@types/pg": "^8.11.6",
     "playwright-core": "1.53.2",
     "serve-handler": "^6.1.6",
     "@empiricalrun/shared-types": "0.12.0"
   },
   "dependencies": {
     "@babel/code-frame": "^7.24.7",
+    "pg": "^8.13.1",
     "adm-zip": "^0.5.16",
     "async-retry": "^1.3.3",
     "authenticator": "^1.1.5",
     "console-log-level": "^1.4.1",
     "puppeteer-extra-plugin-recaptcha": "^3.6.8",
     "rimraf": "^6.0.1",
+    "@empiricalrun/r2-uploader": "^0.5.0",
     "@empiricalrun/llm": "^0.25.1",
-    "@empiricalrun/r2-uploader": "^0.4.0",
-    "@empiricalrun/test-gen": "^0.78.2"
+    "@empiricalrun/test-gen": "^0.78.3"
   },
   "scripts": {
     "dev": "tsc --build --watch",

package/tsconfig.tsbuildinfo

@@ -1 +1 @@
-{"root":["./src/email.ts","./src/index.ts","./src/kv.ts","./src/logger.ts","./src/mailosaur-client.ts","./src/playwright-extensions.ts","./src/auth/google.ts","./src/auth/index.ts","./src/auth/types.ts","./src/captcha/index.ts","./src/captcha/vision.ts","./src/config/index.ts","./src/config/devices/types.ts","./src/overlay-tests/cache.spec.ts","./src/overlay-tests/click.spec.ts","./src/overlay-tests/fixtures.ts","./src/overlay-tests/patch.spec.ts","./src/reporter/empirical-reporter.ts","./src/reporter/uploader.ts","./src/reporter/util.ts","./src/test/constants.ts","./src/test/expect.ts","./src/test/index.ts","./src/test/types.ts","./src/test/video-labels.ts","./src/test/scripts/agent-capabilities.ts","./src/test/scripts/index.ts","./src/test/scripts/locator-highlights.ts","./src/test/scripts/locator-vision.ts","./src/test/scripts/mouse-pointer.ts","./src/test/scripts/types.ts","./src/test/scripts/pw-locator-patch/dismiss-overlays/cache.ts","./src/test/scripts/pw-locator-patch/dismiss-overlays/index.ts","./src/test/scripts/pw-locator-patch/dismiss-overlays/prompt.ts","./src/test/scripts/pw-locator-patch/dismiss-overlays/types.ts","./src/test/scripts/pw-locator-patch/dismiss-overlays/utils.ts","./src/test/scripts/pw-locator-patch/highlight/click.ts","./src/test/scripts/pw-locator-patch/highlight/expect.ts","./src/test/scripts/pw-locator-patch/highlight/hover.ts","./src/test/scripts/pw-locator-patch/highlight/inner-text.ts","./src/test/scripts/pw-locator-patch/highlight/input-value.ts","./src/test/scripts/pw-locator-patch/highlight/is-checked.ts","./src/test/scripts/pw-locator-patch/highlight/is-disabled.ts","./src/test/scripts/pw-locator-patch/highlight/is-editable.ts","./src/test/scripts/pw-locator-patch/highlight/text-content.ts","./src/test/scripts/pw-locator-patch/utils/index.ts","./src/test/scripts/pw-locator-patch/vision/query.ts"],"version":"5.8.3"}
+{"root":["./src/email.ts","./src/index.ts","./src/kv.ts","./src/logger.ts","./src/mailosaur-client.ts","./src/playwright-extensions.ts","./src/postgres.ts","./src/auth/google.ts","./src/auth/index.ts","./src/auth/types.ts","./src/captcha/index.ts","./src/captcha/vision.ts","./src/config/index.ts","./src/config/devices/types.ts","./src/overlay-tests/cache.spec.ts","./src/overlay-tests/click.spec.ts","./src/overlay-tests/fixtures.ts","./src/overlay-tests/patch.spec.ts","./src/reporter/empirical-reporter.ts","./src/reporter/uploader.ts","./src/reporter/util.ts","./src/test/constants.ts","./src/test/expect.ts","./src/test/index.ts","./src/test/types.ts","./src/test/video-labels.ts","./src/test/scripts/agent-capabilities.ts","./src/test/scripts/index.ts","./src/test/scripts/locator-highlights.ts","./src/test/scripts/locator-vision.ts","./src/test/scripts/mouse-pointer.ts","./src/test/scripts/types.ts","./src/test/scripts/pw-locator-patch/dismiss-overlays/cache.ts","./src/test/scripts/pw-locator-patch/dismiss-overlays/index.ts","./src/test/scripts/pw-locator-patch/dismiss-overlays/prompt.ts","./src/test/scripts/pw-locator-patch/dismiss-overlays/types.ts","./src/test/scripts/pw-locator-patch/dismiss-overlays/utils.ts","./src/test/scripts/pw-locator-patch/highlight/click.ts","./src/test/scripts/pw-locator-patch/highlight/expect.ts","./src/test/scripts/pw-locator-patch/highlight/hover.ts","./src/test/scripts/pw-locator-patch/highlight/inner-text.ts","./src/test/scripts/pw-locator-patch/highlight/input-value.ts","./src/test/scripts/pw-locator-patch/highlight/is-checked.ts","./src/test/scripts/pw-locator-patch/highlight/is-disabled.ts","./src/test/scripts/pw-locator-patch/highlight/is-editable.ts","./src/test/scripts/pw-locator-patch/highlight/text-content.ts","./src/test/scripts/pw-locator-patch/utils/index.ts","./src/test/scripts/pw-locator-patch/vision/query.ts"],"version":"5.8.3"}