box-ui-elements 23.4.0-beta.10 → 23.4.0-beta.12

package/src/elements/content-sidebar/stories/tests/MetadataSidebarRedesign-visual.stories.tsx

@@ -425,6 +425,30 @@ export const MetadataInstanceEditorAIEnabled: StoryObj<typeof MetadataSidebarRed
     },
 };
 
+export const MetadataInstanceEditorAIEnabledAdvancedExtractAgent: StoryObj<typeof MetadataSidebarRedesign> = {
+    args: {
+        features: {
+            ...mockFeatures,
+            'metadata.aiSuggestions.enabled': true,
+            'metadata.extractAdvancedAgents.enabled': true,
+        },
+    },
+    play: async ({ canvasElement }) => {
+        const canvas = within(canvasElement);
+        // Edit one instance
+        const editButton = await canvas.findByRole('button', { name: 'Edit My Template' });
+        await userEvent.click(editButton);
+
+        // Find the SplitButton
+        const splitButton = await canvas.findByRole('button', { name: /Autofill/ });
+        expect(splitButton).toBeVisible();
+
+        await userEvent.hover(splitButton);
+        const dropdownButton = await canvas.findByRole('button', { name: 'See agent options.' });
+        expect(dropdownButton).toBeVisible();
+    },
+};
+
 export const ShowErrorWhenAIAPIIsUnavailable: StoryObj<typeof MetadataSidebarRedesign> = {
     args: {
         features: {

package/src/features/metadata-instance-editor/__tests__/CascadePolicy.test.js

@@ -7,6 +7,11 @@ import { screen, render, within } from '../../../test-utils/testing-library';
 import CascadePolicy from '../CascadePolicy';
 
 describe('features/metadata-instance-editor/CascadePolicy', () => {
+    beforeEach(() => {
+        // reset any previous tests that may have set localStorage
+        localStorage.removeItem('aiAgent');
+    });
+
     test('should correctly render cascade policy read only mode', () => {
         const wrapper = shallow(<CascadePolicy id="fakeId" isCascadingEnabled shouldShowCascadeOptions />);
         expect(wrapper).toMatchSnapshot();
@@ -100,12 +105,12 @@ describe('features/metadata-instance-editor/CascadePolicy', () => {
                     shouldShowCascadeOptions
                 />,
             );
-            expect(screen.getByRole('button', { name: 'Agent Basic' })).toBeInTheDocument();
+            expect(screen.getByRole('combobox', { name: 'Basic' })).toBeInTheDocument();
         });
 
         test('should not render AI agent selector when canUseAIFolderExtractionAgentSelector is false', () => {
             render(<CascadePolicy canEdit canUseAIFolderExtraction shouldShowCascadeOptions />);
-            expect(screen.queryByRole('button', { name: 'Agent Basic' })).not.toBeInTheDocument();
+            expect(screen.queryByRole('combobox', { name: 'Basic' })).not.toBeInTheDocument();
         });
     });
 

package/src/features/metadata-instance-editor/__tests__/Instance.test.js

@@ -863,7 +863,7 @@ describe('Instance Component - React Testing Library', () => {
             const editButton = screen.queryByRole('button', { name: 'Edit Metadata' });
             if (editButton) await userEvent.click(editButton); // Enter edit mode to ensure CascadePolicy options are visible
 
-            expect(screen.getByRole('button', { name: 'Agent Basic' })).toBeInTheDocument();
+            expect(screen.getByRole('combobox', { name: 'Basic' })).toBeInTheDocument();
         });
 
         test('should pass isExistingAIExtractionCascadePolicy=true to CascadePolicy if policy is ai_extract', async () => {

package/src/features/metadata-instance-editor/__tests__/Instances.test.js

@@ -265,7 +265,7 @@ describe('Instances component - canUseAIFolderExtractionAgentSelector prop', ()
         const editButton = screen.getByRole('button', { name: 'Edit Metadata' });
         await userEvent.click(editButton);
 
-        expect(screen.getByRole('button', { name: 'Agent Basic' })).toBeInTheDocument();
+        expect(screen.getByRole('combobox', { name: 'Basic' })).toBeInTheDocument();
     });
 
     test('should not show agent selector in child Instance if canUseAIFolderExtractionAgentSelector is false', async () => {
@@ -280,7 +280,7 @@ describe('Instances component - canUseAIFolderExtractionAgentSelector prop', ()
         const editButton = screen.getByRole('button', { name: 'Edit Metadata' });
         await userEvent.click(editButton);
 
-        expect(screen.queryByRole('button', { name: 'Agent Basic' })).not.toBeInTheDocument();
+        expect(screen.queryByRole('combobox', { name: 'Basic' })).not.toBeInTheDocument();
     });
 
     test('should not show agent selector in child Instance if canUseAIFolderExtractionAgentSelector is undefined', async () => {
@@ -291,6 +291,6 @@ describe('Instances component - canUseAIFolderExtractionAgentSelector prop', ()
         const editButton = screen.getByRole('button', { name: 'Edit Metadata' });
         await userEvent.click(editButton);
 
-        expect(screen.queryByRole('button', { name: 'Agent Basic' })).not.toBeInTheDocument();
+        expect(screen.queryByRole('combobox', { name: 'Basic' })).not.toBeInTheDocument();
     });
 });

package/src/features/metadata-instance-editor/__tests__/MetadataInstanceEditor.test.js

@@ -547,7 +547,7 @@ describe('MetadataInstanceEditor - canUseAIFolderExtractionAgentSelector prop',
         const editButton = await screen.findByRole('button', { name: 'Edit Metadata' }, { timeout: 3000 });
         await userEvent.click(editButton);
 
-        expect(screen.getByRole('button', { name: 'Agent Basic' })).toBeInTheDocument();
+        expect(screen.getByRole('combobox', { name: 'Basic' })).toBeInTheDocument();
     });
 
     test('should not show agent selector if canUseAIFolderExtractionAgentSelector is false', async () => {
@@ -562,7 +562,7 @@ describe('MetadataInstanceEditor - canUseAIFolderExtractionAgentSelector prop',
         const editButton = await screen.findByRole('button', { name: 'Edit Metadata' });
         await userEvent.click(editButton);
 
-        expect(screen.queryByRole('button', { name: 'Agent Basic' })).not.toBeInTheDocument();
+        expect(screen.queryByRole('combobox', { name: 'Basic' })).not.toBeInTheDocument();
     });
 
     test('should not show agent selector if canUseAIFolderExtractionAgentSelector is undefined', async () => {
@@ -573,6 +573,6 @@ describe('MetadataInstanceEditor - canUseAIFolderExtractionAgentSelector prop',
         const editButton = await screen.findByRole('button', { name: 'Edit Metadata' });
         await userEvent.click(editButton);
 
-        expect(screen.queryByRole('button', { name: 'Agent Basic' })).not.toBeInTheDocument();
+        expect(screen.queryByRole('combobox', { name: 'Basic' })).not.toBeInTheDocument();
     });
 });

package/src/utils/sleep.js

@@ -0,0 +1,2 @@
+const sleep = ms => new Promise(resolve => setTimeout(resolve, ms));
+export default sleep;