org.eclipse.nebula.widgets.nattable.extension.builder/test/org/eclipse/nebula/widgets/nattable/extension/builder/model/ColumnCategoriesModelAssemblerTest.java

/*******************************************************************************
 * Copyright (c) 2012 Original authors and others.
 * All rights reserved. This program and the accompanying materials
 * are made available under the terms of the Eclipse Public License v1.0
 * which accompanies this distribution, and is available at
 * http://www.eclipse.org/legal/epl-v10.html
 * 
 * Contributors:
 *     Original authors and others - initial API and implementation
 ******************************************************************************/
package org.eclipse.nebula.widgets.nattable.extension.builder.model;

import org.eclipse.nebula.widgets.nattable.columnCategories.ColumnCategoriesModel;
import org.eclipse.nebula.widgets.nattable.columnCategories.Node;
import org.eclipse.nebula.widgets.nattable.extension.builder.model.TableColumn;
import org.eclipse.nebula.widgets.nattable.extension.builder.util.ColumnCategoriesModelAssembler;
import org.junit.Assert;
import org.junit.Test;

public class ColumnCategoriesModelAssemblerTest {

    @Test
    public void assembleModel() throws Exception {
        TableColumn[] columnProps = new TableColumn[] {
                new TableColumn(0, "a"),
                new TableColumn(1, "b").setCategory("C1"),
                new TableColumn(2, "c").setCategory("C1"),
                new TableColumn(3, "d").setCategory("C2"),
                new TableColumn(4, "e").setCategory("C3"),
                new TableColumn(5, "f") };

        ColumnCategoriesModel model = ColumnCategoriesModelAssembler
                .setupColumnCategories(columnProps);

        Node rootCategory = model.getRootCategory().getChildren().get(0);
        Assert.assertEquals(5, rootCategory.getNumberOfChildren());

        Node c1 = rootCategory.getChildren().get(2);
        Assert.assertEquals("C1", c1.getData());
        Assert.assertEquals(2, c1.getChildren().size());

        Node c2 = rootCategory.getChildren().get(3);
        Assert.assertEquals("C2", c2.getData());
        Assert.assertEquals(1, c2.getChildren().size());

        Node c3 = rootCategory.getChildren().get(4);
        Assert.assertEquals("C3", c3.getData());
        Assert.assertEquals(1, c3.getChildren().size());
    }

    @Test
    public void assembleModelAddingAllToRoot() throws Exception {
        TableColumn[] columnProps = new TableColumn[] {
                new TableColumn(0, "a"), new TableColumn(1, "b"),
                new TableColumn(2, "c"), new TableColumn(3, "d"),
                new TableColumn(4, "e"), new TableColumn(5, "f") };

        ColumnCategoriesModel model = ColumnCategoriesModelAssembler
                .setupColumnCategories(columnProps);

        Node rootCategory = model.getRootCategory().getChildren().get(0);
        Assert.assertEquals(6, rootCategory.getNumberOfChildren());
    }

}