org.pentaho.di.trans.steps.excelinput.PoiWorkBookTest.java Source code

Introduction

Here is the source code for org.pentaho.di.trans.steps.excelinput.PoiWorkBookTest.java

Source

/*! ******************************************************************************
 *
 * Pentaho Data Integration
 *
 * Copyright (C) 2002-2015 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.excelinput;

import static org.junit.Assert.*;

import java.io.File;
import java.io.IOException;
import java.io.RandomAccessFile;
import java.nio.channels.FileChannel;
import java.nio.channels.FileLock;
import java.util.Date;

import org.junit.Test;
import org.pentaho.di.core.exception.KettleException;
import org.pentaho.di.core.spreadsheet.KCell;
import org.pentaho.di.core.spreadsheet.KCellType;
import org.pentaho.di.core.spreadsheet.KSheet;
import org.pentaho.di.core.spreadsheet.KWorkbook;
import org.apache.commons.io.FileUtils;

public class PoiWorkBookTest {

    @Test
    public void testReadData() throws KettleException {
        readData();
    }

    @Test
    public void testFileDoesNotChange() throws KettleException, IOException {
        File fileBeforeRead = new File("testfiles/sample-file.xlsx");
        readData();
        File fileAfterRead = new File("testfiles/sample-file.xlsx");
        assertTrue(FileUtils.contentEquals(fileBeforeRead, fileAfterRead));
    }

    @Test
    public void testResourceFree() throws Exception {
        FileLock lock = null;
        RandomAccessFile randomAccessFile = null;
        try {
            readData();
            File fileAfterRead = new File("testfiles/sample-file.xlsx");
            randomAccessFile = new RandomAccessFile(fileAfterRead, "rw");
            FileChannel fileChannel = randomAccessFile.getChannel();
            lock = fileChannel.tryLock();
            // check that we could lock file
            assertTrue(lock.isValid());
        } finally {
            if (lock != null) {
                lock.release();
            }
            if (randomAccessFile != null) {
                randomAccessFile.close();
            }
        }
    }

    private void readData() throws KettleException {
        KWorkbook workbook = WorkbookFactory.getWorkbook(SpreadSheetType.POI, "testfiles/sample-file.xlsx", null);
        int numberOfSheets = workbook.getNumberOfSheets();
        assertEquals(3, numberOfSheets);
        KSheet sheet1 = workbook.getSheet(0);
        assertEquals("Sheet1", sheet1.getName());
        sheet1 = workbook.getSheet("Sheet1");
        assertEquals("Sheet1", sheet1.getName());

        assertEquals(5, sheet1.getRows());

        KCell[] row = sheet1.getRow(2);
        assertEquals(KCellType.LABEL, row[1].getType());
        assertEquals("One", row[1].getValue());
        assertEquals(KCellType.DATE, row[2].getType());
        assertEquals(new Date(1283817600000L), row[2].getValue());
        assertEquals(KCellType.NUMBER, row[3].getType());
        assertEquals(Double.valueOf("75"), row[3].getValue());
        assertEquals(KCellType.BOOLEAN, row[4].getType());
        assertEquals(Boolean.valueOf(true), row[4].getValue());
        assertEquals(KCellType.NUMBER_FORMULA, row[5].getType());
        assertEquals(Double.valueOf("75"), row[5].getValue());

        row = sheet1.getRow(3);
        assertEquals(KCellType.LABEL, row[1].getType());
        assertEquals("Two", row[1].getValue());
        assertEquals(KCellType.DATE, row[2].getType());
        assertEquals(new Date(1283904000000L), row[2].getValue());
        assertEquals(KCellType.NUMBER, row[3].getType());
        assertEquals(Double.valueOf("42"), row[3].getValue());
        assertEquals(KCellType.BOOLEAN, row[4].getType());
        assertEquals(Boolean.valueOf(false), row[4].getValue());
        assertEquals(KCellType.NUMBER_FORMULA, row[5].getType());
        assertEquals(Double.valueOf("117"), row[5].getValue());

        row = sheet1.getRow(4);
        assertEquals(KCellType.LABEL, row[1].getType());
        assertEquals("Three", row[1].getValue());
        assertEquals(KCellType.DATE, row[2].getType());
        assertEquals(new Date(1283990400000L), row[2].getValue());
        assertEquals(KCellType.NUMBER, row[3].getType());
        assertEquals(Double.valueOf("93"), row[3].getValue());
        assertEquals(KCellType.BOOLEAN, row[4].getType());
        assertEquals(Boolean.valueOf(true), row[4].getValue());
        assertEquals(KCellType.NUMBER_FORMULA, row[5].getType());
        assertEquals(Double.valueOf("210"), row[5].getValue());

        try {
            sheet1.getRow(5);
            fail("No out of bounds exception thrown when expected");
        } catch (ArrayIndexOutOfBoundsException e) {
            // OK!
        }
        workbook.close();
    }
}