/*!
* This program is free software; you can redistribute it and/or modify it under the
* terms of the GNU Lesser General Public License, version 2.1 as published by the Free Software
* Foundation.
*
* You should have received a copy of the GNU Lesser General Public License along with this
* program; if not, you can obtain a copy at http://www.gnu.org/licenses/old-licenses/lgpl-2.1.html
* or from the Free Software Foundation, Inc.,
* 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
*
* This program is distributed in the hope that it will be useful, but WITHOUT ANY WARRANTY;
* without even the implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.
* See the GNU Lesser General Public License for more details.
*
* Copyright (c) 2002-2013 Pentaho Corporation.. All rights reserved.
*/
package org.pentaho.reporting.designer.core.util;
import java.util.ArrayList;
import javax.swing.table.AbstractTableModel;
import junit.framework.TestCase;
import org.pentaho.reporting.designer.core.util.table.GroupedTableModel;
import org.pentaho.reporting.designer.core.util.table.GroupingHeader;
import org.pentaho.reporting.designer.core.util.table.GroupingModel;
/**
* Todo: Document Me
*
* @author Thomas Morgner
* @noinspection HardCodedStringLiteral
*/
public class GroupedTableModelTest extends TestCase
{
private static class GroupModel extends AbstractTableModel implements GroupingModel
{
private ArrayList headers;
private ArrayList data;
private GroupModel()
{
headers = new ArrayList();
data = new ArrayList();
}
public void add(final GroupingHeader header, final Object data)
{
headers.add(header);
this.data.add(data);
}
public int getRowCount()
{
return data.size();
}
public int getColumnCount()
{
return 2;
}
public Object getValueAt(final int rowIndex, final int columnIndex)
{
if (columnIndex == 0)
{
return headers.get(rowIndex);
}
return data.get(rowIndex);
}
public GroupingHeader getGroupHeader(final int index)
{
return (GroupingHeader) headers.get(index);
}
public boolean isHeaderRow(final int index)
{
return data.get(index) == null;
}
}
protected static class TestGroupedTableModel extends GroupedTableModel
{
protected TestGroupedTableModel(final GroupingModel parent)
{
super(parent);
}
@Override
public int mapFromModel(final int row)
{
return super.mapFromModel(row);
}
@Override
public int mapToModel(final int row)
{
return super.mapToModel(row);
}
}
public GroupedTableModelTest()
{
}
public GroupedTableModelTest(final String s)
{
super(s);
}
@Override
protected void setUp() throws Exception
{
super.setUp();
}
public void testEmptyMapping()
{
final GroupModel model = new GroupModel();
final TestGroupedTableModel groupedTableModel = new TestGroupedTableModel(model);
assertEquals(groupedTableModel.getRowCount(), model.getRowCount());
}
public void testSimpleMapping()
{
final GroupModel model = new GroupModel();
final GroupingHeader header1 = new GroupingHeader("1");
final GroupingHeader header2 = new GroupingHeader("2");
final GroupingHeader header3 = new GroupingHeader("3");
final GroupingHeader header4 = new GroupingHeader("4");
model.add(header1, null);
model.add(header2, null);
model.add(header2, "2a");
model.add(header2, "2b");
model.add(header2, "2c");
model.add(header3, null);
model.add(header3, "3a");
model.add(header3, "3b");
model.add(header4, null);
final TestGroupedTableModel groupedTableModel = new TestGroupedTableModel(model);
assertEquals(groupedTableModel.getRowCount(), model.getRowCount());
for (int row = 0; row < model.getRowCount(); row++)
{
assertEquals(groupedTableModel.getValueAt(row, 0), model.getValueAt(row, 0));
assertEquals(groupedTableModel.getValueAt(row, 1), model.getValueAt(row, 1));
assertEquals(groupedTableModel.mapFromModel(row), row);
}
}
public void testInnerCollapseMapping()
{
final GroupModel model = new GroupModel();
final GroupingHeader header1 = new GroupingHeader("1");
final GroupingHeader header2 = new GroupingHeader("2");
header2.setCollapsed(true);
final GroupingHeader header3 = new GroupingHeader("3");
final GroupingHeader header4 = new GroupingHeader("4");
model.add(header1, null);
model.add(header1, "1a");
model.add(header2, null);
model.add(header2, "2a");
model.add(header2, "2b");
model.add(header2, "2c");
model.add(header3, null);
model.add(header3, "3a");
model.add(header3, "3b");
model.add(header4, null);
model.add(header4, "4a");
final TestGroupedTableModel groupedTableModel = new TestGroupedTableModel(model);
assertEquals("Row-Count", 8, groupedTableModel.getRowCount());
assertEquals("Header1", header1, groupedTableModel.getValueAt(0, 0));
assertEquals("Header1", header1, groupedTableModel.getValueAt(1, 0));
assertEquals("Header2", header2, groupedTableModel.getValueAt(2, 0));
assertEquals("Header3", header3, groupedTableModel.getValueAt(3, 0));
assertEquals("Header3", header3, groupedTableModel.getValueAt(4, 0));
assertEquals("Header3", header3, groupedTableModel.getValueAt(5, 0));
assertEquals("Header4", header4, groupedTableModel.getValueAt(6, 0));
assertEquals("Header4", header4, groupedTableModel.getValueAt(7, 0));
assertEquals("Back-Mapping", 0, groupedTableModel.mapFromModel(0));
assertEquals("Back-Mapping", 1, groupedTableModel.mapFromModel(1));
assertEquals("Back-Mapping", 2, groupedTableModel.mapFromModel(2));
assertEquals("Back-Mapping", 2, groupedTableModel.mapFromModel(3));
assertEquals("Back-Mapping", 2, groupedTableModel.mapFromModel(4));
assertEquals("Back-Mapping", 2, groupedTableModel.mapFromModel(5));
assertEquals("Back-Mapping", 3, groupedTableModel.mapFromModel(6));
assertEquals("Back-Mapping", 4, groupedTableModel.mapFromModel(7));
assertEquals("Back-Mapping", 5, groupedTableModel.mapFromModel(8));
assertEquals("Back-Mapping", 6, groupedTableModel.mapFromModel(9));
assertEquals("Back-Mapping", 7, groupedTableModel.mapFromModel(10));
}
public void testFirstCollapseMapping()
{
final GroupModel model = new GroupModel();
final GroupingHeader header1 = new GroupingHeader("1");
header1.setCollapsed(true);
final GroupingHeader header2 = new GroupingHeader("2");
final GroupingHeader header3 = new GroupingHeader("3");
final GroupingHeader header4 = new GroupingHeader("4");
model.add(header1, null);
model.add(header1, "1a");
model.add(header2, null);
model.add(header2, "2a");
model.add(header2, "2b");
model.add(header2, "2c");
model.add(header3, null);
model.add(header3, "3a");
model.add(header3, "3b");
model.add(header4, null);
model.add(header4, "4a");
final TestGroupedTableModel groupedTableModel = new TestGroupedTableModel(model);
assertEquals("Row-Count", 10, groupedTableModel.getRowCount());
assertEquals("Header1", header1, groupedTableModel.getValueAt(0, 0));
assertEquals("Header2", header2, groupedTableModel.getValueAt(1, 0));
assertEquals("Header2", header2, groupedTableModel.getValueAt(2, 0));
assertEquals("Header2", header2, groupedTableModel.getValueAt(3, 0));
assertEquals("Header2", header2, groupedTableModel.getValueAt(4, 0));
assertEquals("Header3", header3, groupedTableModel.getValueAt(5, 0));
assertEquals("Header3", header3, groupedTableModel.getValueAt(6, 0));
assertEquals("Header3", header3, groupedTableModel.getValueAt(7, 0));
assertEquals("Header4", header4, groupedTableModel.getValueAt(8, 0));
assertEquals("Header4", header4, groupedTableModel.getValueAt(9, 0));
assertEquals("Back-Mapping", 0, groupedTableModel.mapFromModel(0));
assertEquals("Back-Mapping", 0, groupedTableModel.mapFromModel(1));
assertEquals("Back-Mapping", 1, groupedTableModel.mapFromModel(2));
assertEquals("Back-Mapping", 2, groupedTableModel.mapFromModel(3));
assertEquals("Back-Mapping", 3, groupedTableModel.mapFromModel(4));
assertEquals("Back-Mapping", 4, groupedTableModel.mapFromModel(5));
assertEquals("Back-Mapping", 5, groupedTableModel.mapFromModel(6));
assertEquals("Back-Mapping", 6, groupedTableModel.mapFromModel(7));
assertEquals("Back-Mapping", 7, groupedTableModel.mapFromModel(8));
assertEquals("Back-Mapping", 8, groupedTableModel.mapFromModel(9));
assertEquals("Back-Mapping", 9, groupedTableModel.mapFromModel(10));
}
public void testLastCollapseMapping()
{
final GroupModel model = new GroupModel();
final GroupingHeader header1 = new GroupingHeader("1");
final GroupingHeader header2 = new GroupingHeader("2");
final GroupingHeader header3 = new GroupingHeader("3");
final GroupingHeader header4 = new GroupingHeader("4");
header4.setCollapsed(true);
model.add(header1, null);
model.add(header1, "1a");
model.add(header2, null);
model.add(header2, "2a");
model.add(header2, "2b");
model.add(header2, "2c");
model.add(header3, null);
model.add(header3, "3a");
model.add(header3, "3b");
model.add(header4, null);
model.add(header4, "4a");
model.add(header4, "4b");
final TestGroupedTableModel groupedTableModel = new TestGroupedTableModel(model);
assertEquals("Row-Count", 10, groupedTableModel.getRowCount());
assertEquals("Header1", header1, groupedTableModel.getValueAt(0, 0));
assertEquals("Header1", header1, groupedTableModel.getValueAt(1, 0));
assertEquals("Header2", header2, groupedTableModel.getValueAt(2, 0));
assertEquals("Header2", header2, groupedTableModel.getValueAt(3, 0));
assertEquals("Header2", header2, groupedTableModel.getValueAt(4, 0));
assertEquals("Header2", header2, groupedTableModel.getValueAt(5, 0));
assertEquals("Header3", header3, groupedTableModel.getValueAt(6, 0));
assertEquals("Header3", header3, groupedTableModel.getValueAt(7, 0));
assertEquals("Header3", header3, groupedTableModel.getValueAt(8, 0));
assertEquals("Header4", header4, groupedTableModel.getValueAt(9, 0));
assertEquals("Back-Mapping", 0, groupedTableModel.mapFromModel(0));
assertEquals("Back-Mapping", 1, groupedTableModel.mapFromModel(1));
assertEquals("Back-Mapping", 2, groupedTableModel.mapFromModel(2));
assertEquals("Back-Mapping", 3, groupedTableModel.mapFromModel(3));
assertEquals("Back-Mapping", 4, groupedTableModel.mapFromModel(4));
assertEquals("Back-Mapping", 5, groupedTableModel.mapFromModel(5));
assertEquals("Back-Mapping", 6, groupedTableModel.mapFromModel(6));
assertEquals("Back-Mapping", 7, groupedTableModel.mapFromModel(7));
assertEquals("Back-Mapping", 8, groupedTableModel.mapFromModel(8));
assertEquals("Back-Mapping", 9, groupedTableModel.mapFromModel(9));
assertEquals("Back-Mapping", 9, groupedTableModel.mapFromModel(10));
assertEquals("Back-Mapping", 9, groupedTableModel.mapFromModel(11));
}
public void testAllCollapseMapping()
{
final GroupModel model = new GroupModel();
final GroupingHeader header1 = new GroupingHeader("1");
header1.setCollapsed(true);
final GroupingHeader header2 = new GroupingHeader("2");
header2.setCollapsed(true);
final GroupingHeader header3 = new GroupingHeader("3");
header3.setCollapsed(true);
final GroupingHeader header4 = new GroupingHeader("4");
header4.setCollapsed(true);
model.add(header1, null);
model.add(header1, "1a");
model.add(header2, null);
model.add(header2, "2a");
model.add(header2, "2b");
model.add(header2, "2c");
model.add(header3, null);
model.add(header3, "3a");
model.add(header3, "3b");
model.add(header4, null);
model.add(header4, "4a");
final TestGroupedTableModel groupedTableModel = new TestGroupedTableModel(model);
assertEquals("Row-Count", 4, groupedTableModel.getRowCount());
assertEquals("Header1", header1, groupedTableModel.getValueAt(0, 0));
assertEquals("Header2", header2, groupedTableModel.getValueAt(1, 0));
assertEquals("Header3", header3, groupedTableModel.getValueAt(2, 0));
assertEquals("Header4", header4, groupedTableModel.getValueAt(3, 0));
assertEquals("Back-Mapping", 0, groupedTableModel.mapFromModel(0));
assertEquals("Back-Mapping", 0, groupedTableModel.mapFromModel(1));
assertEquals("Back-Mapping", 1, groupedTableModel.mapFromModel(2));
assertEquals("Back-Mapping", 1, groupedTableModel.mapFromModel(3));
assertEquals("Back-Mapping", 1, groupedTableModel.mapFromModel(4));
assertEquals("Back-Mapping", 1, groupedTableModel.mapFromModel(5));
assertEquals("Back-Mapping", 2, groupedTableModel.mapFromModel(6));
assertEquals("Back-Mapping", 2, groupedTableModel.mapFromModel(7));
assertEquals("Back-Mapping", 2, groupedTableModel.mapFromModel(8));
assertEquals("Back-Mapping", 3, groupedTableModel.mapFromModel(9));
assertEquals("Back-Mapping", 3, groupedTableModel.mapFromModel(10));
}
}