feat(tree): add unit tests

plouc · May 5, 2024 · adff587 · adff587
1 parent 70db97a
commit adff587
Show file tree

Hide file tree

Showing 5 changed files with 431 additions and 7 deletions.
diff --git a/packages/tree/src/Label.tsx b/packages/tree/src/Label.tsx
@@ -7,10 +7,8 @@ export const Label = <Datum,>({ label, animatedProps }: LabelComponentProps<Datu
 
     return (
         <animated.g
-            transform={to([animatedProps.x, animatedProps.y], (x, y) => `translate(${x}, ${y})`)}
-            style={{
-                pointerEvents: 'none',
-            }}
+            data-testid={`label.${label.id}`}
+            transform={to([animatedProps.x, animatedProps.y], (x, y) => `translate(${x},${y})`)}
         >
             <animated.g transform={animatedProps.rotation.to(rotation => `rotate(${rotation})`)}>
                 {theme.labels.text.outlineWidth > 0 && (
@@ -29,6 +27,7 @@ export const Label = <Datum,>({ label, animatedProps }: LabelComponentProps<Datu
                     </text>
                 )}
                 <text
+                    data-testid={`label.${label.id}.label`}
                     style={theme.labels.text}
                     textAnchor={label.textAnchor}
                     dominantBaseline={label.baseline}

diff --git a/packages/tree/src/Labels.tsx b/packages/tree/src/Labels.tsx
@@ -57,13 +57,17 @@ export const Labels = <Datum,>({
     })
 
     return (
-        <>
+        <g
+            style={{
+                pointerEvents: 'none',
+            }}
+        >
             {transition((animatedProps, label) =>
                 createElement(labelComponent, {
                     label,
                     animatedProps,
                 })
             )}
-        </>
+        </g>
     )
 }
diff --git a/packages/tree/src/Tree.tsx b/packages/tree/src/Tree.tsx
@@ -172,8 +172,10 @@ const InnerTree = <Datum,>({
             links,
             innerWidth,
             innerHeight,
+            linkGenerator,
+            setCurrentNode,
         }),
-        [nodes, links, innerWidth, innerHeight]
+        [nodes, links, innerWidth, innerHeight, linkGenerator, setCurrentNode]
     )
 
     return (

diff --git a/packages/tree/src/types.ts b/packages/tree/src/types.ts
@@ -162,6 +162,8 @@ export interface CustomLayerProps<Datum> {
     links: ComputedLink<Datum>[]
     innerWidth: number
     innerHeight: number
+    linkGenerator: LinkGenerator
+    setCurrentNode: (node: ComputedNode<Datum> | null) => void
 }
 export type CustomSvgLayer<Datum> = FunctionComponent<CustomLayerProps<Datum>>