chore: fix autoscaling for team plan users (#1021)

We were autoscaling everyone's integrations

Author: geclos

packages/core/src/services/mcpServers/autoScaleService.test.ts

@@ -6,6 +6,7 @@ import { SubscriptionPlan } from '../../plans'
 import { autoScaleInactiveServers } from './autoScaleService'
 import { createWorkspace, createMcpServer } from '../../tests/factories'
 import { setupQueues } from '../../jobs'
+import { workspaces } from '../../schema'
 
 vi.mock('../../jobs', () => ({
   setupQueues: vi.fn(),
@@ -87,10 +88,18 @@ describe('autoScaleInactiveServers', () => {
   it('should not enqueue jobs for servers on non-hobby plans', async () => {
     // Create a workspace with a team plan
     const { workspace } = await createWorkspace()
+
+    // Create a team subscription
+    const currentSubscription = await database
+      .insert(subscriptions)
+      .values({ workspaceId: workspace.id, plan: SubscriptionPlan.TeamV1 })
+      .returning()
+
+    // Assing the subscription to the workspace
     await database
-      .update(subscriptions)
-      .set({ plan: SubscriptionPlan.TeamV1 })
-      .where(eq(subscriptions.workspaceId, workspace.id))
+      .update(workspaces)
+      .set({ currentSubscriptionId: currentSubscription[0]!.id })
+      .where(eq(workspaces.id, workspace.id))
 
     // Create an inactive MCP server
     await createMcpServer({

packages/core/src/services/mcpServers/autoScaleService.ts

@@ -5,6 +5,7 @@ import { database } from '../../client'
 import { and, eq, lt, inArray, gt } from 'drizzle-orm'
 import { SubscriptionPlan } from '../../plans'
 import { setupQueues } from '../../jobs'
+import { workspaces } from '../../schema'
 
 const INACTIVITY_THRESHOLD_MINUTES = 10
 const SCALE_DOWN_REPLICAS = 0
@@ -22,9 +23,10 @@ export async function autoScaleInactiveServers(db = database) {
     const inactiveServers = await db
       .select({ id: mcpServers.id })
       .from(mcpServers)
+      .innerJoin(workspaces, eq(mcpServers.workspaceId, workspaces.id))
       .innerJoin(
         subscriptions,
-        eq(mcpServers.workspaceId, subscriptions.workspaceId),
+        eq(workspaces.currentSubscriptionId, subscriptions.id),
       )
       .where(
         and(