fix: handle invalid widget index (#6584)

vaadin · Oct 10, 2024 · c0569cc · c0569cc
1 parent 1804bc9
commit c0569cc
Show file tree

Hide file tree

Showing 2 changed files with 26 additions and 2 deletions.
diff --git a/...nt/vaadin-dashboard-flow/src/main/java/com/vaadin/flow/component/dashboard/Dashboard.java b/...nt/vaadin-dashboard-flow/src/main/java/com/vaadin/flow/component/dashboard/Dashboard.java
@@ -98,8 +98,11 @@ public void addWidgetAtIndex(int index, DashboardWidget widget) {
             throw new IllegalArgumentException(
                     "Cannot add a widget with a negative index.");
         }
-        // The case when the index is bigger than the children count is handled
-        // inside the method below
+        if (index > widgets.size()) {
+            throw new IllegalArgumentException(String.format(
+                    "Cannot add a widget with index %d when there are %d widgets",
+                    index, widgets.size()));
+        }
         doAddWidget(index, widget);
         updateClient();
     }

diff --git a/...dashboard-flow/src/test/java/com/vaadin/flow/component/dashboard/tests/DashboardTest.java b/...dashboard-flow/src/test/java/com/vaadin/flow/component/dashboard/tests/DashboardTest.java
@@ -87,6 +87,27 @@ public void addWidgetAtIndex_widgetIsCorrectlyAdded() {
         assertWidgets(dashboard, widget1, widget3, widget2);
     }
 
+    @Test
+    public void addWidgetAtInvalidIndex_exceptionIsThrown() {
+        DashboardWidget widget1 = new DashboardWidget();
+        DashboardWidget widget2 = new DashboardWidget();
+        dashboard.add(widget1);
+        fakeClientCommunication();
+        Assert.assertThrows(IllegalArgumentException.class,
+                () -> dashboard.addWidgetAtIndex(2, widget2));
+        fakeClientCommunication();
+        assertWidgets(dashboard, widget1);
+    }
+
+    @Test
+    public void addWidgetAtNegativeIndex_exceptionIsThrown() {
+        DashboardWidget widget = new DashboardWidget();
+        Assert.assertThrows(IllegalArgumentException.class,
+                () -> dashboard.addWidgetAtIndex(-1, widget));
+        fakeClientCommunication();
+        assertWidgets(dashboard);
+    }
+
     @Test
     public void addNullWidgetAtIndex_exceptionIsThrown() {
         Assert.assertThrows(NullPointerException.class,