test: raw-query golden harness for @prisma/adapter-pg (TRI-13039 spike)

ericallam · ericallam · commit 707dff7e4e34 · 2026-08-07T23:20:49.000+01:00
Spike evidence only, not for merge. Adds a testcontainers harness that runs the same
raw queries through the current Rust-engine Prisma client and an @prisma/adapter-pg
client against the same Postgres, plus a tracing gate.

- Deliverable A: engine spans (prisma:client:operation, prisma:engine:db_query,
  prisma:engine:connection) survive the adapter under driverAdapters-alone.
- Deliverable B: 10/11 raw-result shapes byte-identical; only unqualified SQL on a
  non-public schema diverges (#28128), inert for the public-schema prod DBs.

Enables the driverAdapters preview flag on both schemas (regen required, not committed).
diff --git a/internal-packages/database/prisma/schema.prisma b/internal-packages/database/prisma/schema.prisma
@@ -8,7 +8,7 @@ generator client {
   provider        = "prisma-client-js"
   output          = "../generated/prisma"
   binaryTargets   = ["native", "debian-openssl-1.1.x"]
-  previewFeatures = ["metrics"]
+  previewFeatures = ["metrics", "driverAdapters"]
 }
 
 model User {
diff --git a/internal-packages/run-ops-database/prisma/schema.prisma b/internal-packages/run-ops-database/prisma/schema.prisma
@@ -7,7 +7,7 @@ generator client {
   provider        = "prisma-client-js"
   output          = "../generated/run-ops"
   binaryTargets   = ["native", "debian-openssl-1.1.x"]
-  previewFeatures = ["metrics"]
+  previewFeatures = ["metrics", "driverAdapters"]
 }
 
 // ─────────────────────────────────────────────────────────────────────────────
diff --git a/internal-packages/testcontainers/package.json b/internal-packages/testcontainers/package.json
@@ -20,9 +20,16 @@
   },
   "devDependencies": {
     "@internal/run-ops-database": "workspace:*",
+    "@opentelemetry/instrumentation": "0.218.0",
+    "@opentelemetry/sdk-trace-base": "2.7.1",
+    "@opentelemetry/sdk-trace-node": "2.7.1",
+    "@prisma/adapter-pg": "6.14.0",
+    "@prisma/instrumentation": "6.14.0",
     "@testcontainers/postgresql": "^11.14.0",
+    "@types/pg": "^8.11.10",
     "@testcontainers/redis": "^11.14.0",
     "@trigger.dev/core": "workspace:*",
+    "pg": "8.15.6",
     "std-env": "^3.9.0",
     "testcontainers": "^11.14.0",
     "tinyexec": "^0.3.0"
diff --git a/internal-packages/testcontainers/src/adapterGolden.test.ts b/internal-packages/testcontainers/src/adapterGolden.test.ts
@@ -0,0 +1,255 @@
+import { writeFileSync } from "node:fs";
+import { Prisma } from "@trigger.dev/database";
+import { expect } from "vitest";
+import { postgresTest } from "./index";
+import {
+  createAdapterClient,
+  createClientPair,
+  createRustClient,
+  describe,
+  runShape,
+  type ShapeCase,
+  type ShapeResult,
+} from "./adapterGolden";
+
+function safeWrite(path: string, contents: string) {
+  try {
+    writeFileSync(path, contents);
+  } catch {
+    void 0;
+  }
+}
+
+const shapes: ShapeCase[] = [
+  {
+    id: 1,
+    name: "text[] param with explicit cast (= ANY(${ids}::text[]))",
+    callSite: "PostgresRunStore.ts L2159-2161 (documented binding workaround)",
+    upstream: "#24338",
+    setup: async (c) => {
+      await c.$executeRawUnsafe(`CREATE TABLE g1 (id text)`);
+      await c.$executeRawUnsafe(`INSERT INTO g1 (id) VALUES ('a'),('b'),('c')`);
+    },
+    run: (c) => {
+      const ids = ["a", "c"];
+      return c.$queryRaw`SELECT id FROM g1 WHERE id = ANY(${ids}::text[]) ORDER BY id`;
+    },
+  },
+  {
+    id: 2,
+    name: "array results (array_agg + text[] column)",
+    callSite: "text[] columns / array_agg readers",
+    upstream: "#27823",
+    setup: async (c) => {
+      await c.$executeRawUnsafe(`CREATE TABLE g2 (id text, tags text[])`);
+      await c.$executeRawUnsafe(`INSERT INTO g2 (id, tags) VALUES ('x', ARRAY['a','b'])`);
+    },
+    run: (c) => c.$queryRaw`SELECT tags, array_agg(id ORDER BY id) AS ids FROM g2 GROUP BY tags`,
+  },
+  {
+    id: 3,
+    name: "bigint from COUNT(*)",
+    callSite: "PostgresRunStore.ts:2162,2171; DeploymentListPresenter.server.ts:322,168",
+    upstream: "#23926",
+    setup: async (c) => {
+      await c.$executeRawUnsafe(`CREATE TABLE g3 (id int)`);
+      await c.$executeRawUnsafe(`INSERT INTO g3 (id) VALUES (1),(2),(3)`);
+    },
+    run: (c) => c.$queryRaw`SELECT COUNT(*) AS c FROM g3`,
+  },
+  {
+    id: 4,
+    name: "bigint ns timestamp (keyset pagination)",
+    callSite: "taskEventStore.server.ts L339-386",
+    run: (c) => c.$queryRaw`SELECT 1723058400000000000::int8 AS t`,
+  },
+  {
+    id: 5,
+    name: "NUMERIC -> Prisma.Decimal",
+    callSite: "QueueListPresenter.server.ts:383",
+    setup: async (c) => {
+      await c.$executeRawUnsafe(`CREATE TABLE g5 (n numeric(30,4))`);
+      await c.$executeRawUnsafe(`INSERT INTO g5 (n) VALUES ('12345.6789')`);
+    },
+    run: (c) => c.$queryRaw`SELECT n FROM g5`,
+  },
+  {
+    id: 6,
+    name: "enum cast (CAST('LOG'::text AS enum))",
+    callSite: "taskEventStore.server.ts:196,224,274,303,327",
+    setup: async (c) => {
+      await c.$executeRawUnsafe(`CREATE TYPE g6_kind AS ENUM ('LOG','SPAN')`);
+    },
+    run: (c) => c.$queryRaw`SELECT CAST('LOG'::text AS g6_kind) AS kind`,
+  },
+  {
+    id: 7,
+    name: "jsonb with cast parameters (to_jsonb(${v}::text/::int))",
+    callSite: "dashboardPreferences.server.ts:145,175",
+    upstream: "#24338",
+    run: (c) => {
+      const theme = "dark";
+      const contrast = 1;
+      return c.$queryRaw`
+        SELECT jsonb_set(
+                 jsonb_set('{}'::jsonb, '{theme}', to_jsonb(${theme}::text)),
+                 '{contrast}', to_jsonb(${contrast}::int)
+               ) AS prefs`;
+    },
+  },
+  {
+    id: 8,
+    name: "timestamptz round-trip (Date param)",
+    callSite: "Date -> timestamptz writers",
+    upstream: "#28629",
+    run: (c) => {
+      const d = new Date("2026-08-07T12:34:56.789Z");
+      return c.$queryRaw`SELECT ${d}::timestamptz AS ts`;
+    },
+  },
+  {
+    id: 9,
+    name: "IN with varying arity",
+    callSite: "boundedIn() (TRI-4480)",
+    upstream: "#21803",
+    setup: async (c) => {
+      await c.$executeRawUnsafe(`CREATE TABLE g9 (id text)`);
+      await c.$executeRawUnsafe(`INSERT INTO g9 (id) VALUES ('a'),('b'),('c'),('d')`);
+    },
+    run: async (c) => {
+      const one = await c.$queryRaw`SELECT id FROM g9 WHERE id IN (${"a"}) ORDER BY id`;
+      const three =
+        await c.$queryRaw`SELECT id FROM g9 WHERE id IN (${"a"},${"c"},${"d"}) ORDER BY id`;
+      return { arity1: one, arity3: three };
+    },
+  },
+  {
+    id: 11,
+    name: "nulls, empty result set, zero-row RETURNING",
+    callSite: "classic divergence points",
+    setup: async (c) => {
+      await c.$executeRawUnsafe(`CREATE TABLE g11 (id text)`);
+      await c.$executeRawUnsafe(`INSERT INTO g11 (id) VALUES ('a')`);
+    },
+    run: async (c) => {
+      const nulls = await c.$queryRaw`SELECT NULL::text AS a, NULL::int AS b`;
+      const empty = await c.$queryRaw`SELECT id FROM g11 WHERE 1=0`;
+      const returning = await c.$queryRaw`UPDATE g11 SET id = id WHERE 1=0 RETURNING id`;
+      return { nulls, empty, returning };
+    },
+  },
+];
+
+function renderTable(results: ShapeResult[]): string {
+  const rows = results.map((r) => {
+    const status = r.identical ? "IDENTICAL" : r.adapterErrored ? "ADAPTER ERROR" : "DIVERGES";
+    return [
+      `### Shape ${r.id}: ${r.name}`,
+      `- call site: ${r.callSite}${r.upstream ? ` (upstream ${r.upstream})` : ""}`,
+      `- result: **${status}**`,
+      `- rust:    \`${r.rust}\``,
+      `- adapter: \`${r.adapter}\``,
+      "",
+    ].join("\n");
+  });
+  const pass = results.filter((r) => r.identical).length;
+  return [`## Golden matrix: ${pass}/${results.length} identical`, "", ...rows].join("\n");
+}
+
+postgresTest(
+  "Deliverable B — raw-result equivalence matrix",
+  async ({ postgresContainer }) => {
+    const pair = await createClientPair(postgresContainer.getConnectionUri());
+    const results: ShapeResult[] = [];
+
+    try {
+      for (const shape of shapes) {
+        const result = await runShape(pair, shape);
+        results.push(result);
+        const tag = result.identical ? "OK  " : result.adapterErrored ? "ERR " : "DIFF";
+        console.log(`[TRI-13039][B] ${tag} shape ${result.id}: ${result.name}`);
+        if (!result.identical) {
+          console.log(`    rust:    ${result.rust}`);
+          console.log(`    adapter: ${result.adapter}`);
+        }
+      }
+    } finally {
+      await pair.disconnect();
+    }
+
+    const md = renderTable(results);
+    console.log("\n" + md);
+    safeWrite("/Users/eric/code/triggerdotdev/isolated/adapter-pg-spike-work/golden-matrix.md", md);
+
+    for (const r of results) {
+      expect(r.rust, `shape ${r.id} (${r.name}) must be byte-identical`).toBe(r.adapter);
+    }
+  },
+  180000
+);
+
+postgresTest(
+  "Deliverable B shape 10 — unqualified SQL relying on search_path / {schema} (#28128)",
+  async ({ postgresContainer }) => {
+    const baseUri = postgresContainer.getConnectionUri();
+
+    const admin = createRustClient(baseUri);
+    await admin.$executeRawUnsafe(`CREATE SCHEMA s1`);
+    await admin.$executeRawUnsafe(`CREATE TABLE s1.t (id text)`);
+    await admin.$executeRawUnsafe(`INSERT INTO s1.t (id) VALUES ('inschema')`);
+    await admin.$executeRawUnsafe(`CREATE TABLE public.pt (id text)`);
+    await admin.$executeRawUnsafe(`INSERT INTO public.pt (id) VALUES ('inpublic')`);
+    await admin.$disconnect();
+
+    const capture = async (fn: () => Promise<unknown>) => {
+      try {
+        return { ok: true as const, desc: describe(await fn()) };
+      } catch (err: any) {
+        return {
+          ok: false as const,
+          desc: `${err?.code}: ${String(err?.message).split("\n").pop()}`,
+        };
+      }
+    };
+
+    const nonPublicAdapter = createAdapterClient(baseUri, "s1");
+    const publicAdapter = createAdapterClient(baseUri, "public");
+    try {
+      const nonPublic = await capture(() =>
+        nonPublicAdapter.$queryRaw(Prisma.sql([`SELECT id FROM t ORDER BY id`]))
+      );
+      const publicSchema = await capture(() =>
+        publicAdapter.$queryRaw(Prisma.sql([`SELECT id FROM pt ORDER BY id`]))
+      );
+
+      console.log(`[TRI-13039][B] shape 10 adapter {schema:s1} unqualified: ${nonPublic.desc}`);
+      console.log(
+        `[TRI-13039][B] shape 10 adapter {schema:public} unqualified: ${publicSchema.desc}`
+      );
+      safeWrite(
+        "/Users/eric/code/triggerdotdev/isolated/adapter-pg-spike-work/golden-shape10.md",
+        [
+          "## Shape 10: unqualified SQL relying on search_path (#28128)",
+          "",
+          "The adapter's `{schema}` option does NOT set a session search_path for raw SQL.",
+          `- {schema:s1} + unqualified \`t\` (non-public): ${nonPublic.ok ? "OK" : "FAILS"} -> \`${nonPublic.desc}\``,
+          `- {schema:public} + unqualified \`pt\` (public): ${publicSchema.ok ? "OK" : "FAILS"} -> \`${publicSchema.desc}\``,
+          "",
+          "Prod impact: both DATABASE_URL and RUN_OPS_DATABASE_URL use ?schema=public, so unqualified",
+          "raw SQL resolves fine under the adapter. Only a non-public schema would break.",
+        ].join("\n")
+      );
+
+      expect(nonPublic.ok, "non-public schema breaks unqualified raw SQL (confirms #28128)").toBe(
+        false
+      );
+      expect(publicSchema.ok, "public schema resolves unqualified raw SQL under the adapter").toBe(
+        true
+      );
+      expect(publicSchema.desc).toBe(`[{id: "inpublic"}]`);
+    } finally {
+      await Promise.allSettled([nonPublicAdapter.$disconnect(), publicAdapter.$disconnect()]);
+    }
+  },
+  180000
+);
diff --git a/internal-packages/testcontainers/src/adapterGolden.ts b/internal-packages/testcontainers/src/adapterGolden.ts
@@ -0,0 +1,99 @@
+import { PrismaClient, Prisma } from "@trigger.dev/database";
+import { PrismaPg } from "@prisma/adapter-pg";
+
+export type ClientPair = {
+  rust: PrismaClient;
+  adapter: PrismaClient;
+  disconnect: () => Promise<void>;
+};
+
+export function createRustClient(url: string): PrismaClient {
+  return new PrismaClient({ datasources: { db: { url } } });
+}
+
+export function createAdapterClient(url: string, schema?: string): PrismaClient {
+  const adapter = new PrismaPg({ connectionString: url }, schema ? { schema } : undefined);
+  return new PrismaClient({ adapter } as unknown as ConstructorParameters<typeof PrismaClient>[0]);
+}
+
+export async function createClientPair(url: string): Promise<ClientPair> {
+  const rust = createRustClient(url);
+  const adapter = createAdapterClient(url);
+  return {
+    rust,
+    adapter,
+    disconnect: async () => {
+      await Promise.allSettled([rust.$disconnect(), adapter.$disconnect()]);
+    },
+  };
+}
+
+export function describe(value: unknown): string {
+  if (value === null) return "null";
+  if (value === undefined) return "undefined";
+  if (typeof value === "bigint") return `bigint(${value.toString()})`;
+  if (Prisma.Decimal.isDecimal(value)) return `Decimal(${(value as Prisma.Decimal).toString()})`;
+  if (value instanceof Date) return `Date(${value.toISOString()})`;
+  if (Buffer.isBuffer(value)) return `Buffer(${value.toString("hex")})`;
+  if (Array.isArray(value)) return `[${value.map(describe).join(", ")}]`;
+  if (typeof value === "object") {
+    const entries = Object.entries(value as Record<string, unknown>)
+      .map(([k, v]) => `${k}: ${describe(v)}`)
+      .join(", ");
+    return `{${entries}}`;
+  }
+  if (typeof value === "string") return `"${value}"`;
+  return `${typeof value}(${String(value)})`;
+}
+
+export function deepEqual(a: unknown, b: unknown): boolean {
+  return describe(a) === describe(b);
+}
+
+export type ShapeCase = {
+  id: number;
+  name: string;
+  callSite: string;
+  upstream?: string;
+  setup?: (rust: PrismaClient) => Promise<void>;
+  run: (client: PrismaClient) => Promise<unknown>;
+};
+
+export type ShapeResult = {
+  id: number;
+  name: string;
+  callSite: string;
+  upstream?: string;
+  rust: string;
+  adapter: string;
+  identical: boolean;
+  adapterErrored: boolean;
+};
+
+async function capture(fn: () => Promise<unknown>): Promise<{ desc: string; errored: boolean }> {
+  try {
+    const value = await fn();
+    return { desc: describe(value), errored: false };
+  } catch (err) {
+    const message = err instanceof Error ? err.message : String(err);
+    return { desc: `ERROR: ${message.split("\n")[0]}`, errored: true };
+  }
+}
+
+export async function runShape(pair: ClientPair, shape: ShapeCase): Promise<ShapeResult> {
+  if (shape.setup) {
+    await shape.setup(pair.rust);
+  }
+  const rust = await capture(() => shape.run(pair.rust));
+  const adapter = await capture(() => shape.run(pair.adapter));
+  return {
+    id: shape.id,
+    name: shape.name,
+    callSite: shape.callSite,
+    upstream: shape.upstream,
+    rust: rust.desc,
+    adapter: adapter.desc,
+    identical: rust.desc === adapter.desc,
+    adapterErrored: adapter.errored,
+  };
+}
diff --git a/internal-packages/testcontainers/src/adapterTracing.test.ts b/internal-packages/testcontainers/src/adapterTracing.test.ts
diff --git a/pnpm-lock.yaml b/pnpm-lock.yaml

Original file line number	Diff line number	Diff line change
`@@ -8,7 +8,7 @@ generator client {`
`8`	`8`	`provider = "prisma-client-js"`
`9`	`9`	`output = "../generated/prisma"`
`10`	`10`	`binaryTargets = ["native", "debian-openssl-1.1.x"]`
`11`		`- previewFeatures = ["metrics"]`
	`11`	`+ previewFeatures = ["metrics", "driverAdapters"]`
`12`	`12`	`}`
`13`	`13`
`14`	`14`	`model User {`
Original file line number	Diff line number	Diff line change
`@@ -7,7 +7,7 @@ generator client {`
`7`	`7`	`provider = "prisma-client-js"`
`8`	`8`	`output = "../generated/run-ops"`
`9`	`9`	`binaryTargets = ["native", "debian-openssl-1.1.x"]`
`10`		`- previewFeatures = ["metrics"]`
	`10`	`+ previewFeatures = ["metrics", "driverAdapters"]`
`11`	`11`	`}`
`12`	`12`
`13`	`13`	`// ─────────────────────────────────────────────────────────────────────────────`