feat: update openness and height for all nodes, not only updated

Author: Lodin

__tests__/FixedSizeTree.spec.tsx

@@ -291,16 +291,16 @@ describe('FixedSizeTree', () => {
       });
 
       it('resets current openness to default', async () => {
+        const {records} = component.state();
+
+        for (const id in records) {
+          records[id].isOpen = false;
+        }
+
         // Imitate closing the foo-1 node
         component.setState({
           order: ['foo-1'],
-          records: {
-            ...component.state().records,
-            'foo-1': {
-              ...component.state().records['foo-1'],
-              isOpen: false,
-            },
-          },
+          records,
         });
 
         await treeInstance.recomputeTree({useDefaultOpenness: true});

__tests__/VariableSizeTree.spec.tsx

@@ -323,16 +323,16 @@ describe('VariableSizeTree', () => {
       });
 
       it('resets current openness to default', async () => {
+        const {records} = component.state();
+
+        for (const id in records) {
+          records[id].isOpen = false;
+        }
+
         // Imitate closing the foo-1 node
         component.setState({
           order: ['foo-1'],
-          records: {
-            ...component.state().records,
-            'foo-1': {
-              ...component.state().records['foo-1'],
-              isOpen: false,
-            },
-          },
+          records,
         });
 
         await treeInstance.recomputeTree({useDefaultOpenness: true});

src/FixedSizeTree.tsx

@@ -112,6 +112,12 @@ export default class FixedSizeTree<T> extends React.PureComponent<
     const records = {...prevRecords};
     const iter = treeWalker(refreshNodes);
 
+    if (useDefaultOpenness) {
+      for (const id in records) {
+        records[id].isOpen = records[id].data.isOpenByDefault;
+      }
+    }
+
     let isPreviousOpened = false;
 
     while (true) {

src/VariableSizeTree.tsx

@@ -157,6 +157,18 @@ export default class VariableSizeTree<T> extends React.PureComponent<
     const records = {...prevRecords};
     const iter = treeWalker(refreshNodes);
 
+    if (useDefaultHeight || useDefaultOpenness) {
+      for (const id in records) {
+        if (useDefaultHeight) {
+          records[id].height = records[id].data.defaultHeight;
+        }
+
+        if (useDefaultOpenness) {
+          records[id].isOpen = records[id].data.isOpenByDefault;
+        }
+      }
+    }
+
     let isPreviousOpened = false;
 
     while (true) {