fix: extend enum in all (#27265)

Author: TeddyCr

openmetadata-ui/src/main/resources/ui/src/components/common/Form/JSONSchema/JsonSchemaWidgets/TreeSelectWidget.test.tsx

@@ -37,6 +37,10 @@ const mockProps = {
 };
 
 describe('Test TreeSelectWidget Component', () => {
+  beforeEach(() => {
+    jest.clearAllMocks();
+  });
+
   it('Should render tree select component', async () => {
     render(<TreeSelectWidget {...mockProps} />);
 
@@ -96,4 +100,72 @@ describe('Test TreeSelectWidget Component', () => {
 
     expect(mockOnChange).toHaveBeenCalledTimes(1);
   });
+
+  it('Should pass through "all" when expandAllValue is not set', async () => {
+    const passThroughProps = {
+      ...mockProps,
+      value: [],
+    };
+
+    render(<TreeSelectWidget {...passThroughProps} />);
+
+    const treeSelectInput = await findByRole(
+      screen.getByTestId('tree-select-widget'),
+      'combobox'
+    );
+
+    await act(async () => {
+      userEvent.click(treeSelectInput);
+    });
+
+    await waitFor(() => screen.getByText('All'));
+
+    await act(async () => {
+      fireEvent.click(screen.getByText('All'));
+    });
+
+    expect(mockOnChange).toHaveBeenCalledWith(['all']);
+  });
+
+  it('Should expand "all" into individual enum values when expandAllValue is true', async () => {
+    const expandAllProps = {
+      ...mockProps,
+      options: {
+        enumOptions: [
+          { label: 'PROFILE', value: 'PROFILE' },
+          { label: 'TEST_CASE_RESULTS', value: 'TEST_CASE_RESULTS' },
+          { label: 'ENTITY_HISTORY', value: 'ENTITY_HISTORY' },
+        ],
+      },
+      schema: {
+        ...mockProps.schema,
+        expandAllValue: true,
+      },
+      value: [],
+    };
+
+    render(<TreeSelectWidget {...expandAllProps} />);
+
+    const treeSelectInput = await findByRole(
+      screen.getByTestId('tree-select-widget'),
+      'combobox'
+    );
+
+    await act(async () => {
+      userEvent.click(treeSelectInput);
+    });
+
+    await waitFor(() => screen.getByText('All'));
+
+    await act(async () => {
+      fireEvent.click(screen.getByText('All'));
+    });
+
+    expect(mockOnChange).toHaveBeenCalledWith([
+      'PROFILE',
+      'TEST_CASE_RESULTS',
+      'ENTITY_HISTORY',
+    ]);
+    expect(mockOnChange).not.toHaveBeenCalledWith(['all']);
+  });
 });

openmetadata-ui/src/main/resources/ui/src/components/common/Form/JSONSchema/JsonSchemaWidgets/TreeSelectWidget.tsx

@@ -14,10 +14,12 @@ import Icon from '@ant-design/icons/lib/components/Icon';
 import { WidgetProps } from '@rjsf/utils';
 import { TreeSelect } from 'antd';
 import { startCase } from 'lodash';
-import { FC, ReactNode, useMemo } from 'react';
+import { FC, ReactNode, useCallback, useMemo } from 'react';
 import { ReactComponent as ArrowIcon } from '../../../../../assets/svg/ic-arrow-down.svg';
 import { TEXT_BODY_COLOR } from '../../../../../constants/constants';
 
+const ALL_VALUE = 'all';
+
 const TreeSelectWidget: FC<WidgetProps> = ({
   onFocus,
   onBlur,
@@ -28,8 +30,8 @@ const TreeSelectWidget: FC<WidgetProps> = ({
     () => [
       {
         title: 'All',
-        value: 'all',
-        key: 'all',
+        value: ALL_VALUE,
+        key: ALL_VALUE,
         children: rest.options.enumOptions?.map((node) => ({
           title: startCase(node.label),
           value: node.value,
@@ -40,6 +42,32 @@ const TreeSelectWidget: FC<WidgetProps> = ({
     [rest.options.enumOptions]
   );
 
+  // When `expandAllValue` is set on the schema, the synthetic "all" parent
+  // node is a UI-only convenience: the backend enum does not include "all",
+  // so we expand it back into the full list of enum values before submit.
+  // Without this, selecting "All" would persist a value that violates the
+  // schema (e.g. MetadataExporterApp eventTypes).
+  const shouldExpandAll = Boolean(rest.schema.expandAllValue);
+
+  const handleChange = useCallback(
+    (value: unknown) => {
+      if (
+        shouldExpandAll &&
+        Array.isArray(value) &&
+        value.includes(ALL_VALUE)
+      ) {
+        const expanded =
+          rest.options.enumOptions?.map((option) => option.value) ?? [];
+        onChange(expanded);
+
+        return;
+      }
+
+      onChange(value);
+    },
+    [onChange, shouldExpandAll, rest.options.enumOptions]
+  );
+
   return (
     <TreeSelect
       allowClear
@@ -63,7 +91,7 @@ const TreeSelectWidget: FC<WidgetProps> = ({
       }
       treeData={treeData}
       onBlur={() => onBlur(rest.id, rest.value)}
-      onChange={(value) => onChange(value)}
+      onChange={handleChange}
       onFocus={() => onFocus(rest.id, rest.value)}
       {...rest}>
       {rest.children as ReactNode}