@hubspot/cli 7.8.0-experimental.0 → 7.8.0

package/lib/projects/__tests__/LocalDevProcess.test.js

@@ -1,10 +1,13 @@
 import path from 'path';
 import { translateForLocalDev } from '@hubspot/project-parsing-lib';
 import { handleProjectUpload } from '../upload.js';
+import { handleProjectDeploy } from '../deploy.js';
 import { getProjectConfig } from '../config.js';
+import { fetchProject } from '@hubspot/local-dev-lib/api/projects';
+import { isHubSpotHttpError } from '@hubspot/local-dev-lib/errors/index';
 import LocalDevProcess from '../localDev/LocalDevProcess.js';
 import LocalDevLogger from '../localDev/LocalDevLogger.js';
-import DevServerManagerV2 from '../localDev/DevServerManagerV2.js';
+import DevServerManager from '../localDev/DevServerManager.js';
 import { ENVIRONMENTS } from '@hubspot/local-dev-lib/constants/environments';
 import { vi } from 'vitest';
 // Mock @hubspot/ui-extensions-dev-server
@@ -19,9 +22,12 @@ vi.mock('@hubspot/ui-extensions-dev-server', () => ({
 vi.mock('open');
 vi.mock('@hubspot/project-parsing-lib');
 vi.mock('../upload');
+vi.mock('../deploy');
 vi.mock('../config');
+vi.mock('@hubspot/local-dev-lib/api/projects');
+vi.mock('@hubspot/local-dev-lib/errors/index');
 vi.mock('../localDev/LocalDevLogger');
-vi.mock('../localDev/DevServerManagerV2');
+vi.mock('../localDev/DevServerManager');
 // Tests for LocalDevProcess and LocalDevState
 describe('LocalDevProcess', () => {
     let mockLocalDevLogger;
@@ -37,11 +43,36 @@ describe('LocalDevProcess', () => {
         projectConfig: mockProjectConfig,
         targetProjectAccountId: 123,
         targetTestingAccountId: 456,
-        projectId: 789,
+        projectData: {
+            id: 789,
+            name: 'test-project',
+            portalId: 123,
+            createdAt: 0,
+            deletedAt: 0,
+            isLocked: false,
+            updatedAt: 0,
+            latestBuild: {
+                activitySource: { type: 'HUBSPOT_USER', userId: 456 },
+                buildId: 123,
+                createdAt: '2023-01-01T00:00:00Z',
+                deployableState: 'DEPLOYABLE',
+                deployStatusTaskLocator: { id: 'task-123', links: [] },
+                enqueuedAt: '2023-01-01T00:00:00Z',
+                finishedAt: '2023-01-01T00:05:00Z',
+                isAutoDeployEnabled: false,
+                portalId: 123,
+                projectName: 'test-project',
+                startedAt: '2023-01-01T00:01:00Z',
+                status: 'SUCCESS',
+                subbuildStatuses: [],
+                uploadMessage: 'Build completed',
+                autoDeployId: 0,
+                platformVersion: '2025.2',
+            },
+        },
         initialProjectNodes: {},
         initialProjectProfileData: {},
         env: ENVIRONMENTS.PROD,
-        projectName: 'test-project',
     };
     beforeEach(() => {
         vi.clearAllMocks();
@@ -62,6 +93,9 @@ describe('LocalDevProcess', () => {
             uploadSuccess: vi.fn(),
             fileChangeError: vi.fn(),
             uploadWarning: vi.fn(),
+            deployInitiated: vi.fn(),
+            deployError: vi.fn(),
+            deploySuccess: vi.fn(),
         };
         mockDevServerManager = {
             setup: vi.fn().mockResolvedValue(undefined),
@@ -71,7 +105,9 @@ describe('LocalDevProcess', () => {
         };
         // Mock constructors
         LocalDevLogger.mockImplementation(() => mockLocalDevLogger);
-        DevServerManagerV2.mockImplementation(() => mockDevServerManager);
+        DevServerManager.mockImplementation(() => mockDevServerManager);
+        // Mock external functions
+        isHubSpotHttpError.mockReturnValue(false);
         // Create process instance
         process = new LocalDevProcess(mockOptions);
         // Mock process.exit
@@ -141,9 +177,14 @@ describe('LocalDevProcess', () => {
             handleProjectUpload.mockResolvedValue({
                 uploadError: new Error('Upload failed'),
             });
-            const success = await process.uploadProject();
+            const result = await process.uploadProject();
             expect(mockLocalDevLogger.uploadError).toHaveBeenCalledWith(new Error('Upload failed'));
-            expect(success).toBe(false);
+            expect(result).toEqual({
+                uploadSuccess: false,
+                buildSuccess: false,
+                deploySuccess: false,
+                deployId: undefined,
+            });
         });
         it('should handle successful upload', async () => {
             await process.handleConfigFileChange();
@@ -152,38 +193,120 @@ describe('LocalDevProcess', () => {
             });
             handleProjectUpload.mockResolvedValue({
                 uploadError: null,
+                result: {
+                    deployResult: {
+                        id: 'deploy-123',
+                        deployId: 123,
+                        status: 'SUCCESS',
+                    },
+                },
+            });
+            fetchProject.mockResolvedValue({
+                data: {
+                    id: 789,
+                    name: 'test-project',
+                    portalId: 123,
+                    createdAt: 0,
+                    deletedAt: 0,
+                    isLocked: false,
+                    updatedAt: 0,
+                    latestBuild: { id: 'build-1', status: 'SUCCESS' },
+                    deployedBuild: { id: 'build-1', status: 'SUCCESS' },
+                },
             });
-            const success = await process.uploadProject();
+            const result = await process.uploadProject();
+            expect(fetchProject).toHaveBeenCalledWith(mockOptions.targetProjectAccountId, mockOptions.projectConfig.name);
             expect(mockLocalDevLogger.uploadSuccess).toHaveBeenCalled();
             // @ts-expect-error accessing private property for testing
             expect(process.state.uploadWarnings.size).toBe(0);
-            expect(success).toBe(true);
+            expect(result).toEqual({
+                uploadSuccess: true,
+                buildSuccess: true,
+                deploySuccess: true,
+                deployId: 123,
+            });
         });
-        it('should reset projectNodesAtLastUpload', async () => {
-            const mockNodes = { node1: { uid: 'node1' } };
-            const initialProjectNodes = { existingNode: { uid: 'existingNode' } };
+        it('should reset projectNodesAtLastUpload if deploy is successful', async () => {
+            const mockInitialNodes = {
+                node1: {
+                    uid: 'node1',
+                    componentType: 'APP',
+                    localDev: {
+                        componentRoot: '/test/path',
+                        componentConfigPath: '/test/path/config.json',
+                        configUpdatedSinceLastUpload: false,
+                    },
+                    componentDeps: {},
+                    metaFilePath: '/test/path',
+                    config: { name: 'Node 1' },
+                    files: [],
+                },
+            };
+            const mockNewNodes = {
+                node1: {
+                    uid: 'node2',
+                    componentType: 'APP',
+                    localDev: {
+                        componentRoot: '/test/path',
+                        componentConfigPath: '/test/path/config.json',
+                        configUpdatedSinceLastUpload: false,
+                    },
+                    componentDeps: {},
+                    metaFilePath: '/test/path',
+                    config: { name: 'Node 2' },
+                    files: [],
+                },
+            };
             // @ts-expect-error accessing private property for testing
-            process.state._projectNodesAtLastUpload = initialProjectNodes;
+            process.state.projectNodesAtLastDeploy = mockInitialNodes;
             getProjectConfig.mockResolvedValue({
                 projectConfig: mockOptions.projectConfig,
             });
             handleProjectUpload.mockResolvedValue({
                 uploadError: null,
+                result: {
+                    deployResult: {
+                        id: 'deploy-123',
+                        deployId: 456,
+                        status: 'SUCCESS',
+                    },
+                },
             });
             translateForLocalDev.mockResolvedValue({
-                intermediateNodesIndexedByUid: mockNodes,
+                intermediateNodesIndexedByUid: mockNewNodes,
             });
-            const success = await process.uploadProject();
-            // Verify translateForLocalDev was called without projectNodesAtLastUpload option
+            fetchProject.mockResolvedValue({
+                data: {
+                    id: 789,
+                    name: 'test-project',
+                    portalId: 123,
+                    createdAt: 0,
+                    deletedAt: 0,
+                    isLocked: false,
+                    updatedAt: 0,
+                    latestBuild: { id: 'build-1', status: 'SUCCESS' },
+                    deployedBuild: { id: 'build-1', status: 'SUCCESS' },
+                },
+            });
+            const result = await process.uploadProject();
+            // Verify translateForLocalDev was called during updateProjectNodesAfterDeploy
             expect(translateForLocalDev).toHaveBeenCalledWith({
                 projectSourceDir: path.join(mockOptions.projectDir, mockOptions.projectConfig.srcDir),
                 platformVersion: mockOptions.projectConfig.platformVersion,
                 accountId: mockOptions.targetProjectAccountId,
-            }, { projectNodesAtLastUpload: undefined });
+            }, {
+                profile: undefined,
+                projectNodesAtLastUpload: undefined,
+            });
             // Verify projectNodesAtLastUpload was reset to the new nodes
             // @ts-expect-error accessing private property for testing
-            expect(process.state.projectNodesAtLastUpload).toEqual(mockNodes);
-            expect(success).toBe(true);
+            expect(process.state.projectNodesAtLastDeploy).toEqual(mockNewNodes);
+            expect(result).toEqual({
+                uploadSuccess: true,
+                buildSuccess: true,
+                deploySuccess: true,
+                deployId: 456,
+            });
         });
     });
     describe('handleFileChange()', () => {
@@ -252,4 +375,93 @@ describe('LocalDevProcess', () => {
             expect(listener).toHaveBeenCalledTimes(1);
         });
     });
+    describe('deployLatestBuild()', () => {
+        beforeEach(() => {
+            vi.clearAllMocks();
+        });
+        it('should successfully deploy latest build', async () => {
+            const mockDeploy = {
+                deployId: 456,
+                buildId: 123,
+                status: 'SUCCESS',
+                enqueuedAt: '2023-01-01T00:00:00Z',
+                startedAt: '2023-01-01T00:01:00Z',
+                finishedAt: '2023-01-01T00:05:00Z',
+                portalId: 123,
+                projectName: 'test-project',
+                userId: 789,
+                source: 'HUBSPOT_USER',
+                subdeployStatuses: [],
+            };
+            handleProjectDeploy.mockResolvedValue(mockDeploy);
+            const result = await process.deployLatestBuild();
+            expect(mockLocalDevLogger.deployInitiated).toHaveBeenCalled();
+            expect(handleProjectDeploy).toHaveBeenCalledWith(123, // targetProjectAccountId
+            'test-project', // projectName
+            123, // buildId
+            true, // useV2Api
+            false // force
+            );
+            expect(mockLocalDevLogger.deploySuccess).toHaveBeenCalled();
+            expect(result).toEqual({
+                success: true,
+                deployId: 456,
+            });
+        });
+        it('should deploy with force parameter', async () => {
+            const mockDeploy = {
+                deployId: 456,
+                buildId: 123,
+                status: 'SUCCESS',
+                enqueuedAt: '2023-01-01T00:00:00Z',
+                startedAt: '2023-01-01T00:01:00Z',
+                finishedAt: '2023-01-01T00:05:00Z',
+                portalId: 123,
+                projectName: 'test-project',
+                userId: 789,
+                source: 'HUBSPOT_USER',
+                subdeployStatuses: [],
+            };
+            handleProjectDeploy.mockResolvedValue(mockDeploy);
+            const result = await process.deployLatestBuild(true);
+            expect(handleProjectDeploy).toHaveBeenCalledWith(123, // targetProjectAccountId
+            'test-project', // projectName
+            123, // buildId
+            true, // useV2Api
+            true // force
+            );
+            expect(result).toEqual({
+                success: true,
+                deployId: 456,
+            });
+        });
+        it('should return error when no build exists', async () => {
+            // Create a process without latestBuild
+            const optionsWithoutBuild = {
+                ...mockOptions,
+                projectData: {
+                    ...mockOptions.projectData,
+                    latestBuild: undefined,
+                },
+            };
+            const processWithoutBuild = new LocalDevProcess(optionsWithoutBuild);
+            const result = await processWithoutBuild.deployLatestBuild();
+            expect(mockLocalDevLogger.deployInitiated).toHaveBeenCalled();
+            expect(mockLocalDevLogger.deployError).toHaveBeenCalledWith('Error deploying project. No build was found to deploy.');
+            expect(result).toEqual({
+                success: false,
+            });
+            expect(handleProjectDeploy).not.toHaveBeenCalled();
+        });
+        it('should handle deploy failure when no deploy object returned', async () => {
+            handleProjectDeploy.mockResolvedValue(undefined);
+            const result = await process.deployLatestBuild();
+            expect(mockLocalDevLogger.deployInitiated).toHaveBeenCalled();
+            expect(handleProjectDeploy).toHaveBeenCalled();
+            expect(result).toEqual({
+                success: false,
+            });
+            expect(mockLocalDevLogger.deploySuccess).not.toHaveBeenCalled();
+        });
+    });
 });

package/lib/projects/__tests__/LocalDevWebsocketServer.test.js

@@ -1,12 +1,12 @@
 import { WebSocketServer } from 'ws';
 import { isPortManagerServerRunning, requestPorts, } from '@hubspot/local-dev-lib/portManager';
-import { logger } from '@hubspot/local-dev-lib/logger';
+import { uiLogger } from '../../ui/logger.js';
 import { LOCAL_DEV_UI_MESSAGE_RECEIVE_TYPES, LOCAL_DEV_UI_MESSAGE_SEND_TYPES, LOCAL_DEV_SERVER_MESSAGE_TYPES, } from '../../constants.js';
 import LocalDevWebsocketServer from '../localDev/LocalDevWebsocketServer.js';
 import { lib } from '../../../lang/en.js';
 vi.mock('ws');
 vi.mock('@hubspot/local-dev-lib/portManager');
-vi.mock('@hubspot/local-dev-lib/logger');
+vi.mock('../../ui/logger.js');
 describe('LocalDevWebsocketServer', () => {
     let mockLocalDevProcess;
     let mockWebSocket;
@@ -30,8 +30,16 @@ describe('LocalDevWebsocketServer', () => {
         mockLocalDevProcess = {
             addStateListener: vi.fn(),
             removeStateListener: vi.fn(),
-            uploadProject: vi.fn(),
+            uploadProject: vi.fn().mockResolvedValue({}),
             sendDevServerMessage: vi.fn(),
+            projectData: {
+                name: 'test-project',
+                id: 123,
+                latestBuild: { id: 'build-1', status: 'SUCCESS' },
+                deployedBuild: { id: 'build-1', status: 'SUCCESS' },
+            },
+            targetProjectAccountId: 456,
+            targetTestingAccountId: 789,
         };
         // Mock WebSocketServer constructor
         WebSocketServer.mockImplementation(() => mockWebSocketServer);
@@ -51,39 +59,68 @@ describe('LocalDevWebsocketServer', () => {
             await server.start();
             expect(WebSocketServer).toHaveBeenCalledWith({ port: 1234 });
             expect(mockWebSocketServer.on).toHaveBeenCalledWith('connection', expect.any(Function));
-            expect(logger.log).toHaveBeenCalled();
+            expect(uiLogger.log).toHaveBeenCalled();
         });
-        it('should accept connection from valid origin', async () => {
-            isPortManagerServerRunning.mockResolvedValue(true);
-            requestPorts.mockResolvedValue({
-                'local-dev-ui-websocket-server': 1234,
+        describe('valid origins', () => {
+            const validOrigins = [
+                'https://app.hubspot.com',
+                'https://app.hubspotqa.com',
+                'https://local.hubspot.com',
+                'https://local.hubspotqa.com',
+                'https://app-na2.hubspot.com',
+                'https://app-na2.hubspotqa.com',
+                'https://app-na3.hubspot.com',
+                'https://app-na3.hubspotqa.com',
+                'https://app-ap1.hubspot.com',
+                'https://app-ap1.hubspotqa.com',
+                'https://app-eu1.hubspot.com',
+                'https://app-eu1.hubspotqa.com',
+            ];
+            validOrigins.forEach(origin => {
+                it(`should accept connection from ${origin}`, async () => {
+                    isPortManagerServerRunning.mockResolvedValue(true);
+                    requestPorts.mockResolvedValue({
+                        'local-dev-ui-websocket-server': 1234,
+                    });
+                    await server.start();
+                    // Get the connection callback
+                    const connectionCallback = mockWebSocketServer.on.mock
+                        .calls[0][1];
+                    // Simulate connection from valid origin
+                    connectionCallback(mockWebSocket, {
+                        headers: { origin },
+                    });
+                    expect(mockWebSocket.on).toHaveBeenCalledWith('message', expect.any(Function));
+                    expect(mockLocalDevProcess.addStateListener).toHaveBeenCalledWith('projectNodes', expect.any(Function));
+                    expect(mockWebSocket.close).not.toHaveBeenCalled();
+                });
             });
-            await server.start();
-            // Get the connection callback
-            const connectionCallback = mockWebSocketServer.on.mock.calls[0][1];
-            // Simulate connection from valid origin
-            connectionCallback(mockWebSocket, {
-                headers: { origin: 'https://app.hubspot.com' },
-            });
-            expect(mockWebSocket.on).toHaveBeenCalledWith('message', expect.any(Function));
-            expect(mockLocalDevProcess.addStateListener).toHaveBeenCalledWith('projectNodes', expect.any(Function));
-            expect(mockWebSocket.close).not.toHaveBeenCalled();
         });
-        it('should reject connection from invalid origin', async () => {
-            isPortManagerServerRunning.mockResolvedValue(true);
-            requestPorts.mockResolvedValue({
-                'local-dev-ui-websocket-server': 1234,
+        describe('invalid origins', () => {
+            const invalidOrigins = [
+                'https://malicious-site.com',
+                'https://app.malicious-site.com',
+                'https://app.hubspot.com.evil.com',
+            ];
+            invalidOrigins.forEach(origin => {
+                it(`should reject connection from "${origin}"`, async () => {
+                    isPortManagerServerRunning.mockResolvedValue(true);
+                    requestPorts.mockResolvedValue({
+                        'local-dev-ui-websocket-server': 1234,
+                    });
+                    await server.start();
+                    // Get the connection callback
+                    const connectionCallback = mockWebSocketServer.on.mock
+                        .calls[0][1];
+                    // Simulate connection from invalid origin
+                    connectionCallback(mockWebSocket, {
+                        headers: { origin },
+                    });
+                    expect(mockWebSocket.close).toHaveBeenCalledWith(1008, lib.LocalDevWebsocketServer.errors.originNotAllowed(origin));
+                    expect(mockWebSocket.on).not.toHaveBeenCalled();
+                    expect(mockLocalDevProcess.addStateListener).not.toHaveBeenCalled();
+                });
             });
-            await server.start();
-            // Get the connection callback
-            const connectionCallback = mockWebSocketServer.on.mock.calls[0][1];
-            // Simulate connection from invalid origin
-            connectionCallback(mockWebSocket, {
-                headers: { origin: 'https://malicious-site.com' },
-            });
-            expect(mockWebSocket.close).toHaveBeenCalledWith(1008, lib.LocalDevWebsocketServer.errors.originNotAllowed('https://malicious-site.com'));
-            expect(mockWebSocket.on).not.toHaveBeenCalled();
-            expect(mockLocalDevProcess.addStateListener).not.toHaveBeenCalled();
         });
         it('should reject connection with no origin header', async () => {
             isPortManagerServerRunning.mockResolvedValue(true);
@@ -111,7 +148,7 @@ describe('LocalDevWebsocketServer', () => {
             const connectionCallback = mockWebSocketServer.on.mock.calls[0][1];
             // Simulate connection from valid origin
             connectionCallback(mockWebSocket, {
-                headers: { origin: 'https://app.hubspot.com' },
+                headers: { origin: 'https://app-na3.hubspot.com' },
             });
             expect(mockLocalDevProcess.sendDevServerMessage).toHaveBeenCalledWith(LOCAL_DEV_SERVER_MESSAGE_TYPES.WEBSOCKET_SERVER_CONNECTED);
         });
@@ -140,7 +177,7 @@ describe('LocalDevWebsocketServer', () => {
             const messageCallback = mockWebSocket.on.mock.calls[0][1];
             const message = {};
             messageCallback(JSON.stringify(message));
-            expect(logger.error).toHaveBeenCalled();
+            expect(uiLogger.error).toHaveBeenCalled();
         });
         it('should log error for unknown message type', () => {
             const messageCallback = mockWebSocket.on.mock.calls[0][1];
@@ -148,13 +185,13 @@ describe('LocalDevWebsocketServer', () => {
                 type: 'UNKNOWN_TYPE',
             };
             messageCallback(JSON.stringify(message));
-            expect(logger.error).toHaveBeenCalled();
+            expect(uiLogger.error).toHaveBeenCalled();
         });
         it('should log error for invalid JSON', () => {
             const messageCallback = mockWebSocket.on.mock.calls[0][1];
             const invalidJson = 'invalid json';
             messageCallback(invalidJson);
-            expect(logger.error).toHaveBeenCalled();
+            expect(uiLogger.error).toHaveBeenCalled();
         });
     });
     describe('shutdown()', () => {
@@ -205,7 +242,7 @@ describe('LocalDevWebsocketServer', () => {
                 headers: { origin: 'https://app.hubspot.com' },
             });
             connectionCallback(mockWebSocket2, {
-                headers: { origin: 'https://app.hubspotqa.com' },
+                headers: { origin: 'https://app-na2.hubspotqa.com' },
             });
             connectionCallback(mockWebSocket3, {
                 headers: { origin: 'https://local.hubspot.com' },
@@ -215,7 +252,7 @@ describe('LocalDevWebsocketServer', () => {
             expect(mockWebSocket2.on).toHaveBeenCalledWith('message', expect.any(Function));
             expect(mockWebSocket3.on).toHaveBeenCalledWith('message', expect.any(Function));
             // Each connection should trigger state listener setup
-            expect(mockLocalDevProcess.addStateListener).toHaveBeenCalledTimes(9); // 3 listeners per connection * 3 connections
+            expect(mockLocalDevProcess.addStateListener).toHaveBeenCalledTimes(12); // 4 listeners per connection * 3 connections
             // Each connection should trigger dev server message
             expect(mockLocalDevProcess.sendDevServerMessage).toHaveBeenCalledTimes(3);
             expect(mockLocalDevProcess.sendDevServerMessage).toHaveBeenCalledWith(LOCAL_DEV_SERVER_MESSAGE_TYPES.WEBSOCKET_SERVER_CONNECTED);
@@ -225,29 +262,12 @@ describe('LocalDevWebsocketServer', () => {
             expect(mockWebSocket3.close).not.toHaveBeenCalled();
         });
         it('should send project data to each connection independently', () => {
-            // Setup mock project data properties as getters
-            Object.defineProperty(mockLocalDevProcess, 'projectName', {
-                get: () => 'test-project',
-                configurable: true,
-            });
-            Object.defineProperty(mockLocalDevProcess, 'projectId', {
-                get: () => 123,
-                configurable: true,
-            });
-            Object.defineProperty(mockLocalDevProcess, 'targetProjectAccountId', {
-                get: () => 456,
-                configurable: true,
-            });
-            Object.defineProperty(mockLocalDevProcess, 'targetTestingAccountId', {
-                get: () => 789,
-                configurable: true,
-            });
             // Establish multiple connections
             connectionCallback(mockWebSocket1, {
                 headers: { origin: 'https://app.hubspot.com' },
             });
             connectionCallback(mockWebSocket2, {
-                headers: { origin: 'https://app.hubspotqa.com' },
+                headers: { origin: 'https://app-eu1.hubspotqa.com' },
             });
             // Each websocket should receive project data
             expect(mockWebSocket1.send).toHaveBeenCalledWith(JSON.stringify({
@@ -255,6 +275,8 @@ describe('LocalDevWebsocketServer', () => {
                 data: {
                     projectName: 'test-project',
                     projectId: 123,
+                    latestBuild: { id: 'build-1', status: 'SUCCESS' },
+                    deployedBuild: { id: 'build-1', status: 'SUCCESS' },
                     targetProjectAccountId: 456,
                     targetTestingAccountId: 789,
                 },
@@ -264,6 +286,8 @@ describe('LocalDevWebsocketServer', () => {
                 data: {
                     projectName: 'test-project',
                     projectId: 123,
+                    latestBuild: { id: 'build-1', status: 'SUCCESS' },
+                    deployedBuild: { id: 'build-1', status: 'SUCCESS' },
                     targetProjectAccountId: 456,
                     targetTestingAccountId: 789,
                 },
@@ -275,7 +299,7 @@ describe('LocalDevWebsocketServer', () => {
                 headers: { origin: 'https://app.hubspot.com' },
             });
             connectionCallback(mockWebSocket2, {
-                headers: { origin: 'https://app.hubspotqa.com' },
+                headers: { origin: 'https://app-ap1.hubspotqa.com' },
             });
             // Get all the close callbacks for both connections (there should be 2 per connection)
             const closeCallbacks1 = mockWebSocket1.on.mock.calls
@@ -284,16 +308,16 @@ describe('LocalDevWebsocketServer', () => {
             const closeCallbacks2 = mockWebSocket2.on.mock.calls
                 .filter(call => call[0] === 'close')
                 .map(call => call[1]);
-            expect(closeCallbacks1).toHaveLength(3); // projectNodes and appData listeners
-            expect(closeCallbacks2).toHaveLength(3); // projectNodes and appData listeners
+            expect(closeCallbacks1).toHaveLength(4); // projectNodes, appData, devServersStarted, and uploadWarnings listeners
+            expect(closeCallbacks2).toHaveLength(4); // projectNodes, appData, devServersStarted, and uploadWarnings listeners
             // Simulate first connection closing (call all close callbacks)
             closeCallbacks1.forEach(callback => callback());
-            // Should have removed listeners for first connection (2 listeners: projectNodes and appData)
-            expect(mockLocalDevProcess.removeStateListener).toHaveBeenCalledTimes(3);
+            // Should have removed listeners for first connection (4 listeners: projectNodes, appData, devServersStarted, and uploadWarnings)
+            expect(mockLocalDevProcess.removeStateListener).toHaveBeenCalledTimes(4);
             // Simulate second connection closing
             closeCallbacks2.forEach(callback => callback());
             // Should have removed listeners for second connection as well
-            expect(mockLocalDevProcess.removeStateListener).toHaveBeenCalledTimes(6);
+            expect(mockLocalDevProcess.removeStateListener).toHaveBeenCalledTimes(8);
         });
         it('should broadcast state changes to all connected clients', () => {
             // Establish connections
@@ -301,7 +325,7 @@ describe('LocalDevWebsocketServer', () => {
                 headers: { origin: 'https://app.hubspot.com' },
             });
             connectionCallback(mockWebSocket2, {
-                headers: { origin: 'https://app.hubspotqa.com' },
+                headers: { origin: 'https://local.hubspotqa.com' },
             });
             // Get the projectNodes listeners that were registered
             const projectNodesListeners = mockLocalDevProcess.addStateListener.mock.calls
@@ -317,6 +341,8 @@ describe('LocalDevWebsocketServer', () => {
                         componentRoot: '/test/path',
                         componentConfigPath: '/test/path/config.json',
                         configUpdatedSinceLastUpload: false,
+                        removed: false,
+                        parsingErrors: [],
                     },
                     componentDeps: {},
                     metaFilePath: '/test/path',