-
-
- {
- onLoad(reactFlowInstance);
- setReactFlowInstance(reactFlowInstance);
- }}
- onMove={(_e, viewPort) => handleZoomLevel(viewPort.zoom)}
- onNodeClick={(_e, node) => {
- onNodeClick(node);
- _e.stopPropagation();
- }}
- onNodeContextMenu={onNodeContextMenu}
- onNodeDrag={dragHandle}
- onNodeDragStart={dragHandle}
- onNodeDragStop={dragHandle}
- onNodeMouseEnter={onNodeMouseEnter}
- onNodeMouseLeave={onNodeMouseLeave}
- onNodeMouseMove={onNodeMouseMove}
- onNodesChange={onNodesChange}
- onPaneClick={onPaneClick}>
- {updatedLineageData && (
-
- )}
-
-
-
-
- {isDrawerOpen &&
- !isEditMode &&
- (selectedEdgeInfo ? (
-
{
- setIsDrawerOpen(false);
- setSelectedEdgeInfo(undefined);
- }}
- />
- ) : (
-
- ))}
-
- {showDeleteModal && (
- {
- setShowDeleteModal(false);
- }}
- onOk={onRemove}>
- {getModalBodyText(selectedEdge)}
-
+
+ {isFullScreen && (
+
)}
+
+
+
+ {
+ onLoad(reactFlowInstance);
+ setReactFlowInstance(reactFlowInstance);
+ }}
+ onMove={(_e, viewPort) => handleZoomLevel(viewPort.zoom)}
+ onNodeClick={(_e, node) => {
+ onNodeClick(node);
+ _e.stopPropagation();
+ }}
+ onNodeContextMenu={onNodeContextMenu}
+ onNodeDrag={dragHandle}
+ onNodeDragStart={dragHandle}
+ onNodeDragStop={dragHandle}
+ onNodeMouseEnter={onNodeMouseEnter}
+ onNodeMouseLeave={onNodeMouseLeave}
+ onNodeMouseMove={onNodeMouseMove}
+ onNodesChange={onNodesChange}
+ onPaneClick={onPaneClick}>
+ {updatedLineageData && (
+
+ )}
+
+
+
+
+ {isDrawerOpen &&
+ !isEditMode &&
+ (selectedEdgeInfo ? (
+
{
+ setIsDrawerOpen(false);
+ setSelectedEdgeInfo(undefined);
+ }}
+ />
+ ) : (
+
+ ))}
+
+ {showDeleteModal && (
+ {
+ setShowDeleteModal(false);
+ }}
+ onOk={onRemove}>
+ {getModalBodyText(selectedEdge)}
+
+ )}
- setPipelineSearchValue(value)}
- onSelect={handlePipelineSelection}
- />
-
+ setPipelineSearchValue(value)}
+ onSelect={handlePipelineSelection}
+ />
+
+
);
};
diff --git a/openmetadata-ui/src/main/resources/ui/src/components/EntityLineage/EntityLineage.interface.ts b/openmetadata-ui/src/main/resources/ui/src/components/EntityLineage/EntityLineage.interface.ts
index 6a276e871d6..ec596d786a6 100644
--- a/openmetadata-ui/src/main/resources/ui/src/components/EntityLineage/EntityLineage.interface.ts
+++ b/openmetadata-ui/src/main/resources/ui/src/components/EntityLineage/EntityLineage.interface.ts
@@ -11,6 +11,7 @@
* limitations under the License.
*/
+import { SourceType } from 'components/searched-data/SearchedData.interface';
import { LoadingState } from 'Models';
import { HTMLAttributes } from 'react';
import { Edge as FlowEdge, FitViewOptions, Node } from 'reactflow';
@@ -33,6 +34,7 @@ export interface EntityLineageProp {
deleted?: boolean;
hasEditAccess?: boolean;
isFullScreen?: boolean;
+ entity?: SourceType;
}
export interface Edge {
diff --git a/openmetadata-ui/src/main/resources/ui/src/components/EntityLineage/Entitylineage.component.test.tsx b/openmetadata-ui/src/main/resources/ui/src/components/EntityLineage/Entitylineage.component.test.tsx
index 4381d44ab46..202265c6f33 100644
--- a/openmetadata-ui/src/main/resources/ui/src/components/EntityLineage/Entitylineage.component.test.tsx
+++ b/openmetadata-ui/src/main/resources/ui/src/components/EntityLineage/Entitylineage.component.test.tsx
@@ -11,7 +11,7 @@
* limitations under the License.
*/
-import { render, screen } from '@testing-library/react';
+import { fireEvent, render, screen } from '@testing-library/react';
import { MOCK_CHILD_MAP, MOCK_LINEAGE_DATA } from 'mocks/Lineage.mock';
import React from 'react';
import { act } from 'react-dom/test-utils';
@@ -107,6 +107,15 @@ jest.mock('../EntityInfoDrawer/EntityInfoDrawer.component', () => {
return jest.fn().mockReturnValue(