org.xframium.device.data.ExcelDataProvider.java Source code

Introduction

Here is the source code for org.xframium.device.data.ExcelDataProvider.java

Source

/*******************************************************************************
 * xFramium
 *
 * Copyright 2016 by Moreland Labs, Ltd. (http://www.morelandlabs.com)
 *
 * Some open source application is free software: you can redistribute 
 * it and/or modify it under the terms of the GNU General Public 
 * License as published by the Free Software Foundation, either 
 * version 3 of the License, or (at your option) any later version.
 *  
 * Some open source application 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 General Public License for more details.
 *  
 * You should have received a copy of the GNU General Public License
 * along with xFramium.  If not, see <http://www.gnu.org/licenses/>.
 *
 * @license GPL-3.0+ <http://spdx.org/licenses/GPL-3.0+>
 *******************************************************************************/
/*
 * 
 */
package org.xframium.device.data;

import java.io.BufferedReader;
import java.io.File;
import java.io.FileInputStream;
import java.io.InputStream;
import org.apache.commons.logging.Log;
import org.apache.commons.logging.LogFactory;
import org.apache.poi.xssf.usermodel.XSSFCell;
import org.apache.poi.xssf.usermodel.XSSFRow;
import org.apache.poi.xssf.usermodel.XSSFSheet;
import org.apache.poi.xssf.usermodel.XSSFWorkbook;
import org.xframium.device.DeviceManager;
import org.xframium.device.SimpleDevice;
import org.xframium.spi.Device;

// TODO: Auto-generated Javadoc
/**
 * The Class CSVDataProvider.
 */
public class ExcelDataProvider implements DataProvider {

    /** The log. */
    private Log log = LogFactory.getLog(ExcelDataProvider.class);

    /** The file name. */
    private File fileName;

    /** The tab name. */
    private String tabName;

    /** The resource name. */
    private String resourceName;

    /** The driver type. */
    private DriverType driverType;

    /**
     * Instantiates a new CSV data provider.
     *
     * @param fileName            the file name
     * @param tabName the tab name
     * @param driverType the driver type
     */
    public ExcelDataProvider(File fileName, String tabName, DriverType driverType) {
        this.fileName = fileName;
        this.tabName = tabName;
        this.driverType = driverType;
    }

    /**
     * Instantiates a new CSV data provider.
     *
     * @param resourceName            the resource name
     * @param tabName the tab name
     * @param driverType the driver type
     */
    public ExcelDataProvider(String resourceName, String tabName, DriverType driverType) {
        this.tabName = tabName;
        this.resourceName = resourceName;
        this.driverType = driverType;
    }

    /*
     * (non-Javadoc)
     * 
     * @see com.perfectoMobile.device.data.DataProvider#readData()
     */
    public void readData() {
        if (fileName == null) {
            if (log.isInfoEnabled())
                log.info("Reading Device Data from Resource " + resourceName);

            readData(getClass().getClassLoader().getResourceAsStream(resourceName));
        } else {
            try {
                readData(new FileInputStream(fileName));
            } catch (Exception e) {
                log.fatal("Could mot read from " + fileName, e);
            }
        }
    }

    /**
     * Gets the cell value.
     *
     * @param cell the cell
     * @return the cell value
     */
    private String getCellValue(XSSFCell cell) {
        if (cell != null) {
            switch (cell.getCellType()) {
            case XSSFCell.CELL_TYPE_BLANK:
                return null;
            case XSSFCell.CELL_TYPE_BOOLEAN:
                return String.valueOf(cell.getBooleanCellValue());
            case XSSFCell.CELL_TYPE_NUMERIC:
                return String.valueOf((int) cell.getNumericCellValue());
            case XSSFCell.CELL_TYPE_STRING:
                return cell.getRichStringCellValue().toString();

            }
        }
        return null;
    }

    /**
     * Read data.
     *
     * @param inputStream the input stream
     */
    private void readData(InputStream inputStream) {
        BufferedReader fileReader = null;

        XSSFWorkbook workbook = null;

        try {
            workbook = new XSSFWorkbook(inputStream);
            XSSFSheet sheet = workbook.getSheet(tabName);

            for (int i = 1; i <= sheet.getLastRowNum(); i++) {
                XSSFRow currentRow = sheet.getRow(i);

                if (getCellValue(currentRow.getCell(0)) == null || getCellValue(currentRow.getCell(0)).isEmpty())
                    break;

                String driverName = "";
                switch (driverType) {
                case APPIUM:
                    if (getCellValue(currentRow.getCell(3)).toUpperCase().equals("IOS"))
                        driverName = "IOS";
                    else if (getCellValue(currentRow.getCell(3)).toUpperCase().equals("ANDROID"))
                        driverName = "ANDROID";
                    else
                        throw new IllegalArgumentException("Appium is not supported on the following OS "
                                + getCellValue(currentRow.getCell(3)).toUpperCase());
                    break;

                case PERFECTO:
                    driverName = "PERFECTO";
                    break;

                case WEB:
                    driverName = "WEB";
                    break;
                }

                Device currentDevice = new SimpleDevice(getCellValue(currentRow.getCell(0)),
                        getCellValue(currentRow.getCell(1)), getCellValue(currentRow.getCell(2)),
                        getCellValue(currentRow.getCell(3)), getCellValue(currentRow.getCell(4)),
                        getCellValue(currentRow.getCell(5)), getCellValue(currentRow.getCell(6)),
                        Integer.parseInt(getCellValue(currentRow.getCell(7))), driverName,
                        Boolean.parseBoolean(getCellValue(currentRow.getCell(8))), null);
                if (currentDevice.isActive()) {
                    if (log.isDebugEnabled())
                        log.debug("Extracted: " + currentDevice);

                    DeviceManager.instance().registerDevice(currentDevice);
                }
            }
        } catch (Exception e) {
            log.fatal("Error reading Excel Element File", e);
        } finally {
            try {
                workbook.close();
            } catch (Exception e) {
            }
        }
    }
}