@dionlarson/playwright-orchestrator-pg 1.3.8 → 1.4.0

package/dist/postgresql-adapter.d.ts

@@ -16,6 +16,7 @@ export declare class PostgreSQLAdapter extends Adapter {
     startShard(runId: string): Promise<TestRunConfig>;
     finishShard(runId: string): Promise<void>;
     dispose(): Promise<void>;
+    cleanupStaleTests(runId: string, staleMinutes: number): Promise<number>;
     getReportData(runId: string): Promise<TestRunReport>;
     private updateTestWithResults;
     private loadTestInfos;

package/dist/postgresql-adapter.js

@@ -26,6 +26,7 @@ export class PostgreSQLAdapter extends Adapter {
     async getNextTest(runId, config) {
         const client = await this.pool.connect();
         try {
+            console.log(`[pg-adapter] getNextTest: BEGIN transaction`);
             await client.query('BEGIN');
             const result = await client.query({
                 text: `WITH next_test AS (
@@ -42,14 +43,17 @@ export class PostgreSQLAdapter extends Adapter {
                 RETURNING *`,
                 values: [runId, TestStatus.Ready, TestStatus.Ongoing],
             });
+            console.log(`[pg-adapter] getNextTest: got ${result.rowCount} rows, order_num=${result.rows[0]?.order_num}`);
             await client.query('COMMIT');
+            console.log(`[pg-adapter] getNextTest: COMMIT`);
             if (result.rowCount === 0)
                 return undefined;
             const { file, line, character, project, timeout, order_num } = result.rows[0];
             return { file, position: `${line}:${character}`, project, timeout, order: order_num };
         }
         catch (e) {
-            await this.pool.query('ROLLBACK');
+            console.log(`[pg-adapter] getNextTest: ERROR ${e}, rolling back`);
+            await client.query('ROLLBACK');
         }
         finally {
             client.release();
@@ -197,6 +201,17 @@ export class PostgreSQLAdapter extends Adapter {
             return;
         await this.pool.end();
     }
+    async cleanupStaleTests(runId, staleMinutes) {
+        const result = await this.pool.query({
+            text: `UPDATE ${this.testsTable}
+                SET status = $1, updated = NOW()
+                WHERE run_id = $2
+                  AND status = $3
+                  AND updated < NOW() - INTERVAL '1 minute' * $4`,
+            values: [TestStatus.Ready, runId, TestStatus.Ongoing, staleMinutes],
+        });
+        return result.rowCount ?? 0;
+    }
     async getReportData(runId) {
         const { rows: [dbConfig], } = await this.pool.query({
             text: `SELECT * FROM ${this.configTable} WHERE id = $1`,
@@ -227,22 +242,28 @@ export class PostgreSQLAdapter extends Adapter {
     async updateTestWithResults(status, { runId, test, config, testResult }) {
         const testId = this.getTestId({ ...test, ...testResult });
         const { rows: [testInfo], } = await this.pool.query({
-            text: `SELECT 
-                id, 
+            text: `SELECT
+                id,
                 ema,
                 (
-                    SELECT COUNT(*) FROM ${this.testInfoHistoryTable} 
+                    SELECT COUNT(*) FROM ${this.testInfoHistoryTable}
                     WHERE status = ${TestStatus.Failed} AND test_info_id = info.id
                 ) AS fails,
                 (
-                    SELECT updated FROM ${this.testInfoHistoryTable} 
-                    WHERE status = ${TestStatus.Passed} AND test_info_id = info.id 
+                    SELECT updated FROM ${this.testInfoHistoryTable}
+                    WHERE status = ${TestStatus.Passed} AND test_info_id = info.id
                     ORDER BY updated DESC LIMIT 1
                 ) AS last_successful_run
             FROM ${this.testInfoTable} info
             WHERE name = $1`,
             values: [testId],
         });
+        // Test not found in database - this shouldn't happen as saveTestRun should create it
+        // But handle gracefully to avoid crashes
+        if (!testInfo) {
+            console.warn(`[pg-adapter] Test not found in database: ${testId}`);
+            return;
+        }
         const report = {
             title: testResult.title,
             status,

package/package.json

@@ -1,6 +1,6 @@
 {
     "name": "@dionlarson/playwright-orchestrator-pg",
-    "version": "1.3.8",
+    "version": "1.4.0",
     "keywords": [],
     "author": "Rostyslav Kudrevatykh",
     "license": "Apache-2.0",