/*
* 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 and Contributors...
* All rights reserved.
*/
package org.pentaho.reporting.engine.classic.core.bugs;
import java.io.ByteArrayInputStream;
import java.io.IOException;
import java.net.URL;
import javax.xml.parsers.DocumentBuilder;
import javax.xml.parsers.DocumentBuilderFactory;
import javax.xml.parsers.ParserConfigurationException;
import junit.framework.TestCase;
import org.apache.poi.openxml4j.exceptions.InvalidFormatException;
import org.apache.poi.ss.usermodel.Workbook;
import org.apache.poi.ss.usermodel.WorkbookFactory;
import org.pentaho.reporting.engine.classic.core.ClassicEngineBoot;
import org.pentaho.reporting.engine.classic.core.MasterReport;
import org.pentaho.reporting.engine.classic.core.ReportProcessingException;
import org.pentaho.reporting.engine.classic.core.modules.output.table.xls.ExcelReportUtil;
import org.pentaho.reporting.engine.classic.core.modules.output.table.xml.XmlTableReportUtil;
import org.pentaho.reporting.engine.classic.core.util.NoCloseOutputStream;
import org.pentaho.reporting.libraries.base.util.MemoryByteArrayOutputStream;
import org.pentaho.reporting.libraries.resourceloader.Resource;
import org.pentaho.reporting.libraries.resourceloader.ResourceException;
import org.pentaho.reporting.libraries.resourceloader.ResourceManager;
import org.w3c.dom.Document;
import org.w3c.dom.Element;
import org.w3c.dom.NodeList;
import org.xml.sax.SAXException;
public class Prd3431Test extends TestCase
{
public Prd3431Test()
{
}
public Prd3431Test(final String name)
{
super(name);
}
public void setUp() throws Exception
{
ClassicEngineBoot.getInstance().start();
}
public void testAsXmlOutput()
throws ResourceException, ReportProcessingException, IOException, SAXException, ParserConfigurationException
{
final URL url = getClass().getResource("Prd-3431.prpt");
assertNotNull(url);
final ResourceManager resourceManager = new ResourceManager();
resourceManager.registerDefaults();
final Resource directly = resourceManager.createDirectly(url, MasterReport.class);
final MasterReport report = (MasterReport) directly.getResource();
final MemoryByteArrayOutputStream mbos = new MemoryByteArrayOutputStream();
XmlTableReportUtil.createFlowXML(report, new NoCloseOutputStream(mbos));
final ByteArrayInputStream bin = new ByteArrayInputStream(mbos.getRaw(), 0, mbos.getLength());
final DocumentBuilder documentBuilder = DocumentBuilderFactory.newInstance().newDocumentBuilder();
final Document document = documentBuilder.parse(bin);
final NodeList table = document.getDocumentElement().getElementsByTagName("table");
assertSheetName((Element) table.item(0), "Summary");
assertSheetName((Element) table.item(1), "AuthorPublisher A");
assertSheetName((Element) table.item(2), "AuthorPublisher B");
assertSheetName((Element) table.item(3), "AuthorPublisher C");
}
public void testAsExcelOutput()
throws ResourceException, ReportProcessingException, IOException,
SAXException, ParserConfigurationException, InvalidFormatException
{
final URL url = getClass().getResource("Prd-3431.prpt");
assertNotNull(url);
final ResourceManager resourceManager = new ResourceManager();
resourceManager.registerDefaults();
final Resource directly = resourceManager.createDirectly(url, MasterReport.class);
final MasterReport report = (MasterReport) directly.getResource();
final MemoryByteArrayOutputStream mbos = new MemoryByteArrayOutputStream();
ExcelReportUtil.createXLS(report, new NoCloseOutputStream(mbos));
final ByteArrayInputStream bin = new ByteArrayInputStream(mbos.getRaw(), 0, mbos.getLength());
final Workbook workbook = WorkbookFactory.create(bin);
assertEquals(4, workbook.getNumberOfSheets());
assertEquals("Summary", workbook.getSheetAt(0).getSheetName());
assertEquals("AuthorPublisher A", workbook.getSheetAt(1).getSheetName());
assertEquals("AuthorPublisher B", workbook.getSheetAt(2).getSheetName());
assertEquals("AuthorPublisher C", workbook.getSheetAt(3).getSheetName());
}
private void assertSheetName(final Element n, final String sheetName)
{
assertEquals(sheetName, n.getAttribute("sheet-name"));
}
}