Merge pull request #3846 from cdr/jsjoeio-test-should-enable-proxy

feat: add tests for shouldEnableProxy

test/unit/helpers.test.ts

@@ -1,5 +1,5 @@
 import { promises as fs } from "fs"
-import { tmpdir } from "../../test/utils/helpers"
+import { tmpdir, useEnv } from "../../test/utils/helpers"
 
 /**
  * This file is for testing test helpers (not core code).
@@ -12,3 +12,30 @@ describe("test helpers", () => {
     expect(fs.access(pathToTempDir)).resolves.toStrictEqual(undefined)
   })
 })
+
+describe("useEnv", () => {
+  beforeAll(() => {
+    jest.resetModules()
+    process.env.TEST_USE_ENV = "test environment variable"
+  })
+  afterAll(() => {
+    delete process.env.TEST_USE_ENV
+  })
+  it("should set and reset the env var", () => {
+    const envKey = "TEST_ENV_VAR"
+    const [setValue, resetValue] = useEnv(envKey)
+    setValue("hello-world")
+    expect(process.env[envKey]).toEqual("hello-world")
+    resetValue()
+    expect(process.env[envKey]).toEqual(undefined)
+  })
+  it("should set and reset the env var where a value was already set", () => {
+    const envKey = "TEST_USE_ENV"
+    expect(process.env[envKey]).toEqual("test environment variable")
+    const [setValue, resetValue] = useEnv(envKey)
+    setValue("hello there")
+    expect(process.env[envKey]).toEqual("hello there")
+    resetValue()
+    expect(process.env[envKey]).toEqual("test environment variable")
+  })
+})

test/unit/node/proxy_agent.test.ts

@@ -0,0 +1,39 @@
+import { shouldEnableProxy } from "../../../src/node/proxy_agent"
+import { useEnv } from "../../utils/helpers"
+
+describe("shouldEnableProxy", () => {
+  const [setHTTPProxy, resetHTTPProxy] = useEnv("HTTP_PROXY")
+  const [setHTTPSProxy, resetHTTPSProxy] = useEnv("HTTPS_PROXY")
+  const [setNoProxy, resetNoProxy] = useEnv("NO_PROXY")
+
+  beforeEach(() => {
+    jest.resetModules() // Most important - it clears the cache
+    resetHTTPProxy()
+    resetNoProxy()
+    resetHTTPSProxy()
+  })
+
+  it("returns true when HTTP_PROXY is set", () => {
+    setHTTPProxy("http://proxy.example.com")
+    expect(shouldEnableProxy()).toBe(true)
+  })
+  it("returns true when HTTPS_PROXY is set", () => {
+    setHTTPSProxy("https://proxy.example.com")
+    expect(shouldEnableProxy()).toBe(true)
+  })
+  it("returns false when NO_PROXY is set", () => {
+    setNoProxy("*")
+    expect(shouldEnableProxy()).toBe(false)
+  })
+  it("should return false when neither HTTP_PROXY nor HTTPS_PROXY is set", () => {
+    expect(shouldEnableProxy()).toBe(false)
+  })
+  it("should return false when NO_PROXY is set to https://example.com", () => {
+    setNoProxy("https://example.com")
+    expect(shouldEnableProxy()).toBe(false)
+  })
+  it("should return false when NO_PROXY is set to http://example.com", () => {
+    setNoProxy("http://example.com")
+    expect(shouldEnableProxy()).toBe(false)
+  })
+})