Test for #7839

svn changeset:21898/svn branch:6.7

1 files changed, 238 insertions, 0 deletions

diff --git a/tests/testbench/com/vaadin/tests/components/treetable/TreeTableCacheOnPartialUpdates.java b/tests/testbench/com/vaadin/tests/components/treetable/TreeTableCacheOnPartialUpdates.java
new file mode 100644
index 0000000000..bc1b632daf
--- /dev/null
+++ b/tests/testbench/com/vaadin/tests/components/treetable/TreeTableCacheOnPartialUpdates.java
@@ -0,0 +1,238 @@
+package com.vaadin.tests.components.treetable;

+

+import java.io.Serializable;

+import java.util.ArrayList;

+import java.util.HashMap;

+import java.util.Map;

+

+import com.vaadin.data.Container;

+import com.vaadin.data.Container.Ordered;

+import com.vaadin.data.Property.ValueChangeEvent;

+import com.vaadin.data.Property.ValueChangeListener;

+import com.vaadin.data.util.BeanItemContainer;

+import com.vaadin.data.util.ContainerHierarchicalWrapper;

+import com.vaadin.tests.components.TestBase;

+import com.vaadin.tests.util.Log;

+import com.vaadin.ui.Button;

+import com.vaadin.ui.Button.ClickEvent;

+import com.vaadin.ui.Button.ClickListener;

+import com.vaadin.ui.Component;

+import com.vaadin.ui.NativeButton;

+import com.vaadin.ui.NativeSelect;

+import com.vaadin.ui.Table;

+import com.vaadin.ui.Table.ColumnGenerator;

