Add missing Grid.removeAllColumns (#8750)

Fixes #8704

2 files changed, 23 insertions, 0 deletions

diff --git a/server/src/main/java/com/vaadin/ui/Grid.java b/server/src/main/java/com/vaadin/ui/Grid.java
index 60abec4e45..51d6b1db61 100644
--- a/server/src/main/java/com/vaadin/ui/Grid.java
+++ b/server/src/main/java/com/vaadin/ui/Grid.java
@@ -2313,6 +2313,15 @@ public class Grid<T> extends AbstractListing<T> implements HasComponents,
     }
 
     /**
+     * Removes all columns from this Grid.
+     */
+    public void removeAllColumns() {
+        for (Column<T, ?> column : getColumns()) {
+            removeColumn(column);
+        }
+    }
+
+    /**
      * Sets the details component generator.
      *
      * @param generator
diff --git a/server/src/test/java/com/vaadin/tests/server/component/grid/GridTest.java b/server/src/test/java/com/vaadin/tests/server/component/grid/GridTest.java
index 40c753943e..c85cf5ee73 100644
--- a/server/src/test/java/com/vaadin/tests/server/component/grid/GridTest.java
+++ b/server/src/test/java/com/vaadin/tests/server/component/grid/GridTest.java
@@ -404,6 +404,20 @@ public class GridTest {
     }
 
     @Test
+    public void removeAllColumns() {
+        grid.removeAllColumns();
+
+        Assert.assertEquals(Collections.emptyList(), grid.getColumns());
+    }
+
+    @Test
+    public void removeAllColumnsInGridWithoutColumns() {
+        grid.removeAllColumns();
+        grid.removeAllColumns();
+        Assert.assertEquals(Collections.emptyList(), grid.getColumns());
+    }
+
+    @Test
     public void setColumns_reorder() {
         // Will remove other columns
         grid.setColumns("length", "foo");