Add unit test for getURLs

Emyrk · Emyrk · commit 3391e84cd509 · 2023-04-27T11:17:03.000-05:00
diff --git a/site/src/contexts/ProxyContext.test.ts b/site/src/contexts/ProxyContext.test.ts
@@ -0,0 +1,20 @@
+import { MockPrimaryRegion, MockRegions, MockHealthyWildRegion } from "testHelpers/entities"
+import { getURLs } from "./ProxyContext"
+
+describe("ProxyContextGetURLs", () => {
+  it.each([
+    ["empty", [], undefined, "", ""],
+    // Primary has no path app URL. Uses relative links
+    ["primary", [MockPrimaryRegion], MockPrimaryRegion, "", MockPrimaryRegion.wildcard_hostname],
+    ["regions selected", MockRegions, MockHealthyWildRegion, MockHealthyWildRegion.path_app_url, MockHealthyWildRegion.wildcard_hostname],
+    // Primary is the default if none selected
+    ["no selected", [MockPrimaryRegion], undefined, "", MockPrimaryRegion.wildcard_hostname],
+    ["regions no select primary default", MockRegions, undefined, "", MockPrimaryRegion.wildcard_hostname],
+    // This should never happen, when there is no primary
+    ["no primary", [MockHealthyWildRegion], undefined, "", ""],
+  ])(`%p`, (_, regions, selected, preferredPathAppURL, preferredWildcardHostname) => {
+      const preferred = getURLs(regions, selected)
+      expect(preferred.preferredPathAppURL).toBe(preferredPathAppURL)
+      expect(preferred.preferredWildcardHostname).toBe(preferredWildcardHostname)
+  })
+})
diff --git a/site/src/contexts/ProxyContext.tsx b/site/src/contexts/ProxyContext.tsx
@@ -143,11 +143,12 @@ export const useProxy = (): ProxyContextValue => {
 /**
  * getURLs is a helper function to calculate the urls to use for a given proxy configuration. By default, it is
  * assumed no proxy is configured and relative paths should be used.
+ * Exported for testing.
  *
  * @param regions Is the list of regions returned by coderd. If this is empty, default behavior is used.
  * @param selectedRegion Is the region the user has selected. If this is undefined, default behavior is used.
  */
-const getURLs = (
+export const getURLs = (
   regions: Region[],
   selectedRegion?: Region,
 ): PreferredProxy => {
diff --git a/site/src/testHelpers/entities.ts b/site/src/testHelpers/entities.ts
@@ -78,8 +78,19 @@ export const MockPrimaryRegion: TypesGen.Region = {
   wildcard_hostname: "*.coder.com",
 }
 
+export const MockHealthyWildRegion: TypesGen.Region = {
+  id: "5e2c1ab7-479b-41a9-92ce-aa85625de52c",
+  name: "haswildcard",
+  display_name: "Subdomain Supported",
+  icon_url: "/emojis/1f319.png",
+  healthy: true,
+  path_app_url: "https://external.com",
+  wildcard_hostname: "*.external.com",
+} 
+
 export const MockRegions: TypesGen.Region[] = [
   MockPrimaryRegion,
+  MockHealthyWildRegion,
   {
     id: "8444931c-0247-4171-842a-569d9f9cbadb",
     name: "unhealthy",
