test(types): call the mothership events route with its context, type the acceptance transaction mock, and shrink the test-types baseline

Author: waleedlatif1

apps/sim/app/api/mothership/events/route.test.ts

@@ -27,8 +27,9 @@ vi.mock('@/lib/mothership/chat/organization-chats', () => mothershipOrganization
 vi.mock('@/lib/mothership/chat-status', () => mothershipChatStatusMock)
 vi.mock('@/lib/workspaces/permissions/utils', () => permissionsMock)
 
-import { GET } from '@/app/api/mothership/events/route'
+import { GET as routeGET } from '@/app/api/mothership/events/route'
 
+const GET = (req: NextRequest) => routeGET(req, { params: Promise.resolve({}) })
 const authorize = mothershipOrganizationChatsMockFns.mockAuthorizeOrganizationChatEvents
 const subscribe = mothershipChatStatusMockFns.mockOnStatusChanged
 

apps/sim/lib/billing/organization.test.ts

@@ -24,6 +24,7 @@ import {
 } from '@sim/testing/mocks/organization-membership.mock'
 import { schemaMock } from '@sim/testing/mocks/schema.mock'
 import { beforeEach, describe, expect, it, vi } from 'vitest'
+import type { DbTransaction } from '@/lib/db/types'
 
 const {
   mockCreateOrganizationWithOwner,
@@ -77,6 +78,8 @@ const { mockAcquireOrganizationMutationLock } = organizationMembershipMockFns
 const { mockAcquireUserBillingIdentityLock } = billingIdentityLockMockFns
 const BILLING_WEBHOOK = { type: 'system', name: 'Billing System' } as const
 const ACCEPTOR = { type: 'user', userId: 'acceptor-1' } as const
+/** The chain mock standing in for the acceptance transaction the Tx variant runs on. */
+const ACCEPTANCE_TX = dbChainMock.db as unknown as DbTransaction
 billingPlanHelpersMockFns.mockIsPaid.mockImplementation((plan) => plan !== 'free')
 
 function queueWhereResponses(responses: unknown[][]) {
@@ -175,7 +178,7 @@ describe('ensureOrganizationForTeamSubscription', () => {
     mockAttachOwnedWorkspacesToOrganizationTx.mockResolvedValueOnce({ usageLimitUserIds: [] })
 
     await ensureOrganizationForTeamSubscriptionTx(
-      dbChainMock.db,
+      ACCEPTANCE_TX,
       {
         id: 'sub-1',
         plan: 'team',
@@ -204,7 +207,7 @@ describe('ensureOrganizationForTeamSubscription', () => {
 
     await expect(
       ensureOrganizationForTeamSubscriptionTx(
-        dbChainMock.db,
+        ACCEPTANCE_TX,
         {
           id: 'sub-1',
           plan: 'team',
@@ -267,16 +270,13 @@ describe('syncSubscriptionUsageLimits', () => {
     mockGetOrganizationIdForSubscriptionReference.mockResolvedValue('org-1')
     mockGetPlanPricing.mockReturnValue({ basePrice: 25 })
 
-    await syncSubscriptionUsageLimits(
-      {
-        id: 'sub-1',
-        plan: 'team_6000',
-        referenceId: 'org-1',
-        status: 'active',
-        seats: 2,
-      },
-      BILLING_WEBHOOK
-    )
+    await syncSubscriptionUsageLimits({
+      id: 'sub-1',
+      plan: 'team_6000',
+      referenceId: 'org-1',
+      status: 'active',
+      seats: 2,
+    })
 
     const update = dbChainMockFns.set.mock.calls[0]?.[0]
     const expression = JSON.stringify(update?.orgUsageLimit)

scripts/check-test-types.baseline.json

@@ -231,14 +231,10 @@
   "apps/sim/app/api/mothership/chats/read/route.test.ts": {
     "TS2554": 3
   },
-  "apps/sim/app/api/mothership/events/route.test.ts": {
-    "TS2554": 10
-  },
   "apps/sim/app/api/organizations/[id]/permission-groups/[groupId]/route.test.ts": {
     "TS2550": 2
   },
   "apps/sim/app/api/organizations/route.test.ts": {
-    "TS2554": 3,
     "TS2769": 1
   },
   "apps/sim/app/api/pinned-items/route.test.ts": {
@@ -458,9 +454,6 @@
     "TS2322": 1,
     "TS2554": 1
   },
-  "apps/sim/app/api/workspaces/route.test.ts": {
-    "TS2554": 2
-  },
   "apps/sim/background/async-execution-correlation.test.ts": {
     "TS2345": 2
   },
@@ -966,9 +959,6 @@
     "TS2345": 2,
     "TS2739": 16
   },
-  "apps/sim/lib/billing/organization.test.ts": {
-    "TS2345": 1
-  },
   "apps/sim/lib/billing/organizations/lock-order.test.ts": {
     "TS2345": 2,
     "TS2352": 1,