diff --git a/packages/core/src/components/Edges/wrapEdge.tsx b/packages/core/src/components/Edges/wrapEdge.tsx
index bd441f01..d19fbd42 100644
--- a/packages/core/src/components/Edges/wrapEdge.tsx
+++ b/packages/core/src/components/Edges/wrapEdge.tsx
@@ -162,6 +162,7 @@ export default (EdgeComponent: ComponentType<EdgeProps>) => {
         onKeyDown={disableKeyboardA11y ? undefined : onKeyDown}
         tabIndex={disableKeyboardA11y ? undefined : 0}
         role={disableKeyboardA11y ? undefined : 'button'}
+        data-testid={`rf__edge-${id}`}
         aria-label={ariaLabel === null ? undefined : ariaLabel ? ariaLabel : `Edge from ${source} to ${target}`}
         aria-describedby={disableKeyboardA11y ? undefined : `${ARIA_EDGE_DESC_KEY}-${rfId}`}
         ref={edgeRef}
diff --git a/packages/core/src/components/Nodes/wrapNode.tsx b/packages/core/src/components/Nodes/wrapNode.tsx
index c681b3f5..8ebb50f9 100644
--- a/packages/core/src/components/Nodes/wrapNode.tsx
+++ b/packages/core/src/components/Nodes/wrapNode.tsx
@@ -169,6 +169,7 @@ export default (NodeComponent: ComponentType<NodeProps>) => {
           ...style,
         }}
         data-id={id}
+        data-testid={`rf__node-${id}`}
         onMouseEnter={onMouseEnterHandler}
         onMouseMove={onMouseMoveHandler}
         onMouseLeave={onMouseLeaveHandler}
diff --git a/packages/core/src/container/ReactFlow/index.tsx b/packages/core/src/container/ReactFlow/index.tsx
index 0a00cb9f..d09a8827 100644
--- a/packages/core/src/container/ReactFlow/index.tsx
+++ b/packages/core/src/container/ReactFlow/index.tsx
@@ -163,7 +163,13 @@ const ReactFlow = forwardRef<ReactFlowRefType, ReactFlowProps>(
     const rfId = useId();
 
     return (
-      <div {...rest} style={{ ...style, ...wrapperStyle }} ref={ref} className={cc(['react-flow', className])}>
+      <div
+        {...rest}
+        style={{ ...style, ...wrapperStyle }}
+        ref={ref}
+        className={cc(['react-flow', className])}
+        data-testid="rf__wrapper"
+      >
         <Wrapper>
           <GraphView
             onInit={onInit}