org.bigtester.ate.model.data.CaseDataProcessor.java Source code

Introduction

Here is the source code for org.bigtester.ate.model.data.CaseDataProcessor.java

Source

/*******************************************************************************
 * ATE, Automation Test Engine
 *
 * Copyright 2015, Montreal PROT, or individual contributors as
 * indicated by the @author tags or express copyright attribution
 * statements applied by the authors.  All third-party contributions are
 * distributed under license by Montreal PROT.
 *
 * 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.bigtester.ate.model.data;

import java.io.IOException;
import java.net.MalformedURLException;
import java.sql.SQLException;
import java.util.ArrayList;
import java.util.List;

import org.apache.commons.lang3.ArrayUtils;
import org.bigtester.ate.GlobalUtils;
import org.bigtester.ate.constant.XsdElementConstants;
import org.bigtester.ate.model.page.page.Homepage;
import org.bigtester.ate.model.page.page.Lastpage;
import org.bigtester.ate.model.page.page.RegularPage;
import org.dbunit.DatabaseUnitException;
import org.eclipse.jdt.annotation.Nullable;
import org.springframework.beans.BeansException;
import org.springframework.beans.FatalBeanException;
import org.springframework.beans.factory.config.BeanDefinition;
import org.springframework.beans.factory.config.BeanFactoryPostProcessor;
import org.springframework.beans.factory.config.ConfigurableListableBeanFactory;
import org.springframework.beans.factory.support.BeanDefinitionValidationException;
import org.springframework.core.io.FileSystemResource;
import org.springframework.core.io.Resource;

// TODO: Auto-generated Javadoc
/**
 * This class CaseDataProcessor defines ....
 * @author Peidong Hu
 *
 */

public class CaseDataProcessor implements BeanFactoryPostProcessor {

    /** The all page names. */
    @Nullable
    transient private String[] allPageNames;

    /** The page bean defs. */
    final transient private List<BeanDefinition> pageBeanDefs = new ArrayList<BeanDefinition>();

    /** The case data files. */
    final transient private List<Resource> caseDataFiles = new ArrayList<Resource>();

    /** The db init. */
    @Nullable
    transient private TestDatabaseInitializer dbInit;

    /**
     * {@inheritDoc}
     */
    @Override
    public void postProcessBeanFactory(@Nullable ConfigurableListableBeanFactory beanFactory)
            throws BeansException {
        if (beanFactory == null)
            throw new IllegalStateException("Spring Container initialization error");
        String[] homePageNames = beanFactory.getBeanNamesForType(Homepage.class, true, false);
        String[] lastPageNames = beanFactory.getBeanNamesForType(Lastpage.class, true, false);
        String[] regularPageNames = beanFactory.getBeanNamesForType(RegularPage.class, true, false);

        allPageNames = ArrayUtils.addAll(homePageNames, lastPageNames);
        allPageNames = ArrayUtils.addAll(allPageNames, regularPageNames);

        for (int i = 0; i < getAllPageNames().length; i++) {
            pageBeanDefs.add(beanFactory.getBeanDefinition(getAllPageNames()[i]));
        }

        for (int j = 0; j < pageBeanDefs.size(); j++) {
            if (null != pageBeanDefs.get(j).getPropertyValues()
                    .getPropertyValue(XsdElementConstants.ATTR_BASEPAGEOBJECT_DATAFILE)) {
                caseDataFiles.add(new FileSystemResource((String) pageBeanDefs.get(j).getPropertyValues()
                        .getPropertyValue(XsdElementConstants.ATTR_BASEPAGEOBJECT_DATAFILE).getValue()));
            }
        }

        if (!caseDataFiles.isEmpty()) {
            dbInit = GlobalUtils.findDBInitializer(beanFactory);
            try {
                getDbInit().setInitXmlFiles(caseDataFiles);
            } catch (IOException e) {
                throw new BeanDefinitionValidationException("Page data file in page attribute can't be read!", e);
            }
            try {
                getDbInit().initialize(beanFactory);
            } catch (MalformedURLException | DatabaseUnitException | SQLException e) {
                throw new FatalBeanException("Case database creation error!", e);
            }
        }
    }

    /**
     * @return the allPageNames
     */
    public String[] getAllPageNames() {

        final String[] allPageNames2 = allPageNames;
        if (null == allPageNames2) {
            throw new IllegalStateException("allPageNames field is not correctly populated");
        } else {
            String[] retVal = new String[allPageNames2.length];
            System.arraycopy(allPageNames2, 0, retVal, 0, allPageNames2.length);
            return retVal;
        }

    }

    /**
     * Gets the page bean defs.
     *
     * @return the page bean defs
     */
    public List<BeanDefinition> getPageBeanDefs() {
        return pageBeanDefs;
    }

    /**
     * @return the dbInit
     */
    public TestDatabaseInitializer getDbInit() {
        final TestDatabaseInitializer dbInit2 = dbInit;
        if (null == dbInit2) {
            throw new IllegalStateException("dbinit field is not correctly populated");
        } else {
            return dbInit2;
        }
    }

}