2 files changed, 44 insertions, 7 deletions
diff --git a/uitest/src/test/java/com/vaadin/tests/components/grid/GridHeightByRowTest.java b/uitest/src/test/java/com/vaadin/tests/components/grid/GridHeightByRowTest.java
new file mode 100644
index 0000000000..e6da5314cc
--- /dev/null
+++ b/uitest/src/test/java/com/vaadin/tests/components/grid/GridHeightByRowTest.java
@@ -0,0 +1,37 @@
+package com.vaadin.tests.components.grid;
+
+import com.vaadin.testbench.elements.ButtonElement;
+import com.vaadin.testbench.elements.GridElement;
+import com.vaadin.testbench.parallel.TestCategory;
+import com.vaadin.tests.tb3.MultiBrowserTest;
+import org.junit.Assert;
+import org.junit.Test;
+
+@TestCategory("grid")
+public class GridHeightByRowTest extends MultiBrowserTest {
+    @Test
+    public void testHeightByRow() {
+        openTestURL();
+
+        GridElement grid = $(GridElement.class).first();
+        ButtonElement addButton = $(ButtonElement.class).caption("Add Data")
+                .first();
+        ButtonElement removeButton = $(ButtonElement.class)
+                .caption("Remove Data").first();
+
+        // 10 rows and the header, each has height of 50
+        Assert.assertEquals(550, grid.getSize().getHeight());
+
+        addButton.click();
+
+        Assert.assertEquals(600, grid.getSize().getHeight());
+
+        addButton.click();
+
+        Assert.assertEquals(650, grid.getSize().getHeight());
+
+        removeButton.click();
+
+        Assert.assertEquals(600, grid.getSize().getHeight());
+    }
+}
diff --git a/uitest/src/test/java/com/vaadin/tests/components/menubar/MenuBarIconsTest.java b/uitest/src/test/java/com/vaadin/tests/components/menubar/MenuBarIconsTest.java
index 3415663ca5..bead1fb71d 100644
--- a/uitest/src/test/java/com/vaadin/tests/components/menubar/MenuBarIconsTest.java
+++ b/uitest/src/test/java/com/vaadin/tests/components/menubar/MenuBarIconsTest.java
@@ -1,17 +1,16 @@
 package com.vaadin.tests.components.menubar;
 
-import static org.junit.Assert.assertEquals;
-import static org.junit.Assert.assertTrue;
-
-import org.junit.Assume;
-import org.junit.Test;
-import org.openqa.selenium.WebElement;
-
 import com.vaadin.server.FontAwesome;
 import com.vaadin.testbench.By;
 import com.vaadin.testbench.elements.MenuBarElement;
 import com.vaadin.testbench.parallel.BrowserUtil;
 import com.vaadin.tests.tb3.SingleBrowserTest;
+import org.junit.Assume;
+import org.junit.Test;
+import org.openqa.selenium.WebElement;
+
+import static org.junit.Assert.assertEquals;
+import static org.junit.Assert.assertTrue;
 
 public class MenuBarIconsTest extends SingleBrowserTest {
 
@@ -70,6 +69,7 @@ public class MenuBarIconsTest extends SingleBrowserTest {
         assertImage(image, moreItem);
 
         moreItem.click();
+        waitForElementPresent(By.className("v-menubar-submenu"));
         WebElement filler5 = moreItem.findElement(By.vaadin("#Filler 5"));
         assertImage(image, filler5);