address gitar

Author: harsh-vador

openmetadata-ui/src/main/resources/ui/src/components/Entity/EntityInfoDrawer/EdgeInfoDrawer.component.tsx

@@ -23,6 +23,7 @@ import SectionWithEdit from '../../../components/common/SectionWithEdit/SectionW
 import { NO_DATA_PLACEHOLDER } from '../../../constants/constants';
 import { LINEAGE_SOURCE } from '../../../constants/Lineage.constants';
 import { CSMode } from '../../../enums/codemirror.enum';
+import { EntityType } from '../../../enums/entity.enum';
 import { AddLineage } from '../../../generated/api/lineage/addLineage';
 import { Source } from '../../../generated/type/entityLineage';
 import { getNameFromFQN } from '../../../utils/CommonUtils';
@@ -39,7 +40,6 @@ import { ModalWithFunctionEditor } from '../../Modals/ModalWithFunctionEditor/Mo
 import { ModalWithQueryEditor } from '../../Modals/ModalWithQueryEditor/ModalWithQueryEditor';
 import './entity-info-drawer.less';
 import { EdgeInfoDrawerInfo } from './EntityInfoDrawer.interface';
-import { EntityType } from '../../../enums/entity.enum';
 
 const EdgeInfoDrawer = ({
   edge,

openmetadata-ui/src/main/resources/ui/src/components/common/DescriptionSection/DescriptionSection.test.tsx

@@ -12,6 +12,7 @@
  */
 import { fireEvent, render, screen, waitFor } from '@testing-library/react';
 import DescriptionSection from './DescriptionSection';
+import { EntityType } from '../../../enums/entity.enum';
 
 // Mock i18n
 jest.mock('react-i18next', () => ({
@@ -84,7 +85,7 @@ describe('DescriptionSection', () => {
 
   describe('Rendering - Empty Description', () => {
     it('renders title and no-data placeholder when description is empty', () => {
-      const { container } = render(<DescriptionSection />);
+      const { container } = render(<DescriptionSection entityType={EntityType.LINEAGE_EDGE} />);
 
       expect(screen.getByText('label.description')).toBeInTheDocument();
       expect(
@@ -108,6 +109,7 @@ describe('DescriptionSection', () => {
         <DescriptionSection
           hasPermission
           showEditButton
+          entityType={EntityType.LINEAGE_EDGE}
           onDescriptionUpdate={jest.fn()}
         />
       );
@@ -120,6 +122,7 @@ describe('DescriptionSection', () => {
     it('does not show edit control when showEditButton is false', () => {
       render(
         <DescriptionSection
+          entityType={EntityType.LINEAGE_EDGE}
           showEditButton={false}
           onDescriptionUpdate={jest.fn()}
         />
@@ -133,7 +136,7 @@ describe('DescriptionSection', () => {
     });
 
     it('does not show edit control when onDescriptionUpdate is not provided', () => {
-      render(<DescriptionSection showEditButton />);
+      render(<DescriptionSection showEditButton entityType={EntityType.LINEAGE_EDGE} />);
 
       const clickable = document.querySelector(
         '.description-header .edit-icon'
@@ -149,6 +152,7 @@ describe('DescriptionSection', () => {
         <DescriptionSection
           hasPermission
           showEditButton
+          entityType={EntityType.LINEAGE_EDGE}
           onDescriptionUpdate={jest.fn()}
         />
       );
@@ -176,6 +180,7 @@ describe('DescriptionSection', () => {
         <DescriptionSection
           hasPermission
           showEditButton
+          entityType={EntityType.LINEAGE_EDGE}
           onDescriptionUpdate={onUpdate}
         />
       );
@@ -199,6 +204,7 @@ describe('DescriptionSection', () => {
       render(
         <DescriptionSection
           description="Some markdown text"
+          entityType={EntityType.LINEAGE_EDGE}
           onDescriptionUpdate={jest.fn()}
         />
       );
@@ -218,6 +224,7 @@ describe('DescriptionSection', () => {
           hasPermission
           showEditButton
           description="Content"
+          entityType={EntityType.LINEAGE_EDGE}
           onDescriptionUpdate={jest.fn()}
         />
       );