/*!
* 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.engine.classic.core.modules.parser.bundle;
import java.io.ByteArrayOutputStream;
import java.io.IOException;
import java.math.BigDecimal;
import junit.framework.TestCase;
import org.pentaho.reporting.engine.classic.core.ClassicEngineBoot;
import org.pentaho.reporting.engine.classic.core.DataFactory;
import org.pentaho.reporting.engine.classic.core.MasterReport;
import org.pentaho.reporting.engine.classic.core.modules.misc.datafactory.sequence.NumberSequence;
import org.pentaho.reporting.engine.classic.core.modules.misc.datafactory.sequence.PerformanceTestSequence;
import org.pentaho.reporting.engine.classic.core.modules.misc.datafactory.sequence.PrinterNamesSequence;
import org.pentaho.reporting.engine.classic.core.modules.misc.datafactory.sequence.Sequence;
import org.pentaho.reporting.engine.classic.core.modules.misc.datafactory.sequence.SequenceDataFactory;
import org.pentaho.reporting.engine.classic.core.modules.misc.datafactory.sequence.SequenceDescription;
import org.pentaho.reporting.engine.classic.core.modules.parser.bundle.writer.BundleWriter;
import org.pentaho.reporting.engine.classic.core.modules.parser.bundle.writer.BundleWriterException;
import org.pentaho.reporting.libraries.base.util.ObjectUtilities;
import org.pentaho.reporting.libraries.repository.ContentIOException;
import org.pentaho.reporting.libraries.resourceloader.Resource;
import org.pentaho.reporting.libraries.resourceloader.ResourceException;
import org.pentaho.reporting.libraries.resourceloader.ResourceManager;
public class SequenceDataSourceWriteTest extends TestCase
{
public SequenceDataSourceWriteTest()
{
}
public SequenceDataSourceWriteTest(final String name)
{
super(name);
}
protected void setUp() throws Exception
{
ClassicEngineBoot.getInstance().start();
}
public void testWriteAndLoad() throws IOException, ContentIOException, BundleWriterException, ResourceException
{
final NumberSequence numberSequence = new NumberSequence();
numberSequence.setParameter("limit", 1234);
numberSequence.setParameter("step", new BigDecimal("1.2"));
numberSequence.setParameter("start", new BigDecimal("4.5"));
numberSequence.setParameter("ascending", Boolean.FALSE);
final PrinterNamesSequence printerSequence = new PrinterNamesSequence();
final PerformanceTestSequence performanceSequence = new PerformanceTestSequence();
performanceSequence.setParameter("limit", 4567);
performanceSequence.setParameter("seed", 1234l);
final PerformanceTestSequence performanceSequence2 = new PerformanceTestSequence();
final SequenceDataFactory sdf = new SequenceDataFactory();
sdf.addSequence("one", numberSequence);
sdf.addSequence("two", printerSequence);
sdf.addSequence("three", performanceSequence);
sdf.addSequence("four", performanceSequence2);
final MasterReport report = new MasterReport();
report.setDataFactory(sdf);
final ByteArrayOutputStream reportDefOutputStream = new ByteArrayOutputStream();
BundleWriter.writeReportToZipStream(report, reportDefOutputStream);
final byte[] reportDefBits = reportDefOutputStream.toByteArray();
final ResourceManager mgr = new ResourceManager();
mgr.registerDefaults();
final Resource directly = mgr.createDirectly(reportDefBits, MasterReport.class);
final MasterReport report2 = (MasterReport) directly.getResource();
final DataFactory dataFactory = report2.getDataFactory();
assertTrue(dataFactory instanceof SequenceDataFactory);
final SequenceDataFactory fac2 = (SequenceDataFactory) dataFactory;
final String[] queryNames = fac2.getQueryNames();
assertTrue(ObjectUtilities.equalArray(queryNames, sdf.getQueryNames()));
for (int i = 0; i < queryNames.length; i++)
{
final String queryName = queryNames[i];
assertEqual(sdf.getSequence(queryName), fac2.getSequence(queryName));
}
}
private void assertEqual (final Sequence s1, final Sequence s2)
{
assertEquals(s1.getClass(), s2.getClass());
final SequenceDescription sd = s1.getSequenceDescription();
final int pc = sd.getParameterCount();
for (int i = 0; i < pc; i++)
{
final String name = sd.getParameterName(i);
assertEquals("Failed at " + name, s1.getParameter(name), s2.getParameter(name));
}
}
}