diff --git a/openmetadata-ui/src/main/resources/ui/src/components/Entity/EntityInfoDrawer/EdgeInfoDrawer.component.tsx b/openmetadata-ui/src/main/resources/ui/src/components/Entity/EntityInfoDrawer/EdgeInfoDrawer.component.tsx
index 97d63d956cb..ec56f9c4032 100644
--- a/openmetadata-ui/src/main/resources/ui/src/components/Entity/EntityInfoDrawer/EdgeInfoDrawer.component.tsx
+++ b/openmetadata-ui/src/main/resources/ui/src/components/Entity/EntityInfoDrawer/EdgeInfoDrawer.component.tsx
@@ -21,8 +21,10 @@ import { Node } from 'reactflow';
import { ReactComponent as EditIcon } from '../../../assets/svg/edit-new.svg';
import DescriptionV1 from '../../../components/common/EntityDescription/DescriptionV1';
import { DE_ACTIVE_COLOR } from '../../../constants/constants';
+import { LINEAGE_SOURCE } from '../../../constants/Lineage.constants';
import { CSMode } from '../../../enums/codemirror.enum';
import { EntityType } from '../../../enums/entity.enum';
+import { Source } from '../../../generated/type/entityLineage';
import { getNameFromFQN } from '../../../utils/CommonUtils';
import { getLineageDetailsObject } from '../../../utils/EntityLineageUtils';
import entityUtilClassBase from '../../../utils/EntityUtilClassBase';
@@ -266,6 +268,15 @@ const EdgeInfoDrawer = ({
)}
+
+ {`${t('label.lineage-source')}`}
+
+
+ {LINEAGE_SOURCE[edgeEntity.source as keyof typeof Source]}
+
+
)}
diff --git a/openmetadata-ui/src/main/resources/ui/src/components/Entity/EntityInfoDrawer/EdgeInfoDrawer.test.tsx b/openmetadata-ui/src/main/resources/ui/src/components/Entity/EntityInfoDrawer/EdgeInfoDrawer.test.tsx
new file mode 100644
index 00000000000..7904fd2bbd9
--- /dev/null
+++ b/openmetadata-ui/src/main/resources/ui/src/components/Entity/EntityInfoDrawer/EdgeInfoDrawer.test.tsx
@@ -0,0 +1,163 @@
+/*
+ * Copyright 2024 Collate.
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ * http://www.apache.org/licenses/LICENSE-2.0
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+import { act, fireEvent, render, screen } from '@testing-library/react';
+import React from 'react';
+import { Edge } from 'reactflow';
+import { MOCK_NODES_AND_EDGES } from '../../../mocks/Lineage.mock';
+import EdgeInfoDrawer from './EdgeInfoDrawer.component';
+
+jest.mock('../../../components/common/EntityDescription/DescriptionV1', () =>
+ jest.fn().mockImplementation(({ onDescriptionUpdate }) => (
+
+ DescriptionV1
+ onDescriptionUpdate('updatedHTML')}>
+ Update Description
+
+
+ ))
+);
+
+jest.mock('../../../utils/CommonUtils', () => ({
+ getNameFromFQN: jest.fn().mockReturnValue('getNameFromFQN'),
+}));
+
+jest.mock('../../../utils/EntityUtils', () => ({
+ getEntityName: jest.fn().mockReturnValue('username'),
+}));
+
+jest.mock('../../common/Loader/Loader', () =>
+ jest.fn().mockImplementation(() => Loader
)
+);
+
+jest.mock('../../Database/SchemaEditor/SchemaEditor', () => {
+ return jest.fn().mockImplementation(() => SchemaEditor.component
);
+});
+
+jest.mock('../../Modals/ModalWithQueryEditor/ModalWithQueryEditor', () => {
+ return jest.fn().mockImplementation(() => ModalWithQueryEditor
);
+});
+
+jest.mock('antd', () => ({
+ ...jest.requireActual('antd'),
+ Drawer: jest.fn().mockImplementation(({ children, title }) => (
+
+ Drawer
+
{title}
+
{children}
+