Fix for: TreeTable with ContainerHierarchicalWrapper not correctly

displaying child items (#15421)

Change-Id: Iadf1faa979fbae412b55551c6622b0429039a21c

1 files changed, 14 insertions, 0 deletions

diff --git a/server/src/com/vaadin/data/util/ContainerHierarchicalWrapper.java b/server/src/com/vaadin/data/util/ContainerHierarchicalWrapper.java
index 0bfec33957..199d186fab 100644
--- a/server/src/com/vaadin/data/util/ContainerHierarchicalWrapper.java
+++ b/server/src/com/vaadin/data/util/ContainerHierarchicalWrapper.java
@@ -450,6 +450,8 @@ public class ContainerHierarchicalWrapper implements Container.Hierarchical,
             // Update parent
             parent.remove(itemId);
 
+            fireItemSetChangeIfAbstractContainer();
+
             return true;
         }
 
@@ -490,10 +492,22 @@ public class ContainerHierarchicalWrapper implements Container.Hierarchical,
             }
         }
 
+        fireItemSetChangeIfAbstractContainer();
+
         return true;
     }
 
     /**
+     * inform container (if it is instance of AbstractContainer) about the
+     * change in hierarchy (#15421)
+     */
+    private void fireItemSetChangeIfAbstractContainer() {
+        if (container instanceof AbstractContainer) {
+            ((AbstractContainer) container).fireItemSetChange();
+        }
+    }
+
+    /**
      * Creates a new Item into the Container, assigns it an automatic ID, and
      * adds it to the hierarchy.
      *