test(testcontainers): register the warmup per collecting suite

Author: nicktrn

internal-packages/testcontainers/src/index.ts

@@ -324,13 +324,7 @@ const withWarmup = <T extends WarmableTestApi>(
   api: T,
   warmUp: (context: any) => Promise<void>
 ): T => {
-  let registered = false;
-
   const register = () => {
-    if (registered) {
-      return;
-    }
-    registered = true;
     api.beforeAll(warmUp, CONTAINER_WARMUP_TIMEOUT_MS);
   };
 

internal-packages/testcontainers/src/warmup.test.ts

@@ -1,8 +1,14 @@
 import { describe, expect, vi } from "vitest";
-import { containerTest } from "./index";
+import { clickhouseTest, containerTest } from "./index";
 
 vi.setConfig({ testTimeout: 10_000 });
 
+describe.skip("a skipped suite that touches the fixture first", () => {
+  containerTest("never runs", async ({ prisma }) => {
+    expect(prisma).toBeDefined();
+  });
+});
+
 describe("container fixture warmup", () => {
   containerTest("the first test is not billed for the container boot", async ({ prisma }) => {
     const rows = await prisma.$queryRawUnsafe<Array<{ ok: number }>>("SELECT 1 as ok");
@@ -16,3 +22,12 @@ describe("container fixture warmup", () => {
     expect(rows[0]?.ok).toBe(2);
   });
 });
+
+describe("worker-scoped fixtures are warmed too", () => {
+  clickhouseTest("clickhouse is up before the first test", async ({ clickhouseClient }) => {
+    const rs = await clickhouseClient.query({ query: "SELECT 1 AS ok", format: "JSONEachRow" });
+    const rows = await rs.json<{ ok: number }>();
+
+    expect(rows[0]?.ok).toBe(1);
+  });
+});