LeagueTestUtilImpl.java

package ejava.projects.eleague.blimpl;

import java.util.Map;

import javax.persistence.EntityManager;

import ejava.projects.eleague.bl.LeagueTestUtil;
import ejava.util.jpa.DBUtil;

public class LeagueTestUtilImpl implements LeagueTestUtil {
    protected static final String DROP_SCRIPT = "ddl/eLeague_drop.ddl";  
    protected static final String CREATE_SCRIPT = "ddl/eLeague_create.ddl";      
    private EntityManager em;
    
    public void setEntityManager(EntityManager em) {
        this.em = em;
    }
    
    protected DBUtil getDBUtil() {
        DBUtil dbUtil = new DBUtil();
        dbUtil.setEntityManager(em);
        dbUtil.addDropPath(DROP_SCRIPT);
        dbUtil.addCreatePath(CREATE_SCRIPT);
        return dbUtil;
    }
    

    public Map<String, Number> drop() throws RuntimeException {
        DBUtil dbUtil = getDBUtil();
        return dbUtil.dropAll();
    }
    
    public void create(Map<String, Number> sequenceVals) throws RuntimeException {
        DBUtil dbUtil = getDBUtil();
        dbUtil.createAll(sequenceVals);
    }

    @Override
    public void reset() throws RuntimeException {
        Map<String, Number> sequenceVals = drop();
        create(sequenceVals);
    }
}