// ---------------------------------------------------------------------------
// DbService integration test
// ---------------------------------------------------------------------------
//
// Regression coverage for the pg/CloudflareSocket hang (see
// personal-notes/pg-cloudflare-sockets-dev.md). This test:
//
// - Runs inside the Cloudflare Workers runtime via
// @cloudflare/vitest-pool-workers
// - 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 { describe, it, expect } from "@effect/vitest";
import { Effect, Layer } from "effect";
import { DbService } from "./db";
import { makeUserStore } from "./user-store";
const program = (body: Effect.Effect) =>
Effect.runPromise(
body.pipe(Effect.provide(DbService.Live), Effect.scoped) as Effect.Effect,
);
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" }),
);
}),
),
DbService.Live,
);
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(DbService.Live)),
) as Effect.Effect<{ id: string; name: string } | null, never, never>;
}) as Effect.Effect<{ id: string; name: string } | null, never, never>,
);
expect(result?.id).toBe(orgId);
expect(result?.name).toBe("Acme");
}, 15_000);
});