fix(apollo-react): add data-test id for case E2E

Author: SreedharAvvari

packages/apollo-react/src/canvas/components/StageNode/StageNode.test.tsx

@@ -4,8 +4,8 @@ import { ReactFlowProvider } from '@uipath/apollo-react/canvas/xyflow/react';
 import { beforeEach, describe, expect, it, vi } from 'vitest';
 import type { ListItem } from '../Toolbox';
 import { StageNode } from './StageNode';
-import { StageHeaderChipType } from './StageNode.types';
 import type { StageNodeProps, StageTaskItem } from './StageNode.types';
+import { StageHeaderChipType } from './StageNode.types';
 
 // Mock DndContext and related components
 vi.mock('@dnd-kit/core', () => ({
@@ -202,6 +202,14 @@ const renderStageNode = (props: Partial<StageNodeProps> = {}) => {
   );
 };
 
+describe('StageNode - Test Hooks', () => {
+  it('renders a stable test id for the stage header', () => {
+    renderStageNode();
+
+    expect(screen.getByTestId('stage-header-stage-1')).toBeInTheDocument();
+  });
+});
+
 describe('StageNode - Replace Task Functionality', () => {
   beforeEach(() => {
     vi.clearAllMocks();

packages/apollo-react/src/canvas/components/StageNode/StageNode.tsx

@@ -593,7 +593,7 @@ const StageNodeComponent = (props: StageNodeProps) => {
       onMouseLeave={handleMouseLeave}
     >
       <StageContainer selected={selected} status={status} width={width} style={taskWidthStyle}>
-        <StageHeader isException={isException}>
+        <StageHeader isException={isException} data-testid={`stage-header-${id}`}>
           <Row gap={Spacing.SpacingMicro} align="center" flex={1} minW={0}>
             {icon}
             <Column py={2} flex={1} minW={0}>