Fix Escalator to properly reset height by rows (#11090)8.5.1

(cherry picked from commit 49f6f45cdf4895f179d9a630df6db096db3cdf66)

1 files changed, 46 insertions, 0 deletions

diff --git a/uitest/src/main/java/com/vaadin/tests/components/grid/GridHeightByRow.java b/uitest/src/main/java/com/vaadin/tests/components/grid/GridHeightByRow.java
new file mode 100644
index 0000000000..5254f9091f
--- /dev/null
+++ b/uitest/src/main/java/com/vaadin/tests/components/grid/GridHeightByRow.java
@@ -0,0 +1,46 @@
+package com.vaadin.tests.components.grid;
+
+import com.vaadin.data.provider.DataProvider;
+import com.vaadin.data.provider.ListDataProvider;
+import com.vaadin.server.VaadinRequest;
+import com.vaadin.shared.ui.grid.HeightMode;
+import com.vaadin.tests.components.AbstractTestUIWithLog;
+import com.vaadin.ui.Button;
+import com.vaadin.ui.Grid;
+
+import java.util.ArrayList;
+import java.util.List;
+
+public class GridHeightByRow extends AbstractTestUIWithLog {
+    @Override
+    protected void setup(VaadinRequest request) {
+        List<String> data = new ArrayList<>();
+        for (int i = 0; i < 10; i++)
+            data.add("Data " + i);
+
+        Grid<String> grid = new Grid<>();
+        grid.addColumn(String::toString).setCaption("Test");
+        ListDataProvider<String> provider = DataProvider.ofCollection(data);
+        grid.setDataProvider(provider);
+
+        grid.setHeightMode(HeightMode.UNDEFINED);
+        grid.setRowHeight(50);
+
+        Button addButton = new Button("Add Data");
+        addButton.addClickListener(event -> {
+            data.add("Data");
+            grid.getDataProvider().refreshAll();
+        });
+
+        Button removeButton = new Button("Remove Data");
+        removeButton.addClickListener(event -> {
+            if (data.isEmpty())
+                return;
+
+            data.remove(0);
+            grid.getDataProvider().refreshAll();
+        });
+
+        addComponents(addButton, removeButton, grid);
+    }
+}