fix: update enzyme wrapper

Author: Lodin

__tests__/FixedSizeTree.spec.tsx

@@ -505,7 +505,7 @@ describe('FixedSizeTree', () => {
           // children. But it sill will set `isOpen` to true.
           'foo-6': true,
         });
-        component.update();
+        component.update(); // Update the wrapper to get the latest changes
 
         const receivedRecords = extractReceivedRecords(
           component.find(FixedSizeList),
@@ -538,6 +538,7 @@ describe('FixedSizeTree', () => {
 
         // @ts-expect-error: Test for non-typescript code.
         await treeInstance.recomputeTree('4');
+        component.update(); // Update the wrapper to get the latest changes
 
         expect(extractReceivedRecords(component.find(FixedSizeList))).toEqual(
           originalRecords,
@@ -552,6 +553,7 @@ describe('FixedSizeTree', () => {
         await treeInstance.recomputeTree({
           'foo-42': false,
         });
+        component.update(); // Update the wrapper to get the latest changes
 
         expect(extractReceivedRecords(component.find(FixedSizeList))).toEqual(
           originalRecords,

__tests__/VariableSizeTree.spec.tsx

@@ -534,7 +534,7 @@ describe('VariableSizeTree', () => {
           // children. But it sill will set `isOpen` to true.
           'foo-6': true,
         });
-        component.update();
+        component.update(); // Update the wrapper to get the latest changes
 
         const receivedRecords = extractReceivedRecords(
           component.find(VariableSizeList),
@@ -567,6 +567,7 @@ describe('VariableSizeTree', () => {
 
         // @ts-expect-error: Test for non-typescript code.
         await treeInstance.recomputeTree('4');
+        component.update(); // Update the wrapper to get the latest changes
 
         expect(
           extractReceivedRecords(component.find(VariableSizeList)),
@@ -581,6 +582,7 @@ describe('VariableSizeTree', () => {
         await treeInstance.recomputeTree({
           'foo-42': false,
         });
+        component.update(); // Update the wrapper to get the latest changes
 
         expect(
           extractReceivedRecords(component.find(VariableSizeList)),