/*
* 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) 2000 - 2013 Pentaho Corporation, Simba Management Limited and Contributors... All rights reserved.
*/
package org.pentaho.reporting.engine.classic.core.function;
import java.net.URL;
import junit.framework.TestCase;
import org.pentaho.reporting.engine.classic.core.ClassicEngineBoot;
import org.pentaho.reporting.engine.classic.core.MasterReport;
import org.pentaho.reporting.engine.classic.core.RelationalGroup;
import org.pentaho.reporting.engine.classic.core.TableDataFactory;
import org.pentaho.reporting.engine.classic.core.event.ReportEvent;
import org.pentaho.reporting.engine.classic.core.testsupport.DebugReportRunner;
import org.pentaho.reporting.libraries.resourceloader.Resource;
import org.pentaho.reporting.libraries.resourceloader.ResourceManager;
public class TotalItemCountTest extends TestCase
{
private static final int[] GROUPCOUNTS = new int[]{
2, 3, 1, 14, 2, 1
};
private static class TotalItemCountVerifyFunction
extends AbstractFunction
{
private int index;
/**
* Creates an unnamed function. Make sure the name of the function is set using {@link #setName} before the function
* is added to the report's function collection.
*/
public TotalItemCountVerifyFunction()
{
setName("verification");
}
/**
* Receives notification that report generation initializes the current run.
* <p/>
* The event carries a ReportState.Started state. Use this to initialize the report.
*
* @param event The event.
*/
public void reportInitialized(ReportEvent event)
{
index = 0;
}
/**
* Receives notification that a group has finished.
*
* @param event the event.
*/
public void groupFinished(ReportEvent event)
{
if (event.getLevel() >= 0)
{
return;
}
assertCount(event);
index += 1;
}
/**
* Receives notification that a group has started.
*
* @param event the event.
*/
public void groupStarted(ReportEvent event)
{
if (event.getLevel() >= 0)
{
return;
}
assertCount(event);
}
private void assertCount(ReportEvent event)
{
// The itemcount function is only valid within the defined group.
if (FunctionUtilities.getCurrentGroup(event).getName().equals("Continent Group"))
{
// the number of continents in the report1
Number n = (Number) event.getDataRow().get("continent-total-gc");
assertEquals("continent-total-gc: " + index, GROUPCOUNTS[index], n.intValue());
}
// the number of continents in the report1 + default group start
Number n2 = (Number) event.getDataRow().get("total-gc");
assertEquals("total-gc", 23, n2.intValue());
}
public Object getValue()
{
return null;
}
}
public TotalItemCountTest()
{
}
public TotalItemCountTest(final String s)
{
super(s);
}
protected void setUp() throws Exception
{
ClassicEngineBoot.getInstance().start();
}
public void testGroupItemCount() throws Exception
{
final URL url = getClass().getResource("aggregate-function-test.xml");
assertNotNull(url);
final ResourceManager resourceManager = new ResourceManager();
resourceManager.registerDefaults();
final Resource directly = resourceManager.createDirectly(url, MasterReport.class);
final MasterReport report = (MasterReport) directly.getResource();
report.addExpression(new EventMonitorFunction());
report.setDataFactory(new TableDataFactory("default", new AggregateTestDataTableModel()));
report.addExpression(new TotalItemCountVerifyFunction());
final RelationalGroup g = report.getGroupByName("default");
if (g != null)
{
report.removeGroup(g);
}
final TotalItemCountFunction f = new TotalItemCountFunction();
f.setName("continent-total-gc");
f.setGroup("Continent Group");
f.setDependencyLevel(1);
report.addExpression(f);
final TotalItemCountFunction f2 = new TotalItemCountFunction();
f2.setName("total-gc");
f2.setDependencyLevel(1);
report.addExpression(f2);
DebugReportRunner.execGraphics2D(report);
}
public void testSimplifiedRun() throws Exception
{
final MasterReport report = new MasterReport();
report.addExpression(new EventMonitorFunction());
report.addExpression(new TotalItemCountVerifyFunction());
report.setDataFactory(new TableDataFactory("default", new AggregateTestDataTableModel()));
report.setQuery("default");
final RelationalGroup rootGroup = (RelationalGroup) report.getRootGroup();
rootGroup.clearFields();
rootGroup.addField("Continent");
rootGroup.setName("Continent Group");
final TotalItemCountFunction f = new TotalItemCountFunction();
f.setName("continent-total-gc");
f.setGroup("Continent Group");
f.setDependencyLevel(1);
report.addExpression(f);
final TotalItemCountFunction f2 = new TotalItemCountFunction();
f2.setName("total-gc");
f2.setDependencyLevel(1);
report.addExpression(f2);
DebugReportRunner.execGraphics2D(report);
}
}