wehub-resource-sync
101 lines
3.1 KiB
TypeScript
101 lines
3.1 KiB
TypeScript
import { describe, it, expect } from "vitest";
|
|
import { tenantContext, type TenantContext } from "../app/services/tenantContext.server";
|
|
|
|
const sample: TenantContext = {
|
|
orgSlug: "acme",
|
|
projectSlug: "web",
|
|
envSlug: "prod",
|
|
};
|
|
|
|
describe("tenantContext", () => {
|
|
it("returns undefined outside run()", () => {
|
|
expect(tenantContext.get()).toBeUndefined();
|
|
});
|
|
|
|
it("returns the active context inside run()", () => {
|
|
tenantContext.run(sample, () => {
|
|
expect(tenantContext.get()).toEqual(sample);
|
|
});
|
|
});
|
|
|
|
it("isolates concurrent async trees", async () => {
|
|
const a: TenantContext = { ...sample, orgSlug: "a" };
|
|
const b: TenantContext = { ...sample, orgSlug: "b" };
|
|
|
|
const [got1, got2] = await Promise.all([
|
|
tenantContext.run(a, async () => {
|
|
await new Promise((r) => setTimeout(r, 10));
|
|
return tenantContext.get()?.orgSlug;
|
|
}),
|
|
tenantContext.run(b, async () => {
|
|
await new Promise((r) => setTimeout(r, 5));
|
|
return tenantContext.get()?.orgSlug;
|
|
}),
|
|
]);
|
|
expect(got1).toBe("a");
|
|
expect(got2).toBe("b");
|
|
});
|
|
|
|
it("supports nested run() overriding", () => {
|
|
const inner: TenantContext = { ...sample, orgSlug: "inner" };
|
|
tenantContext.run(sample, () => {
|
|
tenantContext.run(inner, () => {
|
|
expect(tenantContext.get()?.orgSlug).toBe("inner");
|
|
});
|
|
expect(tenantContext.get()?.orgSlug).toBe("acme");
|
|
});
|
|
});
|
|
|
|
it("enrich() patches the active context in-place", () => {
|
|
tenantContext.run({ ...sample }, () => {
|
|
tenantContext.enrich({
|
|
userId: "usr_1",
|
|
orgId: "org_1",
|
|
projectId: "proj_1",
|
|
envType: "PRODUCTION",
|
|
});
|
|
expect(tenantContext.get()).toMatchObject({
|
|
orgSlug: "acme",
|
|
projectSlug: "web",
|
|
envSlug: "prod",
|
|
userId: "usr_1",
|
|
orgId: "org_1",
|
|
projectId: "proj_1",
|
|
envType: "PRODUCTION",
|
|
});
|
|
});
|
|
});
|
|
|
|
it("enrich() outside run() is a no-op", () => {
|
|
expect(() => tenantContext.enrich({ orgId: "x" })).not.toThrow();
|
|
expect(tenantContext.get()).toBeUndefined();
|
|
});
|
|
|
|
it("supports starting from an empty scope and enriching userId only (non-tenant page)", () => {
|
|
tenantContext.run({}, () => {
|
|
tenantContext.enrich({ userId: "usr_1" });
|
|
expect(tenantContext.get()).toEqual({ userId: "usr_1" });
|
|
});
|
|
});
|
|
|
|
it("enrich() patches do not bleed across concurrent run() scopes", async () => {
|
|
const a: TenantContext = { ...sample, orgSlug: "a" };
|
|
const b: TenantContext = { ...sample, orgSlug: "b" };
|
|
const [got1, got2] = await Promise.all([
|
|
tenantContext.run(a, async () => {
|
|
await new Promise((r) => setTimeout(r, 5));
|
|
tenantContext.enrich({ orgId: "org_a" });
|
|
await new Promise((r) => setTimeout(r, 5));
|
|
return tenantContext.get();
|
|
}),
|
|
tenantContext.run(b, async () => {
|
|
await new Promise((r) => setTimeout(r, 10));
|
|
tenantContext.enrich({ orgId: "org_b" });
|
|
return tenantContext.get();
|
|
}),
|
|
]);
|
|
expect(got1?.orgId).toBe("org_a");
|
|
expect(got2?.orgId).toBe("org_b");
|
|
});
|
|
});
|