Test with a custom ItemSorter

svn changeset:16932/svn branch:6.5

diff --git a/tests/src/com/vaadin/tests/containers/TestItemSorter.java b/tests/src/com/vaadin/tests/containers/TestItemSorter.java
new file mode 100644 (file)
index 0000000..bf7ce61
--- /dev/null
+++ b/tests/src/com/vaadin/tests/containers/TestItemSorter.java
@@ -0,0 +1,94 @@
+package com.vaadin.tests.containers;
+
+import java.util.Collection;
+import java.util.Comparator;
+
+import com.vaadin.data.Item;
+import com.vaadin.data.util.DefaultItemSorter;
+import com.vaadin.data.util.IndexedContainer;
+import com.vaadin.tests.components.TestBase;
+import com.vaadin.ui.Button;
+import com.vaadin.ui.CheckBox;
+import com.vaadin.ui.Table;
+
+public class TestItemSorter extends TestBase {
+
+    private static final Object BUTTON = "Button";
+    private static final Object CHECKBOX = "CheckBox";
+    private static final Object STRING = "String";
+
+    private Table table;
+    private IndexedContainer container;
+
+    @Override
+    protected void setup() {
+        table = new Table("DefaultItemSorter with custom comparator");
+        container = createContainer();
+        populateContainer(container);
+        container.setItemSorter(new DefaultItemSorter(new Comparator<Object>() {
+
+            public int compare(Object o1, Object o2) {
+                if (o1 instanceof CheckBox && o2 instanceof CheckBox) {
+                    Boolean b1 = ((CheckBox) o1).booleanValue();
+                    return b1.compareTo(((CheckBox) o2).booleanValue());
+                } else if (o1 instanceof Button && o2 instanceof Button) {
+                    String caption1 = ((Button) o1).getCaption().toLowerCase();
+                    String caption2 = ((Button) o2).getCaption().toLowerCase();
+                    return caption1.compareTo(caption2);
+
+                } else if (o1 instanceof String && o2 instanceof String) {
+                    return ((String) o1).toLowerCase().compareTo(
+                            ((String) o2).toLowerCase());
+                }
+
+                return 0;
+
+            }
+        }));
+        table.setContainerDataSource(container);
+
+        addComponent(table);
+
+    }
+
+    private static void populateContainer(IndexedContainer container) {
+        container.removeAllItems();
+        String[] strings = new String[] { "Text 1", "Text 2", "true", "false",
+                "Caption 1", "Caption 2" };
+        for (String s : strings) {
+            Object id = container.addItem();
+            Item item = container.getItem(id);
+            item.getItemProperty(STRING).setValue(s);
+            item.getItemProperty(BUTTON).setValue(new Button(s));
+            item.getItemProperty(CHECKBOX).setValue(
+                    new CheckBox("", s.equals("true")));
+        }
+
+    }
+
+    private static IndexedContainer createContainer() {
+        IndexedContainer ic = new IndexedContainer() {
+            @Override
+            public Collection<?> getSortableContainerPropertyIds() {
+                // Default implementation allows sorting only if the property
+                // type can be cast to Comparable
+                return getContainerPropertyIds();
+            }
+        };
+        ic.addContainerProperty(BUTTON, Button.class, null);
+        ic.addContainerProperty(CHECKBOX, CheckBox.class, null);
+        ic.addContainerProperty(STRING, String.class, null);
+        return ic;
+    }
+
+    @Override
+    protected String getDescription() {
+        return "Test that uses a custom ItemSorter to allow sorting Property types that do not implement Comparable";
+    }
+
+    @Override
+    protected Integer getTicketNumber() {
+        return null;
+    }
+
+}