org.pivot4j.ui.poi.ExcelExporterIT.java Source code

Introduction

Here is the source code for org.pivot4j.ui.poi.ExcelExporterIT.java

Source

package org.pivot4j.ui.poi;

/*
 * ====================================================================
 * This software is subject to the terms of the Common Public License
 * Agreement, available at the following URL:
 *   http://www.opensource.org/licenses/cpl.html .
 * You must accept the terms of that agreement to use this software.
 * ====================================================================
 */
import static org.hamcrest.CoreMatchers.equalTo;
import static org.hamcrest.CoreMatchers.is;
import static org.hamcrest.CoreMatchers.notNullValue;
import static org.junit.Assert.assertThat;

import java.io.File;
import java.io.FileOutputStream;
import java.io.IOException;
import java.io.OutputStream;

import org.apache.commons.io.IOUtils;
import org.apache.poi.openxml4j.exceptions.InvalidFormatException;
import org.apache.poi.ss.usermodel.Sheet;
import org.apache.poi.ss.usermodel.Workbook;
import org.apache.poi.ss.usermodel.WorkbookFactory;
import org.junit.Test;
import org.pivot4j.AbstractIntegrationTestCase;
import org.pivot4j.PivotModel;
import org.pivot4j.ui.poi.ExcelExporter;
import org.pivot4j.ui.poi.Format;
import org.pivot4j.ui.table.TableRenderer;

public class ExcelExporterIT extends AbstractIntegrationTestCase {

    private String testQuery = "SELECT {[Measures].[Unit Sales], [Measures].[Store Cost], [Measures].[Store Sales]} ON COLUMNS, "
            + "Hierarchize(Union(Union(CrossJoin({[Time].[1997]}, {[Promotion Media].[All Media]}), "
            + "CrossJoin({[Time].[1997]}, [Promotion Media].[All Media].Children)), "
            + "{([Time].[1998], [Promotion Media].[All Media])})) ON ROWS FROM [Sales]";

    private boolean deleteTestFile = true;

    /**
     * @see org.pivot4j.AbstractIntegrationTestCase#setUp()
     */
    @Override
    public void setUp() throws Exception {
        super.setUp();

        PivotModel model = getPivotModel();
        model.setMdx(testQuery);
        model.initialize();
    }

    @Test
    public void testExportHSSF() throws IOException, InvalidFormatException {
        testExport(Format.HSSF, true, true, false, 17, 6);
        testExport(Format.HSSF, false, true, false, 17, 3);
        testExport(Format.HSSF, true, false, false, 16, 5);
        testExport(Format.HSSF, true, false, true, 16, 3);
    }

    @Test
    public void testExportXSSF() throws IOException, InvalidFormatException {
        testExport(Format.XSSF, true, true, false, 17, 6);
        testExport(Format.XSSF, false, true, false, 17, 3);
        testExport(Format.XSSF, true, false, false, 16, 5);
        testExport(Format.XSSF, true, false, true, 16, 3);
    }

    @Test
    public void testExportSXSSF() throws IOException, InvalidFormatException {
        testExport(Format.SXSSF, true, true, false, 17, 6);
        testExport(Format.SXSSF, false, true, false, 17, 3);
        testExport(Format.SXSSF, true, false, false, 16, 5);
        testExport(Format.SXSSF, true, false, true, 16, 3);
    }

    /**
     * @param format
     * @param showParentMember
     * @param showDimensionTitle
     * @param hideSpans
     * @param rows
     * @param mergedRegions
     * @throws IOException
     * @throws InvalidFormatException
     */
    protected void testExport(Format format, boolean showParentMember, boolean showDimensionTitle,
            boolean hideSpans, int rows, int mergedRegions) throws IOException, InvalidFormatException {
        OutputStream out = null;

        File file = File.createTempFile("pivot4j-", "." + format.getExtension());

        if (deleteTestFile) {
            file.deleteOnExit();
        }

        try {
            out = new FileOutputStream(file);

            TableRenderer renderer = new TableRenderer();

            renderer.setShowParentMembers(showParentMember);
            renderer.setShowDimensionTitle(showDimensionTitle);
            renderer.setHideSpans(hideSpans);

            ExcelExporter exporter = new ExcelExporter(out);
            exporter.setFormat(format);

            renderer.render(getPivotModel(), exporter);
        } finally {
            out.flush();
            IOUtils.closeQuietly(out);
        }

        Workbook workbook = WorkbookFactory.create(file);

        assertThat("Workbook cannot be null.", workbook, is(notNullValue()));

        Sheet sheet = workbook.getSheetAt(0);
        assertThat("Worksheet cannot be null.", sheet, is(notNullValue()));

        assertThat("Invalid worksheet name.", sheet.getSheetName(), is(equalTo("Sales")));

        assertThat("Wrong number of rows.", sheet.getLastRowNum(), is(equalTo(rows)));
        assertThat("Wrong number of merged regions.", sheet.getNumMergedRegions(), is(equalTo(mergedRegions)));
    }
}