fixes #3070, implemented item style generator for tree

svn changeset:9958/svn branch:6.2

1 files changed, 59 insertions, 0 deletions

diff --git a/tests/src/com/vaadin/tests/components/tree/ItemStyleGenerator.java b/tests/src/com/vaadin/tests/components/tree/ItemStyleGenerator.java
new file mode 100644
index 0000000000..c4299ec939
--- /dev/null
+++ b/tests/src/com/vaadin/tests/components/tree/ItemStyleGenerator.java
@@ -0,0 +1,59 @@
+package com.vaadin.tests.components.tree;
+
+import java.io.IOException;
+import java.io.StringBufferInputStream;
+
+import com.vaadin.tests.components.TestBase;
+import com.vaadin.ui.Component;
+import com.vaadin.ui.CustomLayout;
+import com.vaadin.ui.Tree;
+
+public class ItemStyleGenerator extends TestBase {
+
+    private Component styles;
+    private String css = "<style type=\"text/css\">"
+            + ".v-tree-node-red {color: red;}"
+            + ".v-tree-node-green {color: green;}"
+            + ".v-tree-node-caption-blue {color:blue;}" //
+            + "</style>";
+
+    @Override
+    protected String getDescription() {
+        return "Item Style generator can be used to style items.";
+    }
+
+    @Override
+    protected Integer getTicketNumber() {
+        return 3070;
+    }
+
+    @Override
+    protected void setup() {
+        try {
+            styles = new CustomLayout(new StringBufferInputStream(css));
+        } catch (IOException e) {
+            e.printStackTrace();
+        }
+        addComponent(styles);
+
+        Tree tree = new Tree();
+
+        tree.setItemStyleGenerator(new Tree.ItemStyleGenerator() {
+            public String getStyle(Object itemId) {
+                // simple return itemId as css style name
+                return itemId.toString();
+            }
+        });
+
+        tree.addItem("red");
+        tree.setChildrenAllowed("red", false);
+        tree.addItem("green");
+        tree.addItem("green children");
+        tree.setParent("green children", "green");
+        tree.addItem("blue");
+        tree.addItem("non-blue-childnode");
+        tree.setParent("non-blue-childnode", "blue");
+
+        addComponent(tree);
+    }
+}