* has been explicitly set
*/
public Column<T, ?> getHierarchyColumn() {
- return getColumn(getState(false).hierarchyColumnId);
+ return getColumnByInternalId(getState(false).hierarchyColumnId);
}
/**
import com.vaadin.data.TreeData;
import com.vaadin.data.provider.TreeDataProvider;
+import com.vaadin.ui.Grid.Column;
import com.vaadin.ui.TreeGrid;
import com.vaadin.ui.renderers.TextRenderer;
private boolean expandEventFired = false;
private boolean collapseEventFired = false;
+ @Test
public void testChangeRendererOfHierarchyColumn() {
treeGrid.addColumn(Object::toString).setId("foo");
treeGrid.setHierarchyColumn("foo");
Assert.assertFalse("Item not collapsed", treeGrid.isExpanded("Foo"));
Assert.assertTrue("Collapse event not fired", collapseEventFired);
}
+
+ @Test
+ public void testSetAndGetHierarchyColumn() {
+ Column<String, String> column = treeGrid.addColumn(Object::toString)
+ .setId("foo");
+ treeGrid.setHierarchyColumn("foo");
+ Assert.assertEquals("Hierarchy column was not correctly returned",
+ column, treeGrid.getHierarchyColumn());
+ }
+
}