fix(billing): scope operation completion to workspace

Amp-Thread-ID: https://ampcode.com/threads/T-01a01a57-6259-75bd-bb52-c149e0f7bb9c
Co-authored-by: Amp <amp@ampcode.com>

Author: dante01yoon

src/platform/workspace/stores/billingOperationStore.test.ts

@@ -1448,6 +1448,32 @@ describe('billingOperationStore', () => {
       })
     })
 
+    it('does not update a new workspace after cancellation completes', async () => {
+      let finishRefresh: () => void = () => {}
+      mockFetchStatus.mockImplementationOnce(
+        () =>
+          new Promise<void>((resolve) => {
+            finishRefresh = resolve
+          })
+      )
+      vi.mocked(workspaceApi.getBillingOpStatus).mockResolvedValue({
+        id: 'op-1',
+        status: 'succeeded',
+        started_at: new Date().toISOString()
+      })
+
+      const store = useBillingOperationStore()
+      const terminal = store.startOperation('op-1', 'cancel')
+      await vi.advanceTimersByTimeAsync(0)
+      expect(mockFetchStatus).toHaveBeenCalledOnce()
+
+      mockActiveWorkspaceId.value = 'workspace-2'
+      finishRefresh()
+      await terminal
+
+      expect(mockUpdateActiveWorkspace).not.toHaveBeenCalled()
+    })
+
     it('resolves the terminal outcome even when the post-success refresh fails', async () => {
       mockFetchStatus.mockRejectedValueOnce(new Error('refresh failed'))
       vi.mocked(workspaceApi.getBillingOpStatus).mockResolvedValue({

src/platform/workspace/stores/billingOperationStore.ts

@@ -404,6 +404,11 @@ export const useBillingOperationStore = defineStore('billingOperation', () => {
       })
     }
 
+    if (operation.workspaceId !== workspaceStore.activeWorkspaceId) {
+      resolveTerminal(opId)
+      return
+    }
+
     const billingContext = useBillingContext()
     if (operation.type === 'subscription') {
       await Promise.allSettled([billingContext.reconcileSubscriptionSuccess()])
@@ -414,6 +419,11 @@ export const useBillingOperationStore = defineStore('billingOperation', () => {
       ])
     }
 
+    if (operation.workspaceId !== workspaceStore.activeWorkspaceId) {
+      resolveTerminal(opId)
+      return
+    }
+
     if (operation.type === 'cancel') {
       useTeamWorkspaceStore().updateActiveWorkspace({ isSubscribed: false })
       resolveTerminal(opId)