diff --git a/openmetadata-ui/src/main/resources/ui/playwright/utils/lineage.ts b/openmetadata-ui/src/main/resources/ui/playwright/utils/lineage.ts index 7d35a338886..d7f3f24035e 100644 --- a/openmetadata-ui/src/main/resources/ui/playwright/utils/lineage.ts +++ b/openmetadata-ui/src/main/resources/ui/playwright/utils/lineage.ts @@ -660,7 +660,7 @@ export const verifyExportLineagePNG = async ( } await expect( - page.getByTestId('export-type-select').getByTitle('PNG') + page.getByTestId('export-type-select').getByText('PNGBeta') ).toBeVisible(); const [download] = await Promise.all([ diff --git a/openmetadata-ui/src/main/resources/ui/src/components/Entity/EntityExportModalProvider/EntityExportModalProvider.component.tsx b/openmetadata-ui/src/main/resources/ui/src/components/Entity/EntityExportModalProvider/EntityExportModalProvider.component.tsx index 134defd2eb1..e7d18a65455 100644 --- a/openmetadata-ui/src/main/resources/ui/src/components/Entity/EntityExportModalProvider/EntityExportModalProvider.component.tsx +++ b/openmetadata-ui/src/main/resources/ui/src/components/Entity/EntityExportModalProvider/EntityExportModalProvider.component.tsx @@ -10,7 +10,7 @@ * See the License for the specific language governing permissions and * limitations under the License. */ -import { Form, Input, Modal, Select } from 'antd'; +import { Badge, Form, Input, Modal, Select } from 'antd'; import { AxiosError } from 'axios'; import classNames from 'classnames'; import { isString, lowerCase } from 'lodash'; @@ -24,7 +24,10 @@ import React, { } from 'react'; import { useTranslation } from 'react-i18next'; import { useLocation } from 'react-router-dom'; -import { ExportTypes } from '../../../constants/Export.constants'; +import { + BETA_EXPORT_TYPES, + ExportTypes, +} from '../../../constants/Export.constants'; import { getCurrentISODate } from '../../../utils/date-time/DateTimeUtils'; import { isBulkEditRoute } from '../../../utils/EntityBulkEdit/EntityBulkEditUtils'; import exportUtilClassBase from '../../../utils/ExportUtilClassBase'; @@ -274,9 +277,25 @@ export const EntityExportModalProvider = ({