org.pentaho.di.trans.steps.excelwriter.ExcelWriterStepTest.java Source code

Introduction

Here is the source code for org.pentaho.di.trans.steps.excelwriter.ExcelWriterStepTest.java

Source

/*! ******************************************************************************
 *
 * Pentaho Data Integration
 *
 * Copyright (C) 2002-2013 by Pentaho : http://www.pentaho.com
 *
 *******************************************************************************
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with
 * the License. You may obtain a copy of the License at
 *
 *    http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 *
 ******************************************************************************/

package org.pentaho.di.trans.steps.excelwriter;

import org.apache.commons.io.FileUtils;
import org.apache.poi.hssf.usermodel.HSSFWorkbook;
import org.junit.After;
import org.junit.Before;
import org.junit.Test;
import org.pentaho.di.core.logging.LoggingObjectInterface;
import org.pentaho.di.trans.steps.mock.StepMockHelper;

import java.io.ByteArrayOutputStream;
import java.io.File;
import java.io.FileOutputStream;
import java.io.OutputStream;
import java.io.PrintStream;

import static org.junit.Assert.assertTrue;
import static org.junit.Assert.fail;
import static org.mockito.Matchers.any;
import static org.mockito.Mockito.when;

public class ExcelWriterStepTest {

    private static final String SHEET_NAME = "Sheet1";

    private File xlsFile;
    private HSSFWorkbook wb;
    private StepMockHelper<ExcelWriterStepMeta, ExcelWriterStepData> mockHelper;
    private ExcelWriterStep step;

    @Before
    public void setUp() throws Exception {
        xlsFile = File.createTempFile("testXLSProtect", ".xls");
        wb = createWorkbook(xlsFile);
        mockHelper = new StepMockHelper<ExcelWriterStepMeta, ExcelWriterStepData>("Excel Writer Test",
                ExcelWriterStepMeta.class, ExcelWriterStepData.class);
        when(mockHelper.logChannelInterfaceFactory.create(any(), any(LoggingObjectInterface.class)))
                .thenReturn(mockHelper.logChannelInterface);
        step = new ExcelWriterStep(mockHelper.stepMeta, mockHelper.stepDataInterface, 0, mockHelper.transMeta,
                mockHelper.trans);
    }

    @After
    public void tearDown() {
        FileUtils.deleteQuietly(xlsFile);
    }

    @Test
    public void testProtectSheet() throws Exception {
        step.protectSheet(wb.getSheet(SHEET_NAME), "aa");
        assertTrue(wb.getSheet(SHEET_NAME).getProtect());
    }

    @Test
    public void testMaxSheetNameLength() throws Exception {
        PrintStream err = System.err;
        try {
            ByteArrayOutputStream baos = new ByteArrayOutputStream();
            System.setErr(new PrintStream(baos));

            when(mockHelper.initStepMetaInterface.getSheetname()).thenReturn("12345678901234567890123456789012"); // 32 character
            step.init(mockHelper.initStepMetaInterface, mockHelper.initStepDataInterface);

            String content = baos.toString("UTF-8"); // e.g. ISO-8859-1
            if (!content.contains("12345678901234567890123456789012")) {
                fail();
            }
        } finally {
            System.setErr(err);
        }
    }

    private HSSFWorkbook createWorkbook(File file) throws Exception {
        HSSFWorkbook wb = null;
        OutputStream os = null;
        try {
            os = new FileOutputStream(file);
            wb = new HSSFWorkbook();
            wb.createSheet(SHEET_NAME);
            wb.write(os);
        } finally {
            os.flush();
            os.close();
        }
        return wb;
    }

}