Test for #7292

svn changeset:19864/svn branch:6.7

diff --git a/tests/src/com/vaadin/tests/components/treetable/KeepAllItemsVisible.java b/tests/src/com/vaadin/tests/components/treetable/KeepAllItemsVisible.java
new file mode 100644 (file)
index 0000000..a558696
--- /dev/null
+++ b/tests/src/com/vaadin/tests/components/treetable/KeepAllItemsVisible.java
@@ -0,0 +1,126 @@
+package com.vaadin.tests.components.treetable;
+
+import com.vaadin.data.Item;
+import com.vaadin.data.Property.ValueChangeEvent;
+import com.vaadin.data.Property.ValueChangeListener;
+import com.vaadin.tests.components.TestBase;
+import com.vaadin.ui.Button;
+import com.vaadin.ui.Button.ClickEvent;
+import com.vaadin.ui.Tree.CollapseEvent;
+import com.vaadin.ui.Tree.CollapseListener;
+import com.vaadin.ui.Tree.ExpandEvent;
+import com.vaadin.ui.Tree.ExpandListener;
+import com.vaadin.ui.TreeTable;
+
+public class KeepAllItemsVisible extends TestBase implements ExpandListener,
+        CollapseListener {
+
+    private static final String CAPTION = "caption";
+
+    @Override
+    protected void setup() {
+        getMainWindow().getContent().setHeight(null);
+        final TreeTable tt = new TreeTable();
+        tt.setWidth("400px");
+        tt.addContainerProperty(CAPTION, String.class, "");
+        for (int i = 0; i < 20; i++) {
+            String id = "Root " + i;
+            Item item = tt.addItem(id);
+            item.getItemProperty(CAPTION).setValue(id);
+            addChildren(tt, id, 5, 2);
+        }
+
+        tt.setSelectable(true);
+        tt.setImmediate(true);
+
+        tt.addListener((ExpandListener) this);
+        tt.addListener((CollapseListener) this);
+        tt.addListener(new ValueChangeListener() {
+
+            public void valueChange(ValueChangeEvent event) {
+                recalculatePageLength(tt);
+                tt.requestRepaint();
+            }
+        });
+        addComponent(tt);
+
+        recalculatePageLength(tt);
+
+        Button b = new Button("Set pagelength to 10",
+                new Button.ClickListener() {
+
+                    public void buttonClick(ClickEvent event) {
+                        tt.setPageLength(10);
+                    }
+                });
+        addComponent(b);
+        b = new Button("Set pagelength to 20", new Button.ClickListener() {
+
+            public void buttonClick(ClickEvent event) {
+                tt.setPageLength(20);
+            }
+        });
+        addComponent(b);
+    }
+
+    private void recalculatePageLength(TreeTable tt) {
+        tt.setPageLength(getVisibleNodeCount(tt));
+    }
+
+    private int getVisibleNodeCount(TreeTable tt) {
+        int count = 0;
+        for (Object rootId : tt.rootItemIds()) {
+            count += countVisibleNodes(tt, rootId);
+        }
+
+        return count;
+
+    }
+
+    private int countVisibleNodes(TreeTable tt, Object node) {
+        int count = 1;
+        if (!tt.isCollapsed(node) && tt.hasChildren(node)) {
+            for (Object childId : tt.getChildren(node)) {
+                count += countVisibleNodes(tt, childId);
+            }
+        }
+
+        return count;
+    }
+
+    private void addChildren(TreeTable tt, String parentId, int nr,
+            int maxNesting) {
+        if (maxNesting < 1) {
+            return;
+        }
+
+        for (int i = 1; i <= nr; i++) {
+            String id = parentId + "/" + i;
+            Item item = tt.addItem(id);
+            item.getItemProperty(CAPTION).setValue(id);
+            tt.setParent(id, parentId);
+            addChildren(tt, id, nr, maxNesting - 1);
+        }
+
+    }
+
+    @Override
+    protected String getDescription() {
+        return "Keeps the TreeTable pagelength so that all expanded items are shown.";
+    }
+
+    @Override
+    protected Integer getTicketNumber() {
+        return 7292;
+    }
+
+    public void nodeCollapse(CollapseEvent event) {
+        recalculatePageLength((TreeTable) event.getSource());
+
+    }
+
+    public void nodeExpand(ExpandEvent event) {
+        recalculatePageLength((TreeTable) event.getSource());
+    }
+
+}