// --------------------------------------------------------------------------- // DbService integration test // --------------------------------------------------------------------------- // // Regression coverage for the pg/CloudflareSocket hang (see // personal-notes/pg-cloudflare-sockets-dev.md). This test: // // - Runs against the same postgres.js code path the Cloudflare Worker uses // - Talks to a real Postgres (PGlite exposed over a TCP socket by // scripts/test-globalsetup.ts) // - Constructs DbService.Live across multiple independent Effect scopes, // the way api.ts does per request // - Performs real queries against the mirrored accounts/organizations // tables through drizzle // // With the old `pg` + `drizzle-orm/node-postgres` stack, the second scope's // query would hang indefinitely because CloudflareSocket cannot be reused // across request contexts and `Client.end()` never resolved. With // postgres.js the test passes: each scope acquires its own socket and // releases it cleanly. import * as Cloudflare from "alchemy/Cloudflare/Workers/Runtime"; import { describe, it, expect } from "@effect/vitest"; import { Effect, Layer } from "effect"; import { DbService } from "./db"; import { makeUserStore } from "./user-store"; const DbLive = DbService.Live.pipe(Layer.provide(Cloudflare.WorkerEnvironment.layer(process.env))); const program = (body: Effect.Effect) => Effect.runPromise(body.pipe(Effect.provide(DbLive), Effect.scoped)); describe("DbService", () => { it("executes a trivial query end-to-end", async () => { const result = await program( Effect.gen(function* () { const { db } = yield* DbService; const rows = yield* Effect.promise(() => db.execute("select 1 as one")); return rows; }), ); expect( Array.isArray(result) ? result[0] : (result as { rows: unknown[] }).rows[0], ).toBeDefined(); }); it("round-trips an account through the user store", async () => { const id = `user_${crypto.randomUUID()}`; const account = await program( Effect.gen(function* () { const { db } = yield* DbService; const store = makeUserStore(db); return yield* Effect.promise(() => store.ensureAccount(id)); }), ); expect(account.id).toBe(id); }); it("supports multiple sequential scopes (regression: pg socket reuse hang)", async () => { // With `pg`, the second scope's connect()/query would hang because the // CloudflareSocket from the first scope cannot be reused and Client.end() // never completes. postgres.js creates a fresh socket per scope. const idA = `user_${crypto.randomUUID()}`; const idB = `user_${crypto.randomUUID()}`; await program( Effect.gen(function* () { const { db } = yield* DbService; yield* Effect.promise(() => makeUserStore(db).ensureAccount(idA)); }), ); const fetched = await program( Effect.gen(function* () { const { db } = yield* DbService; const store = makeUserStore(db); yield* Effect.promise(() => store.ensureAccount(idB)); return yield* Effect.promise(() => store.getAccount(idA)); }), ); expect(fetched?.id).toBe(idA); }, 15_000); it("supports nested scopes within a single outer scope (regression: /api/scope pattern)", async () => { // Mirrors api.ts: an outer scope resolves the org, then an inner scope // (the HttpApi request handler) re-acquires DbService and queries again. const orgId = `org_${crypto.randomUUID()}`; const outer = Layer.provide( Layer.effectDiscard( Effect.gen(function* () { const { db } = yield* DbService; yield* Effect.promise(() => makeUserStore(db).upsertOrganization({ id: orgId, name: "Acme" }), ); }), ), DbLive, ); const result = await Effect.runPromise( Effect.gen(function* () { // Outer "request" scope. yield* Effect.scoped(Layer.build(outer).pipe(Effect.asVoid)); // Inner "handler" scope — fresh DbService. return yield* Effect.scoped( Effect.gen(function* () { const { db } = yield* DbService; return yield* Effect.promise(() => makeUserStore(db).getOrganization(orgId)); }).pipe(Effect.provide(DbLive)), ); }), ); expect(result?.id).toBe(orgId); expect(result?.name).toBe("Acme"); }, 15_000); });