From 7abf6065ae905947f6ae5739d2ff971fff1e608c Mon Sep 17 00:00:00 2001
From: "rorya.lyj" <rorya.lyj@alibaba-inc.com>
Date: Sun, 16 Aug 2020 11:05:01 +0800
Subject: [PATCH] fix: findDOMNodes

---
 packages/rax-simulator-renderer/src/renderer-view.tsx     | 8 ++++----
 .../rax-simulator-renderer/src/utils/find-dom-nodes.ts    | 6 +++++-
 2 files changed, 9 insertions(+), 5 deletions(-)

diff --git a/packages/rax-simulator-renderer/src/renderer-view.tsx b/packages/rax-simulator-renderer/src/renderer-view.tsx
index ae4a80e8d..8ba5a42e4 100644
--- a/packages/rax-simulator-renderer/src/renderer-view.tsx
+++ b/packages/rax-simulator-renderer/src/renderer-view.tsx
@@ -53,9 +53,10 @@ export class Routes extends Component<{ rendererContainer: SimulatorRendererCont
     const { rendererContainer } = this.props;
     return (
       <Fragment>
-        {rendererContainer.documentInstances.map((instance) => {
+        {rendererContainer.documentInstances.map((instance, index) => {
+          console.log("Routes");
           return (
-            <Renderer rendererContainer={rendererContainer} documentInstance={instance} />
+            <Renderer key={index} rendererContainer={rendererContainer} documentInstance={instance} />
           );
         })}
       </Fragment>
@@ -130,8 +131,7 @@ class Renderer extends Component<{ rendererContainer: SimulatorRendererContainer
         customCreateElement={(Component: any, props: any, children: any) => {
           const { __id, __desingMode, ...viewProps } = props;
           viewProps.componentId = __id;
-          const leaf: any = null;
-          // const leaf = host.document.getNode(__id);
+          const leaf = documentInstance.getNode(__id);
           viewProps._leaf = leaf;
           viewProps._componentName = leaf?.componentName;
 
diff --git a/packages/rax-simulator-renderer/src/utils/find-dom-nodes.ts b/packages/rax-simulator-renderer/src/utils/find-dom-nodes.ts
index 97a54ed36..e93102654 100644
--- a/packages/rax-simulator-renderer/src/utils/find-dom-nodes.ts
+++ b/packages/rax-simulator-renderer/src/utils/find-dom-nodes.ts
@@ -9,5 +9,9 @@ export function raxFindDOMNodes(instance: any): Array<Element | Text> | null {
   if (isElement(instance)) {
     return [instance];
   }
-  return findDOMNode(instance);
+  const result = findDOMNode(instance);
+  if (Array.isArray(result)) {
+    return result;
+  }
+  return [result];
 }