Use HierarchicalContainer as default container for Tree (#3950)

2 files changed, 74 insertions, 25 deletions

diff --git a/server/tests/src/com/vaadin/tests/server/component/tree/TestHasChildren.java b/server/tests/src/com/vaadin/tests/server/component/tree/TestHasChildren.java
deleted file mode 100644
index 66535d3ffb..0000000000
--- a/server/tests/src/com/vaadin/tests/server/component/tree/TestHasChildren.java
+++ /dev/null
@@ -1,25 +0,0 @@
-package com.vaadin.tests.server.component.tree;
-
-import junit.framework.TestCase;
-
-import com.vaadin.ui.Tree;
-
-public class TestHasChildren extends TestCase {
-
-    private Tree tree;
-
-    @Override
-    protected void setUp() {
-        tree = new Tree();
-        tree.addItem("parent");
-        tree.addItem("child");
-        tree.setChildrenAllowed("parent", true);
-        tree.setParent("child", "parent");
-    }
-
-    public void testRemoveChildren() {
-        assertTrue(tree.hasChildren("parent"));
-        tree.removeItem("child");
-        assertFalse(tree.hasChildren("parent"));
-    }
-}
diff --git a/server/tests/src/com/vaadin/tests/server/component/tree/TreeTest.java b/server/tests/src/com/vaadin/tests/server/component/tree/TreeTest.java
new file mode 100644
index 0000000000..16a7091947
--- /dev/null
+++ b/server/tests/src/com/vaadin/tests/server/component/tree/TreeTest.java
@@ -0,0 +1,74 @@
+package com.vaadin.tests.server.component.tree;
+
+import junit.framework.TestCase;
+
+import com.vaadin.data.Container;
+import com.vaadin.data.util.HierarchicalContainer;
+import com.vaadin.data.util.IndexedContainer;
+import com.vaadin.ui.Tree;
+
+public class TreeTest extends TestCase {
+
+    private Tree tree;
+    private Tree tree2;
+    private Tree tree3;
+    private Tree tree4;
+
+    @Override
+    protected void setUp() {
+        tree = new Tree();
+        tree.addItem("parent");
+        tree.addItem("child");
+        tree.setChildrenAllowed("parent", true);
+        tree.setParent("child", "parent");
+
+        tree2 = new Tree("Caption");
+        tree2.addItem("parent");
+        tree2.addItem("child");
+        tree2.setChildrenAllowed("parent", true);
+        tree2.setParent("child", "parent");
+
+        tree3 = new Tree("Caption", null);
+        tree3.addItem("parent");
+        tree3.addItem("child");
+        tree3.setChildrenAllowed("parent", true);
+        tree3.setParent("child", "parent");
+
+        tree4 = new Tree("Caption", new IndexedContainer());
+        tree4.addItem("parent");
+        tree4.addItem("child");
+        tree4.setChildrenAllowed("parent", true);
+        tree4.setParent("child", "parent");
+    }
+
+    public void testRemoveChildren() {
+        assertTrue(tree.hasChildren("parent"));
+        tree.removeItem("child");
+        assertFalse(tree.hasChildren("parent"));
+
+        assertTrue(tree2.hasChildren("parent"));
+        tree2.removeItem("child");
+        assertFalse(tree2.hasChildren("parent"));
+
+        assertTrue(tree3.hasChildren("parent"));
+        tree3.removeItem("child");
+        assertFalse(tree3.hasChildren("parent"));
+
+        assertTrue(tree4.hasChildren("parent"));
+        tree4.removeItem("child");
+        assertFalse(tree4.hasChildren("parent"));
+    }
+
+    public void testContainerTypeIsHierarchical() {
+        assertTrue(HierarchicalContainer.class.isAssignableFrom(tree
+                .getContainerDataSource().getClass()));
+        assertTrue(HierarchicalContainer.class.isAssignableFrom(tree2
+                .getContainerDataSource().getClass()));
+        assertTrue(HierarchicalContainer.class.isAssignableFrom(tree3
+                .getContainerDataSource().getClass()));
+        assertFalse(HierarchicalContainer.class.isAssignableFrom(tree4
+                .getContainerDataSource().getClass()));
+        assertTrue(Container.Hierarchical.class.isAssignableFrom(tree4
+                .getContainerDataSource().getClass()));
+    }
+}