From 3cf15b139605854c9441f603bdb788b2f5e916e6 Mon Sep 17 00:00:00 2001
From: =?utf8?q?Sauli=20T=C3=A4hk=C3=A4p=C3=A4=C3=A4?= <sauli@vaadin.com>
Date: Sat, 29 Aug 2015 23:00:21 +0300
Subject: [PATCH] Unify select all behavior with single row selection. (#17742)

Change-Id: I2aec82d16b2508505e855f1ff3a1ca5de341821c
---
 .../src/com/vaadin/client/widgets/Grid.java   | 12 +++++++
 .../components/grid/GridSelectAllCell.java    | 21 ++++++++++++
 .../grid/GridSelectAllCellTest.java           | 34 +++++++++++++++++++
 3 files changed, 67 insertions(+)
 create mode 100644 uitest/src/com/vaadin/tests/components/grid/GridSelectAllCell.java
 create mode 100644 uitest/src/com/vaadin/tests/components/grid/GridSelectAllCellTest.java

diff --git a/client/src/com/vaadin/client/widgets/Grid.java b/client/src/com/vaadin/client/widgets/Grid.java
index 6145d69305..abd5c91eeb 100644
--- a/client/src/com/vaadin/client/widgets/Grid.java
+++ b/client/src/com/vaadin/client/widgets/Grid.java
@@ -2429,6 +2429,18 @@ public class Grid<T> extends ResizeComposite implements
                         });
                 selectAllCheckBox.setValue(selected);
 
+                addHeaderClickHandler(new HeaderClickHandler() {
+                    @Override
+                    public void onClick(GridClickEvent event) {
+                        CellReference<?> targetCell = event.getTargetCell();
+                        int defaultRowIndex = getHeader().getRows().indexOf(getDefaultHeaderRow());
+
+                        if(targetCell.getColumnIndex() == 0 && targetCell.getRowIndex() == defaultRowIndex) {
+                            selectAllCheckBox.setValue(!selectAllCheckBox.getValue(), true);
+                        }
+                    }
+                });
+
                 // Select all with space when "select all" cell is active
                 addHeaderKeyUpHandler(new HeaderKeyUpHandler() {
                     @Override
diff --git a/uitest/src/com/vaadin/tests/components/grid/GridSelectAllCell.java b/uitest/src/com/vaadin/tests/components/grid/GridSelectAllCell.java
new file mode 100644
index 0000000000..1b31c38994
--- /dev/null
+++ b/uitest/src/com/vaadin/tests/components/grid/GridSelectAllCell.java
@@ -0,0 +1,21 @@
+package com.vaadin.tests.components.grid;
+
+import com.vaadin.annotations.Theme;
+import com.vaadin.server.VaadinRequest;
+import com.vaadin.tests.components.AbstractTestUI;
+import com.vaadin.ui.Grid;
+
+@Theme("valo")
+public class GridSelectAllCell extends AbstractTestUI {
+    @Override
+    protected void setup(VaadinRequest request) {
+        Grid grid = new Grid();
+
+        grid.addColumn("foo", String.class);
+        grid.addRow("bar");
+
+        grid.setSelectionMode(Grid.SelectionMode.MULTI);
+
+        addComponent(grid);
+    }
+}
diff --git a/uitest/src/com/vaadin/tests/components/grid/GridSelectAllCellTest.java b/uitest/src/com/vaadin/tests/components/grid/GridSelectAllCellTest.java
new file mode 100644
index 0000000000..423a9f419b
--- /dev/null
+++ b/uitest/src/com/vaadin/tests/components/grid/GridSelectAllCellTest.java
@@ -0,0 +1,34 @@
+package com.vaadin.tests.components.grid;
+
+import static org.hamcrest.MatcherAssert.assertThat;
+import static org.hamcrest.core.Is.is;
+
+import java.io.IOException;
+
+import org.junit.Test;
+import org.openqa.selenium.By;
+import org.openqa.selenium.WebElement;
+import org.openqa.selenium.interactions.Actions;
+
+import com.vaadin.testbench.elements.GridElement;
+import com.vaadin.tests.tb3.MultiBrowserTest;
+
+public class GridSelectAllCellTest extends MultiBrowserTest {
+
+    @Override
+    public void setup() throws Exception {
+        super.setup();
+
+        openTestURL();
+    }
+
+    @Test
+    public void selectAllCellCanBeClicked() throws IOException {
+        GridElement.GridCellElement selectAllCell = $(GridElement.class).first().getHeaderCell(0, 0);
+
+        new Actions(getDriver()).moveToElement(selectAllCell, 2, 2).click().perform();
+
+        WebElement selectAllCheckbox = selectAllCell.findElement(By.cssSelector("input"));
+        assertThat(selectAllCheckbox.isSelected(), is(true));
+    }
+}
\ No newline at end of file
-- 
2.39.5