From 8a06f4bd72932989668fbba3a5a11aae3e491bed Mon Sep 17 00:00:00 2001
From: Chaiwat Ekkaewnumchai <cekkaewnumchai@fb.com>
Date: Thu, 7 May 2020 03:37:49 -0700
Subject: [PATCH] Unit Test for Multiple Selector Data Handler

Summary: per title

Reviewed By: mweststrate

Differential Revision: D21304005

fbshipit-source-id: 97231d49a2c6fb3faeafa15587bc140bdc1dbea3
---
 .../layout/__tests__/Inspector.node.tsx       | 121 ++++++++++++++++++
 desktop/plugins/layout/package.json           |   3 +
 2 files changed, 124 insertions(+)
 create mode 100644 desktop/plugins/layout/__tests__/Inspector.node.tsx

diff --git a/desktop/plugins/layout/__tests__/Inspector.node.tsx b/desktop/plugins/layout/__tests__/Inspector.node.tsx
new file mode 100644
index 000000000..7515c8079
--- /dev/null
+++ b/desktop/plugins/layout/__tests__/Inspector.node.tsx
@@ -0,0 +1,121 @@
+/**
+ * Copyright (c) Facebook, Inc. and its affiliates.
+ *
+ * This source code is licensed under the MIT license found in the
+ * LICENSE file in the root directory of this source tree.
+ *
+ * @format
+ */
+
+import Inspector, {ElementSelectorNode} from '../Inspector';
+import {PluginClient, Element} from 'flipper';
+import React from 'react';
+import {render} from '@testing-library/react';
+
+let inspectorComponent: Inspector | null = null;
+beforeEach(() => {
+  const mockRoot: Element = {
+    id: '10000',
+    name: '10000',
+    expanded: false,
+    children: [],
+    attributes: [],
+    data: {},
+    decoration: '',
+    extraInfo: {},
+  };
+  const client: PluginClient = {
+    send: () => {},
+    call: () => Promise.resolve(mockRoot),
+    subscribe: () => {},
+    supportsMethod: () => Promise.resolve(false),
+  };
+  render(
+    <Inspector
+      client={client}
+      showsSidebar={false}
+      selectedElement={null}
+      selectedAXElement={null}
+      onSelect={() => {}}
+      onDataValueChanged={() => {}}
+      setPersistedState={() => {}}
+      persistedState={{
+        rootElement: null,
+        rootAXElement: null,
+        elements: {},
+        AXelements: {},
+      }}
+      searchResults={null}
+      ref={(e) => {
+        inspectorComponent = e;
+      }}
+    />,
+  );
+});
+
+function constructTestTree(): ElementSelectorNode {
+  // The tree will be:
+  // 10000 ---> 11000 ---> 11100 ---> 11110
+  //       |           |          +-> 11120
+  //       |           +-> 11200
+  //       +--> 12000 ---> 12100
+  //                   +-> 12200 ---> 12210 ---> 12211
+  //                   +-> 12300 ---> 12310
+  //                              +-> 12320
+  return {
+    10000: {
+      11000: {11100: {11110: {}, 11120: {}}, 11200: {}},
+      12000: {
+        12100: {},
+        12200: {12210: {12211: {}}},
+        12300: {12310: {}, 12320: {}},
+      },
+    },
+  };
+}
+
+test('test getPathFromNode without id', () => {
+  const tree = constructTestTree();
+  const path = inspectorComponent?.getPathForNode(tree, null);
+  let subtree = tree;
+  path?.forEach((id) => {
+    subtree = subtree[id];
+    expect(subtree).toBeDefined();
+  });
+  expect(subtree).toEqual({});
+});
+
+test('test getPathFromNode with id (leaf)', () => {
+  const tree = constructTestTree();
+  const path = inspectorComponent?.getPathForNode(tree, '12320');
+  expect(path).toEqual(['10000', '12000', '12300', '12320']);
+});
+
+test('test getPathFromNode with id (non-leaf)', () => {
+  const tree = constructTestTree();
+  const path = inspectorComponent?.getPathForNode(tree, '12210');
+  expect(path).toEqual(['10000', '12000', '12200', '12210']);
+});
+
+test('test getPathFromNode with non-existing id', () => {
+  const tree = constructTestTree();
+  const path = inspectorComponent?.getPathForNode(tree, '12313');
+  expect(path).toBeNull();
+});
+
+test('test getElementLeaves', () => {
+  const tree = constructTestTree();
+  const leaves = inspectorComponent?.getElementLeaves(tree);
+  expect(leaves).toHaveLength(7);
+  expect(leaves).toEqual(
+    expect.arrayContaining([
+      '11110',
+      '11120',
+      '11200',
+      '12100',
+      '12211',
+      '12310',
+      '12320',
+    ]),
+  );
+});
diff --git a/desktop/plugins/layout/package.json b/desktop/plugins/layout/package.json
index b4da590f9..d6fc04acf 100644
--- a/desktop/plugins/layout/package.json
+++ b/desktop/plugins/layout/package.json
@@ -10,6 +10,9 @@
     "deep-equal": "^2.0.1",
     "lodash": "^4.17.15"
   },
+  "devDependencies": {
+    "@testing-library/react": "^10.0.2"
+  },
   "title": "Layout",
   "icon": "target",
   "bugs": {