162 lines · 5.1 KB · TypeScript
RawDownload
import { describe, expect, it } from "@effect/vitest";import { Effect, Layer } from "effect"; import { ApiKeyService } from "../auth/api-keys";import { UserStoreService } from "../auth/context";import { WorkOSClient, type WorkOSClientService } from "../auth/workos";import { resolveProtectedPrincipal } from "./protected"; const createdAt = new Date("2026-01-01T00:00:00.000Z"); const stubApiKeys = Layer.succeed(ApiKeyService)({  validate: (value: string) =>    Effect.succeed(      value === "valid_user_key"        ? {            accountId: "user_123",            organizationId: "org_123",            keyId: "api_key_123",          }        : null,    ),  listUserKeys: () => Effect.succeed([]),  createUserKey: () => Effect.die("protected API auth test does not create API keys"),  revokeUserKey: () => Effect.void,}); const stubWorkOS = Layer.succeed(  WorkOSClient,  new Proxy({} as WorkOSClientService, {    get: (_target, prop) => {      if (prop === "authenticateRequest") {        return (request: Request) =>          Effect.succeed(            request.headers.get("cookie") === "wos-session=valid"              ? {                  userId: "user_123",                  email: "session@example.com",                  firstName: "Session",                  lastName: "User",                  avatarUrl: null,                  organizationId: "org_123",                  sessionId: "session_123",                  enterpriseSubjectToken: "workos_access_token",                  enterpriseIdentityProviderTokenUrl: "http://workos.test/oauth2/token",                  enterpriseIdentityProviderClientId: "client_test",                  refreshedSession: undefined,                }              : null,          );      }      if (prop === "listUserMemberships") {        return (userId: string) =>          Effect.succeed({            data:              userId === "user_123"                ? [{ userId, organizationId: "org_123", status: "active" }]                : [],          });      }      return () => Effect.die(`unexpected WorkOSClient.${String(prop)} call`);    },  }),); const stubUsers = Layer.succeed(UserStoreService)({  use: (fn) =>    Effect.promise(() =>      fn({        ensureAccount: async (id: string) => ({ id, createdAt }),        getAccount: async (id: string) => ({ id, createdAt }),        upsertOrganization: async (org: { id: string; name: string }) => ({          ...org,          slug: `org-slug-${org.id}`,          createdAt,        }),        getOrganization: async (id: string) => ({          id,          name: `Org ${id}`,          slug: `org-slug-${id}`,          createdAt,        }),        getOrganizationBySlug: async (slug: string) => ({          id: "org_by_slug",          name: `Org ${slug}`,          slug,          createdAt,        }),      }),    ),}); const run = (request: Request) =>  resolveProtectedPrincipal(request).pipe(    Effect.provide(Layer.mergeAll(stubApiKeys, stubWorkOS, stubUsers)),  ); describe("protected API key auth", () => {  it.effect("resolves a valid bearer API key into protected identity", () =>    Effect.gen(function* () {      const identity = yield* run(        new Request("https://executor.test/api/tools", {          headers: { authorization: "Bearer valid_user_key" },        }),      );       expect(identity).toEqual({        accountId: "user_123",        organizationId: "org_123",        organizationName: "Org org_123",        email: "",        name: null,        avatarUrl: null,        roles: [],      });      expect(identity.enterpriseSubjectToken).toBeUndefined();      expect(identity.enterpriseIdentityProviderTokenUrl).toBeUndefined();      expect(identity.enterpriseIdentityProviderClientId).toBeUndefined();    }),  );   it.effect("resolves a sealed WorkOS session with a server-only enterprise subject token", () =>    Effect.gen(function* () {      const identity = yield* run(        new Request("https://executor.test/api/tools", {          headers: { cookie: "wos-session=valid" },        }),      );       expect(identity).toMatchObject({        accountId: "user_123",        organizationId: "org_123",        organizationName: "Org org_123",        email: "session@example.com",        name: "Session User",        enterpriseSubjectToken: "workos_access_token",        enterpriseIdentityProviderTokenUrl: "http://workos.test/oauth2/token",        enterpriseIdentityProviderClientId: "client_test",      });    }),  );   it.effect("rejects invalid bearer API keys", () =>    Effect.gen(function* () {      const error = yield* Effect.flip(        run(          new Request("https://executor.test/api/tools", {            headers: { authorization: "Bearer invalid_user_key" },          }),        ),      );       // The resolver now raises the SHARED `Unauthorized` carrying the same      // machine code; cloud's failure strategy renders it as the byte-identical      // 401 `{ error: "Invalid API key", code: "invalid_api_key" }`.      expect(error).toMatchObject({        _tag: "Unauthorized",        code: "invalid_api_key",        message: "Invalid API key",      });    }),  );});