+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 TreeTableCacheOnPartialUpdates extends TestBase {

+    private Log log = new Log(5);

+

+    public class TestBean implements Serializable {

+        private String col1;

+        private String col2;

+

+        public TestBean() {

+            col1 = "";

+            col2 = "";

+        }

+

+        public TestBean(String col1, String col2) {

+            this.col1 = col1;

+            this.col2 = col2;

+        }

+

+        public String getCol1() {

+            return col1;

+        }

+

+        public void setCol1(String col1) {

+            this.col1 = col1;

+        }

+

+        public String getCol2() {

+            return col2;

+        }

+

+        public void setCol2(String col2) {

+            this.col2 = col2;

+        }

+

+        @Override

+        public String toString() {

+            return "TestBean [col1=" + col1 + ", col2=" + col2 + "]";

+        }

+

+    }

+

+    public class Col3ColumnGenerator implements ColumnGenerator {

+        public Component generateCell(final com.vaadin.ui.Table source,

+                final Object itemId, Object columnId) {

+            TestBean tb = (TestBean) itemId;

+            // if (!tb.getCol1().contains("children")) {

+            // return null;

+            // }

+            String identifier = "Item " + itemId + "/" + columnId;

+            System.out.println("Generating new Button for " + identifier);

+            Button btnCol3 = new NativeButton(identifier);

+            btnCol3.addListener(new Button.ClickListener() {

+                public void buttonClick(ClickEvent event) {

+                    log.log("Button " + event.getButton().getCaption()

+                            + " clicked. Row index: "

+                            + indexOfId(source, itemId));

+                }

+            });

+            return btnCol3;

+        }

+

+    }

+

+    protected int indexOfId(Table source, Object itemId) {

+        Container.Ordered c = (Ordered) source.getContainerDataSource();

+        if (c instanceof Container.Indexed) {

+            return ((Container.Indexed) source).indexOfId(itemId);

+        } else {

+            ArrayList<Object> list = new ArrayList<Object>(source.getItemIds());

+            return list.indexOf(itemId);

+

+        }

+    }

+

+    private TreeTable treeTable;

+    private BeanItemContainer<TestBean> testBeanContainer;

+    private static String[] columnHeaders = new String[] { "Col1", "Col2",

+            "Col3" };

+    private static Object[] visibleColumns = new Object[] { "col1", "col2",

+            "col3" };

+

+    @Override

+    public void setup() {

+        setTheme("reindeer-tests");

+

+        // Force row height to be the same in all browsers so scrolling based on

+        // pixels works as expected

+        Button b = new Button("Show first");

+        addComponent(b);

+        b.addListener(new ClickListener() {

+

+            public void buttonClick(ClickEvent event) {

+                log.log("First visible item id is: "

+                        + treeTable.getCurrentPageFirstItemId());

+            }

+        });

+        NativeSelect cacheRateSelect = new NativeSelect("Cache rate");

+        cacheRateSelect.setImmediate(true);

+        cacheRateSelect.setNullSelectionAllowed(false);

+        cacheRateSelect.addItem(new Integer(0));

+        cacheRateSelect.addItem(new Integer(1));

+        cacheRateSelect.addItem(new Integer(2));

+        cacheRateSelect.setValue(2);

+        cacheRateSelect.addListener(new ValueChangeListener() {

+

+            public void valueChange(ValueChangeEvent event) {

+                treeTable

+                        .setCacheRate((Integer) event.getProperty().getValue());

+            }

+        });

+        addComponent(cacheRateSelect);

+        treeTable = new TreeTable();

+        treeTable.addStyleName("table-equal-rowheight");

+        // treeTable.setPageLength(0);

+        testBeanContainer = new BeanItemContainer<TestBean>(TestBean.class);

+

+        Map<String, Integer> hasChildren = new HashMap<String, Integer>();

+        hasChildren.put("1", 5);

+        hasChildren.put("3", 10);

+        hasChildren.put("5", 20);

+        hasChildren.put("6", 7);

+        hasChildren.put("7", 1);

+        hasChildren.put("40", 20);

+        hasChildren.put("99", 20);

+        treeTable.setContainerDataSource(createContainer(100, hasChildren));

+        treeTable.addGeneratedColumn("col3", new Col3ColumnGenerator());

+        treeTable.addListener(new ExpandListener() {

+

+            public void nodeExpand(ExpandEvent event) {

+                logExpandCollapse(event.getItemId(), "expanded");

+

+            }

+        });

+        treeTable.addListener(new CollapseListener() {

+

+            public void nodeCollapse(CollapseEvent event) {

+                logExpandCollapse(event.getItemId(), "collapsed");

+

+            }

+        });

+        treeTable.setColumnHeaders(columnHeaders);

+        treeTable.setVisibleColumns(visibleColumns);

+        treeTable.setColumnWidth("col1", 150);

+        treeTable.setColumnWidth("col2", 50);

+        treeTable.setHeight("430px");

+        // treeTable.setColumnWidth("col3", 150);

+        addComponent(log);

+        addComponent(treeTable);

+    }

+

+    protected void logExpandCollapse(Object itemId, String operation) {

+        String identifier = "Item " + itemId;

+        log.log("Row " + identifier + " " + operation + ". Row index: "

+                + indexOfId(treeTable, itemId));

+

+    }

+

+    private Container createContainer(int items,

+            Map<String, Integer> hasChildren) {

+        ContainerHierarchicalWrapper container;

+

+        container = new ContainerHierarchicalWrapper(testBeanContainer);

+        populate(container, items, hasChildren);

+        return container;

+    }

+

+    private void populate(Container.Hierarchical container, int items,

+            Map<String, Integer> children) {

+

+        for (int i = 0; i < items; i++) {

+            String row = String.valueOf(i + 1);

+            String rowText = stringForInt(i);

+

+            TestBean itemId = new TestBean(row, rowText);

+            container.addItem(itemId);

+            if (children.containsKey(row)) {

+                itemId.setCol1(itemId.getCol1() + " (children)");

+                for (int j = 0; j < children.get(row); j++) {

+                    TestBean childItemId = new TestBean(row + "." + (j + 1),

+                            rowText + "." + stringForInt(j));

+                    container.addItem(childItemId);

+                    container.setParent(childItemId, itemId);

+                }

+            }

+

+        }

+    }

+

+    private String stringForInt(int i) {

+        int charsAllowed = 26; // a-z

+        if (i >= charsAllowed) {

+            return stringForInt(i / charsAllowed - 1)

+                    + stringForInt(i % charsAllowed);

+        } else {

+            return String.valueOf((char) (i + 'A'));

+        }

+    }

+

+    @Override

+    protected String getDescription() {

+        // TODO Auto-generated method stub

+        return null;

+    }

+

+    @Override

+    protected Integer getTicketNumber() {

+        // TODO Auto-generated method stub

+        return null;

+    }

+}
\ No newline at end of file