fix(e2e): derive usage from entitled coverage

Co-authored-by: Cursor <cursoragent@cursor.com>

Author: 2 people

apps/sim/e2e/fixtures/scenario-billing.ts

@@ -0,0 +1,58 @@
+import type { ResolvedScenario, ScenarioSubscription } from './scenario'
+
+export function isEntitledSubscription(subscription: ScenarioSubscription): boolean {
+  return subscription.status === 'active' || subscription.status === 'past_due'
+}
+
+export function sameBillingReference(
+  left: ScenarioSubscription,
+  right: ScenarioSubscription
+): boolean {
+  return billingReferenceIdentity(left) === billingReferenceIdentity(right)
+}
+
+export function initialSubscriptionStatus(
+  scenario: ResolvedScenario,
+  definition: ScenarioSubscription
+): 'active' | 'past_due' | 'canceled' {
+  if (definition.status !== 'lapsed') return definition.status
+  const hasEntitledReplacement = scenario.definition.subscriptions.some(
+    (candidate) =>
+      candidate.key !== definition.key &&
+      isEntitledSubscription(candidate) &&
+      sameBillingReference(candidate, definition)
+  )
+  return hasEntitledReplacement ? 'canceled' : 'active'
+}
+
+export function expectedUsageLimit(scenario: ResolvedScenario, userKey: string): string | null {
+  const organizationKeys = new Set(
+    scenario.definition.organizationMemberships
+      .filter((membership) => membership.userKey === userKey)
+      .map((membership) => membership.organizationKey)
+  )
+  const hasEntitledOrganizationSubscription = scenario.definition.subscriptions.some(
+    (candidate) =>
+      candidate.billingReference.kind === 'organization' &&
+      organizationKeys.has(candidate.billingReference.organizationKey) &&
+      isEntitledSubscription(candidate)
+  )
+  if (hasEntitledOrganizationSubscription) return null
+
+  const personalSubscription = scenario.definition.subscriptions.find(
+    (candidate) =>
+      candidate.billingReference.kind === 'user' &&
+      candidate.billingReference.userKey === userKey &&
+      isEntitledSubscription(candidate)
+  )
+  if (personalSubscription?.plan.startsWith('pro_')) {
+    return String(Number(personalSubscription.plan.split('_')[1]) / 200)
+  }
+  return '5'
+}
+
+function billingReferenceIdentity(subscription: ScenarioSubscription): string {
+  return subscription.billingReference.kind === 'user'
+    ? `user/${subscription.billingReference.userKey}`
+    : `organization/${subscription.billingReference.organizationKey}`
+}

apps/sim/e2e/foundation/scenario-validation.spec.ts

@@ -7,6 +7,7 @@ import {
   type ScenarioOrganizationMembership,
   type ScenarioWorkspaceGrant,
 } from '../fixtures/scenario'
+import { expectedUsageLimit } from '../fixtures/scenario-billing'
 import {
   ScenarioValidationError,
   validateScenario,
@@ -254,7 +255,15 @@ test.describe('pure scenario validation', () => {
         : persona
     )
 
-    expect(() => validateScenario(scenario)).not.toThrow()
+    const resolved = validateScenario(scenario)
+    if (lapsed.billingReference.kind !== 'organization') {
+      throw new Error('Expected an organization subscription')
+    }
+    const ownerUserKey = resolved.organizationsByKey.get(
+      lapsed.billingReference.organizationKey
+    )?.ownerUserKey
+    if (!ownerUserKey) throw new Error('Expected an organization owner')
+    expect(expectedUsageLimit(resolved, ownerUserKey)).toBeNull()
   })
 
   test('rejects permission groups without active Enterprise organization/workspace scope', () => {

apps/sim/e2e/scripts/seed-world.ts

@@ -40,6 +40,7 @@ import {
 import { createWorkspace, grantWorkspacePermission } from '../fixtures/factories/workspaces'
 import { E2eHttpClient } from '../fixtures/http-client'
 import type { ResolvedScenario, ScenarioSubscription } from '../fixtures/scenario'
+import { expectedUsageLimit, initialSubscriptionStatus } from '../fixtures/scenario-billing'
 import { validateScenario, validateScenarioSet } from '../fixtures/validate-scenario'
 import { createSettingsPersonaScenarios } from '../settings/personas'
 import { writeSyntheticSecretCanary } from '../support/leak-canary'
@@ -729,57 +730,6 @@ function resolveBillingReference(world: E2EWorld, subscription: ScenarioSubscrip
       ).id
 }
 
-function initialSubscriptionStatus(
-  scenario: ResolvedScenario,
-  definition: ScenarioSubscription
-): 'active' | 'past_due' | 'canceled' {
-  if (definition.status !== 'lapsed') return definition.status
-  const hasEntitledReplacement = scenario.definition.subscriptions.some(
-    (candidate) =>
-      candidate.key !== definition.key &&
-      (candidate.status === 'active' || candidate.status === 'past_due') &&
-      sameBillingReference(candidate, definition)
-  )
-  return hasEntitledReplacement ? 'canceled' : 'active'
-}
-
-function sameBillingReference(left: ScenarioSubscription, right: ScenarioSubscription): boolean {
-  const identity = (subscription: ScenarioSubscription): string =>
-    subscription.billingReference.kind === 'user'
-      ? `user/${subscription.billingReference.userKey}`
-      : `organization/${subscription.billingReference.organizationKey}`
-  return identity(left) === identity(right)
-}
-
-function expectedUsageLimit(scenario: ResolvedScenario, userKey: string): string | null {
-  const membership = scenario.definition.organizationMemberships.find(
-    (candidate) => candidate.userKey === userKey
-  )
-  if (membership) {
-    const organizationSubscription = scenario.definition.subscriptions.find(
-      (candidate) =>
-        candidate.billingReference.kind === 'organization' &&
-        candidate.billingReference.organizationKey === membership.organizationKey
-    )
-    if (
-      organizationSubscription?.status === 'active' ||
-      organizationSubscription?.status === 'past_due'
-    ) {
-      return null
-    }
-  }
-  const personalSubscription = scenario.definition.subscriptions.find(
-    (candidate) =>
-      candidate.billingReference.kind === 'user' &&
-      candidate.billingReference.userKey === userKey &&
-      (candidate.status === 'active' || candidate.status === 'past_due')
-  )
-  if (personalSubscription?.plan.startsWith('pro_')) {
-    return String(Number(personalSubscription.plan.split('_')[1]) / 200)
-  }
-  return '5'
-}
-
 function matchesEnterpriseMetadata(
   metadata: unknown,
   definition: ScenarioSubscription,