junitstrategy.java

来自「一个java 代码生成器」· Java 代码 · 共 94 行

/**
 * Copyright (c) 2002, Siddhartha P. Chandurkar [e-mail]
 * All rights reserved.
 * Licensed under the Academic Free License version 1.1
 * See the file LICENSE.TXT for details.
 * LICENSE.txt is located in the directory  <install-directory>\Jenerator
 * of your Jenertaor Installation.
 *
 */
package com.jenerator.transformer;

import com.jenerator.regen.ExtracterException;
import com.jenerator.regen.ReplacerException;
import com.jenerator.struct.ConfigStructs;
import com.jenerator.struct.ejb.EntityStruct;
import com.jenerator.struct.ejb.MDBStruct;
import com.jenerator.struct.ejb.SessionStruct;
import com.jenerator.util.Configurator;
import org.apache.log4j.Logger;

import java.io.File;
import java.util.Vector;

/**
 * JUnitStrategy
 *
 * @author Siddhartha P. Chandurkar
 * @version 0.9.0
 */
public class JUnitStrategy extends TransformStrategy {


    //ATTRIBUTES
    private static Logger log = Logger.getLogger(com.jenerator.transformer.JUnitStrategy.class.getName());

    /**
     * This method overides the super class <code>transform()</code> method
     */
    public void transform() throws TransformFailedException {
        try {
            ConfigStructs configStructs = Configurator.getInstance().getConfigStructs();
            Vector ejbStructs = configStructs.getEjbStructs();
            for (int i = 0; i < ejbStructs.size(); i++) {
                if (ejbStructs.elementAt(i) instanceof com.jenerator.struct.ejb.EntityStruct) {
                    EntityStruct entityStruct = (EntityStruct) ejbStructs.elementAt(i);
                    File file = new File(genDir + entityStruct.getPackageDir(), entityStruct.getEjbClass() + "Test.java");
                    if (file.exists()) {
                        extract = extractManualCode(file);
                        log.debug(extract);
                        processor.process(entityStruct.toXml(), "unit-test-entity.xsl", entityStruct.getEjbClass() + "Test.java", entityStruct.getPackageDir());
                        replaceManualCode(file, extract);
                    } else {
                        processor.process(entityStruct.toXml(), "unit-test-entity.xsl", entityStruct.getEjbClass() + "Test.java", entityStruct.getPackageDir());
                    }
                } else if (ejbStructs.elementAt(i) instanceof com.jenerator.struct.ejb.SessionStruct) {
                    SessionStruct sessionStruct = (SessionStruct) ejbStructs.elementAt(i);
                    File file = new File(genDir + sessionStruct.getPackageDir(), sessionStruct.getEjbClass() + "Test.java");
                    if (file.exists()) {
                        extract = extractManualCode(file);
                        log.debug(extract);
                        processor.process(sessionStruct.toXml(), "unit-test-session.xsl", sessionStruct.getEjbClass() + "Test.java", sessionStruct.getPackageDir());
                        replaceManualCode(file, extract);
                    } else {
                        processor.process(sessionStruct.toXml(), "unit-test-session.xsl", sessionStruct.getEjbClass() + "Test.java", sessionStruct.getPackageDir());
                    }
                } else if (ejbStructs.elementAt(i) instanceof com.jenerator.struct.ejb.MDBStruct) {
                    MDBStruct mdbStruct = (MDBStruct) ejbStructs.elementAt(i);
                    File file = new File(genDir + mdbStruct.getPackageDir(), mdbStruct.getEjbClass() + "Test.java");
                    if (file.exists()) {
                        extract = extractManualCode(file);
                        log.debug(extract);
                        processor.process(mdbStruct.toXml(), "unit-test-mdb.xsl", mdbStruct.getEjbClass() + "Test.java", mdbStruct.getPackageDir());
                        replaceManualCode(file, extract);
                    } else {
                        processor.process(mdbStruct.toXml(), "unit-test-mdb.xsl", mdbStruct.getEjbClass() + "Test.java", mdbStruct.getPackageDir());
                    }
                }

            }
        } catch (NotProcessableException ex) {
            log.fatal(ex);
            throw new TransformFailedException(ex);
        } catch (ExtracterException e) {
            throw new TransformFailedException(e);

        } catch (ReplacerException e) {
            throw new TransformFailedException(e);
        }
        //try

    }

}