Fix Grid DetailsGenerator being removed on Container change (#19593)

Change-Id: Ied328804260b1a7bb35093fca1d972735559cf41

1 files changed, 22 insertions, 0 deletions

diff --git a/server/tests/src/com/vaadin/tests/server/component/grid/GridContainerTest.java b/server/tests/src/com/vaadin/tests/server/component/grid/GridContainerTest.java
index 527568eac8..079487d01f 100644
--- a/server/tests/src/com/vaadin/tests/server/component/grid/GridContainerTest.java
+++ b/server/tests/src/com/vaadin/tests/server/component/grid/GridContainerTest.java
@@ -19,11 +19,33 @@ import org.junit.Assert;
 import org.junit.Test;
 
 import com.vaadin.data.util.IndexedContainer;
+import com.vaadin.ui.Component;
 import com.vaadin.ui.Grid;
+import com.vaadin.ui.Grid.DetailsGenerator;
+import com.vaadin.ui.Grid.RowReference;
+import com.vaadin.ui.Label;
 
 public class GridContainerTest {
 
     @Test
+    public void testDetailsGeneratorDoesNotResetOnContainerChange() {
+        Grid grid = new Grid();
+        DetailsGenerator detGen = new DetailsGenerator() {
+
+            @Override
+            public Component getDetails(RowReference rowReference) {
+                return new Label("Empty details");
+            }
+        };
+        grid.setDetailsGenerator(detGen);
+
+        grid.setContainerDataSource(createContainer());
+
+        Assert.assertEquals("DetailsGenerator changed", detGen,
+                grid.getDetailsGenerator());
+    }
+
+    @Test
     public void testSetContainerTwice() throws Exception {
 
         TestGrid grid = new TestGrid();