Create file and function to AddTenantToUrl

packages/backend-core/tests/tenancy/tenancy.spec.ts

@@ -0,0 +1,50 @@
+import {
+  DEFAULT_TENANT_ID,
+  getTenantId,
+  getTenantIDFromAppID,
+  isMultiTenant,
+} from "../../src/context"
+import env from "../../src/environment"
+import {
+  BBContext,
+  TenantResolutionStrategy,
+  GetTenantIdOptions,
+} from "@budibase/types"
+import { Header } from "../../src/constants"
+import {
+  addTenantToUrl,
+  isUserInAppTenant,
+  getTenantIDFromCtx,
+} from "../../src/tenancy"
+
+// mock the `getTenantId` and `isMultiTenant` functions
+jest.mock("../../src/context", () => ({
+  getTenantId: jest.fn(() => "budibase"),
+  isMultiTenant: jest.fn(() => true),
+}))
+
+describe("addTenantToUrl", () => {
+  it("should append tenantId parameter to the URL", () => {
+    const url = "https://example.com"
+    const expectedUrl = "https://example.com?tenantId=budibase"
+    expect(addTenantToUrl(url)).toEqual(expectedUrl)
+  })
+
+  it("should append tenantId parameter to the URL query string", () => {
+    const url = "https://example.com?foo=bar"
+    const expectedUrl = "https://example.com?foo=bar&tenantId=default"
+    expect(addTenantToUrl(url)).toEqual(expectedUrl)
+  })
+
+  it("should not append tenantId parameter to the URL if isMultiTenant is false", () => {
+    // mock the `isMultiTenant` function to return false
+    jest.mock("../../src/context", () => ({
+      getTenantId: jest.fn(() => "default"),
+      isMultiTenant: jest.fn(() => false),
+    }))
+
+    const url = "https://example.com"
+    const expectedUrl = "https://example.com"
+    expect(addTenantToUrl(url)).toEqual(expectedUrl)
+  })
+})