From 4e9cc377b10c3a1784a16c1515bfb31e333fed52 Mon Sep 17 00:00:00 2001
From: moklick <info@moritzklack.com>
Date: Wed, 14 Sep 2022 15:32:27 +0200
Subject: [PATCH] chore(node-edges): add data-testid

---
 src/components/Edges/wrapEdge.tsx | 1 +
 src/components/Nodes/wrapNode.tsx | 1 +
 2 files changed, 2 insertions(+)

diff --git a/src/components/Edges/wrapEdge.tsx b/src/components/Edges/wrapEdge.tsx
index 0f2938bd..631decc6 100644
--- a/src/components/Edges/wrapEdge.tsx
+++ b/src/components/Edges/wrapEdge.tsx
@@ -157,6 +157,7 @@ export default (EdgeComponent: ComponentType<EdgeProps>) => {
           targetHandleId={targetHandleId}
           markerStart={markerStartUrl}
           markerEnd={markerEndUrl}
+          data-testid={`rf__edge-${id}`}
         />
         {handleEdgeUpdate && (
           <g
diff --git a/src/components/Nodes/wrapNode.tsx b/src/components/Nodes/wrapNode.tsx
index d6a8d0cc..5efee901 100644
--- a/src/components/Nodes/wrapNode.tsx
+++ b/src/components/Nodes/wrapNode.tsx
@@ -135,6 +135,7 @@ export default (NodeComponent: ComponentType<NodeProps>) => {
         onContextMenu={onContextMenuHandler}
         onClick={onSelectNodeHandler}
         onDoubleClick={onDoubleClickHandler}
+        data-testid={`rf__node-${id}`}
         data-id={id}
       >
         <Provider value={id}>