Reactor setting env vars in tests

Author: cristianrgreco

packages/testcontainers/src/container-runtime/auth/get-auth-config.test.ts

@@ -14,20 +14,23 @@ describe("get auth config", () => {
   });
 
   afterEach(() => {
-    delete process.env.DOCKER_AUTH_CONFIG;
+    vi.unstubAllEnvs();
     vi.resetModules();
   });
 
   it("should use DOCKER_AUTH_CONFIG environment variable as Docker config", async () => {
-    process.env.DOCKER_AUTH_CONFIG = JSON.stringify({
-      auths: {
-        "https://registry.example.com": {
-          email: "[email protected]",
-          username: "user",
-          password: "pass",
+    vi.stubEnv(
+      "DOCKER_AUTH_CONFIG",
+      JSON.stringify({
+        auths: {
+          "https://registry.example.com": {
+            email: "[email protected]",
+            username: "user",
+            password: "pass",
+          },
         },
-      },
-    });
+      })
+    );
     const { getAuthConfig } = await import("./get-auth-config");
     expect(await getAuthConfig("https://registry.example.com")).toEqual({
       username: "user",

packages/testcontainers/src/container-runtime/image-name.test.ts

@@ -10,6 +10,10 @@ describe("ContainerImage", () => {
     expect(imageName.equals(new ImageName("anotherRegistry", "image", "tag"))).toBe(false);
   });
 
+  afterEach(() => {
+    vi.unstubAllEnvs();
+  });
+
   describe("string", () => {
     it("should work with registry", () => {
       const imageName = new ImageName("registry", "image", "tag");
@@ -57,21 +61,12 @@ describe("ContainerImage", () => {
     it.each(["custom.com/registry", "custom.com/registry/"])(
       "should substitute no registry with the one provided via TESTCONTAINERS_HUB_IMAGE_NAME_PREFIX when provided registry is %s",
       (customRegistry: string) => {
-        const oldEnvValue = process.env.TESTCONTAINERS_HUB_IMAGE_NAME_PREFIX;
-        try {
-          process.env.TESTCONTAINERS_HUB_IMAGE_NAME_PREFIX = customRegistry;
-          const imageName = new ImageName(undefined, "image", "tag");
-          expect(imageName.registry).toBe("custom.com");
-          expect(imageName.image).toBe("registry/image");
-          expect(imageName.tag).toBe("tag");
-          expect(imageName.string).toBe("custom.com/registry/image:tag");
-        } finally {
-          if (oldEnvValue === undefined) {
-            delete process.env.TESTCONTAINERS_HUB_IMAGE_NAME_PREFIX;
-          } else {
-            process.env.TESTCONTAINERS_HUB_IMAGE_NAME_PREFIX = oldEnvValue;
-          }
-        }
+        vi.stubEnv("TESTCONTAINERS_HUB_IMAGE_NAME_PREFIX", customRegistry);
+        const imageName = new ImageName(undefined, "image", "tag");
+        expect(imageName.registry).toBe("custom.com");
+        expect(imageName.image).toBe("registry/image");
+        expect(imageName.tag).toBe("tag");
+        expect(imageName.string).toBe("custom.com/registry/image:tag");
       }
     );
   });
@@ -176,18 +171,8 @@ describe("ContainerImage", () => {
   ])(
     "fromString with TESTCONTAINERS_HUB_IMAGE_NAME_PREFIX set to $customRegistry",
     ({ customRegistry, expectedRegistry, expectedImagePrefix }) => {
-      let oldEnvValue: string | undefined;
       beforeEach(() => {
-        oldEnvValue = process.env.TESTCONTAINERS_HUB_IMAGE_NAME_PREFIX;
-        process.env.TESTCONTAINERS_HUB_IMAGE_NAME_PREFIX = customRegistry;
-      });
-
-      afterEach(() => {
-        if (oldEnvValue === undefined) {
-          delete process.env.TESTCONTAINERS_HUB_IMAGE_NAME_PREFIX;
-        } else {
-          process.env.TESTCONTAINERS_HUB_IMAGE_NAME_PREFIX = oldEnvValue;
-        }
+        vi.stubEnv("TESTCONTAINERS_HUB_IMAGE_NAME_PREFIX", customRegistry);
       });
 
       it("should work", () => {

packages/testcontainers/src/generic-container/generic-container-reuse.test.ts

@@ -4,7 +4,7 @@ import { GenericContainer } from "./generic-container";
 
 describe("GenericContainer reuse", { timeout: 180_000 }, () => {
   afterEach(() => {
-    process.env.TESTCONTAINERS_REUSE_ENABLE = undefined;
+    vi.unstubAllEnvs();
   });
 
   it("should not reuse the container by default", async () => {
@@ -64,7 +64,7 @@ describe("GenericContainer reuse", { timeout: 180_000 }, () => {
   });
 
   it("should not reuse the container when TESTCONTAINERS_REUSE_ENABLE is set to false", async () => {
-    process.env.TESTCONTAINERS_REUSE_ENABLE = "false";
+    vi.stubEnv("TESTCONTAINERS_REUSE_ENABLE", "false");
 
     const name = `there_can_only_be_one_${randomUuid()}`;
     const container = await new GenericContainer("cristianrgreco/testcontainer:1.1.14")
@@ -90,7 +90,7 @@ describe("GenericContainer reuse", { timeout: 180_000 }, () => {
   it.each(["true", undefined])(
     "should reuse the container when TESTCONTAINERS_REUSE_ENABLE is set to %s",
     async (reuseEnable: string | undefined) => {
-      process.env.TESTCONTAINERS_REUSE_ENABLE = reuseEnable;
+      vi.stubEnv("TESTCONTAINERS_REUSE_ENABLE", reuseEnable);
 
       const name = `there_can_only_be_one_${randomUuid()}`;
       const container1 = await new GenericContainer("cristianrgreco/testcontainer:1.1.14")

packages/testcontainers/src/reaper/reaper.test.ts

@@ -8,13 +8,13 @@ describe("Reaper", { timeout: 120_000 }, () => {
 
   beforeEach(async () => {
     vi.resetModules();
-    vitest.unstubAllEnvs();
+    vi.unstubAllEnvs();
 
     client = await getContainerRuntimeClient();
   });
 
   it("should create disabled reaper when TESTCONTAINERS_RYUK_DISABLED=true", async () => {
-    vitest.stubEnv("TESTCONTAINERS_RYUK_DISABLED", "true");
+    vi.stubEnv("TESTCONTAINERS_RYUK_DISABLED", "true");
 
     vi.spyOn(client.container, "list").mockResolvedValue([]);
     const reaper = await getReaper();
@@ -29,7 +29,7 @@ describe("Reaper", { timeout: 120_000 }, () => {
 
   it("should use custom port when TESTCONTAINERS_RYUK_PORT is set", async () => {
     const customPort = (await new RandomUniquePortGenerator().generatePort()).toString();
-    vitest.stubEnv("TESTCONTAINERS_RYUK_PORT", customPort);
+    vi.stubEnv("TESTCONTAINERS_RYUK_PORT", customPort);
 
     vi.spyOn(client.container, "list").mockResolvedValue([]);
     const reaper = await getReaper();
@@ -60,7 +60,7 @@ describe("Reaper", { timeout: 120_000 }, () => {
   });
 
   it("should propagate TESTCONTAINERS_RYUK_VERBOSE into Reaper container", async () => {
-    vitest.stubEnv("TESTCONTAINERS_RYUK_VERBOSE", "true");
+    vi.stubEnv("TESTCONTAINERS_RYUK_VERBOSE", "true");
 
     vi.spyOn(client.container, "list").mockResolvedValue([]);
     const reaper = await getReaper();