test: add enterprise feature mock utilities

2026-08-03 07:46:42 +02:00 · 2026-06-13 16:22:01 +08:00
parent 86d6f50ea6
commit f6d5479334
2 changed files with 85 additions and 0 deletions
@@ -0,0 +1,34 @@
+import { vi } from "vitest";
+import type { EnterpriseFeature } from "@snapotter/enterprise";
+
+export function mockEnterpriseFeatures(features: EnterpriseFeature[]) {
+  vi.doMock("@snapotter/enterprise", () => ({
+    isFeatureEnabled: (feature: string) =>
+      features.includes(feature as EnterpriseFeature),
+    getActiveLicense: () => ({
+      org: "test-org",
+      plan: "enterprise" as const,
+      features,
+      seats: 100,
+      expiresAt: new Date(
+        Date.now() + 365 * 24 * 60 * 60 * 1000,
+      ).toISOString(),
+      issuedAt: new Date().toISOString(),
+    }),
+    initEnterprise: vi.fn(),
+    loadS3Storage: vi.fn(),
+    ENTERPRISE_FEATURES: features,
+    PLAN_FEATURES: { team: [], enterprise: features },
+  }));
+}
+
+export function mockNoEnterprise() {
+  vi.doMock("@snapotter/enterprise", () => ({
+    isFeatureEnabled: () => false,
+    getActiveLicense: () => null,
+    initEnterprise: vi.fn(),
+    loadS3Storage: vi.fn(),
+    ENTERPRISE_FEATURES: [],
+    PLAN_FEATURES: { team: [], enterprise: [] },
+  }));
+}