impactiontest.java

OBPM是一个开源

package cn.myapps.core.expimp.imp.action;

import java.io.File;
import java.util.Collection;
import java.util.HashMap;
import java.util.Iterator;
import java.util.LinkedHashMap;
import java.util.Map;

import junit.framework.TestCase;
import cn.myapps.base.dao.PersistenceUtils;
import cn.myapps.base.dao.ValueObject;
import cn.myapps.base.ejb.BaseProcess;
import cn.myapps.constans.Environment;
import cn.myapps.core.deploy.application.ejb.ApplicationProcess;
import cn.myapps.core.deploy.module.ejb.ModuleProcess;
import cn.myapps.core.dynaform.activity.ejb.ActivityProcess;
import cn.myapps.core.dynaform.component.ejb.ComponentProcess;
import cn.myapps.core.dynaform.document.ejb.DocumentProcess;
import cn.myapps.core.dynaform.dts.exp.mappingconfig.ejb.MappingConfigProcess;
import cn.myapps.core.dynaform.form.ejb.FormProcess;
import cn.myapps.core.dynaform.view.ejb.ColumnProcess;
import cn.myapps.core.dynaform.view.ejb.ViewProcess;
import cn.myapps.core.expimp.imp.ejb.ImpObject;
import cn.myapps.core.image.repository.ejb.ImageRepositoryProcess;
import cn.myapps.core.macro.repository.ejb.RepositoryProcess;
import cn.myapps.core.report.query.ejb.QueryProcess;
import cn.myapps.core.report.reportconfig.ejb.ReportConfigProcess;
import cn.myapps.core.report.tablecolumn.ejb.TableColumnProcess;
import cn.myapps.core.resource.ejb.ResourceProcess;
import cn.myapps.core.style.repository.ejb.StyleRepositoryProcess;
import cn.myapps.core.workflow.storage.definition.ejb.BillDefiProcess;
import cn.myapps.util.ProcessFactory;

public class ImpActionTest extends TestCase {
	ImpAction action;
	Environment evt;
	protected void setUp() throws Exception {
		super.setUp();
		action = new ImpAction();
		evt = new Environment(); // 所在路径c:\downloads\export
		evt.setWebcontextBasePath("c:/");
	}

	protected void tearDown() throws Exception {
		super.tearDown();
	}

	/*
	 * Test method for 'cn.myapps.core.expimp.imp.action.ImpAction.doUpload()'
	 */
	public void testDoUpload() throws Exception {
		File dir = new File("c:/downloads/export/");
		File impFile = dir.listFiles()[0];

		action.setImpFile(impFile);
		action.setImpFileFileName(impFile.getName());
		action.doUpload();

		deleteAllDir(impFile);

		doImp();
	}

	/*
	 * Test method for 'cn.myapps.core.expimp.imp.action.ImpAction.doImp()'
	 */
	public void doImp() throws Exception {
		ImpObject _impobj = new ImpObject();
		_impobj.setImpAll(true);
		action.set_impobj(_impobj);
		action.doImp();

		Map values = getValueObjects();
		deleteAll(values);

		String fileDir = action.getImpFilePath();
		deleteAllDir(new File(fileDir));
	}

	private Map getValueObjects() throws Exception {
		Map rtn = new LinkedHashMap();

		//PersistenceUtils.getSessionSignal().sessionSignal++;
		ApplicationProcess appProcess = (ApplicationProcess) ProcessFactory
				.createProcess(ApplicationProcess.class);
		Collection apps = appProcess.doSimpleQuery(null, null);
		assertTrue(apps.size() > 0);

		ModuleProcess mdProcess = (ModuleProcess) ProcessFactory
				.createProcess(ModuleProcess.class);
		Collection module = mdProcess.doSimpleQuery(null, null);
		assertTrue(module.size() > 0);

		StyleRepositoryProcess styleProcess = (StyleRepositoryProcess) ProcessFactory
				.createProcess(StyleRepositoryProcess.class);
		Collection style = styleProcess.doSimpleQuery(null, null);
		assertTrue(style.size() > 0);

		BillDefiProcess bdProcess = (BillDefiProcess) ProcessFactory
				.createProcess(BillDefiProcess.class);
		Collection bill = bdProcess.doSimpleQuery(null, null);
		assertTrue(bill.size() > 0);

		FormProcess fmProcess = (FormProcess) ProcessFactory
				.createProcess(FormProcess.class);
		Collection form = fmProcess.doSimpleQuery(null, null);
		assertTrue(form.size() > 0);

		ViewProcess viewProcess = (ViewProcess) ProcessFactory
				.createProcess(ViewProcess.class);
		Collection view = viewProcess.doSimpleQuery(null, null);
		assertTrue(view.size() > 0);

		ActivityProcess actProcess = (ActivityProcess) ProcessFactory
				.createProcess(ActivityProcess.class);
		Collection activity = actProcess.doSimpleQuery(null, null);
		assertTrue(activity.size() > 0);

		DocumentProcess docProcess = (DocumentProcess) ProcessFactory
				.createProcess(DocumentProcess.class);
		Collection doc = docProcess.doSimpleQuery(null, null);
		assertTrue(doc.size() > 0);

		ColumnProcess columnProcess = (ColumnProcess) ProcessFactory
				.createProcess(ColumnProcess.class);
		Collection column = columnProcess.doSimpleQuery(null, null);
		assertTrue(column.size() > 0);

		QueryProcess qyProcess = (QueryProcess) ProcessFactory
				.createProcess(QueryProcess.class);
		Collection query = qyProcess.doSimpleQuery(null, null);
		assertTrue(query.size() > 0);

		ReportConfigProcess repcgProcess = (ReportConfigProcess) ProcessFactory
				.createProcess(ReportConfigProcess.class);
		Collection reportconfig = repcgProcess.doSimpleQuery(null, null);
		assertTrue(reportconfig.size() > 0);

		MappingConfigProcess mpcgProcess = (MappingConfigProcess) ProcessFactory
				.createProcess(MappingConfigProcess.class);
		Collection mappingconfig = mpcgProcess.doSimpleQuery(null, null);
		assertTrue(mappingconfig.size() > 0);

		TableColumnProcess tcProcess = (TableColumnProcess) ProcessFactory
				.createProcess(TableColumnProcess.class);
		Collection tablecolumn = tcProcess.doSimpleQuery(null, null);
		assertTrue(tablecolumn.size() > 0);

		ComponentProcess cpProcess = (ComponentProcess) ProcessFactory
				.createProcess(ComponentProcess.class);
		Collection comps = cpProcess.doSimpleQuery(null, null);
		assertTrue(comps.size() > 0);

		ResourceProcess resProcess = (ResourceProcess) ProcessFactory
				.createProcess(ResourceProcess.class);
		Collection res = resProcess.doSimpleQuery(null, null);
		assertTrue(comps.size() > 0);

		ImageRepositoryProcess imagerepProcess = (ImageRepositoryProcess) ProcessFactory
				.createProcess(ImageRepositoryProcess.class);
		Collection imagerep = imagerepProcess.doSimpleQuery(null, null);
		assertTrue(comps.size() > 0);

		RepositoryProcess repProcess = (RepositoryProcess) ProcessFactory
				.createProcess(RepositoryProcess.class);
		Collection reps = repProcess.doSimpleQuery(null, null);
		assertTrue(reps.size() > 0);

		rtn.put(getIds(reportconfig), repcgProcess);
		rtn.put(getIds(mappingconfig), mpcgProcess);
		rtn.put(getIds(query), qyProcess);
		rtn.put(getIds(doc), docProcess);
		rtn.put(getIds(activity), actProcess);
		rtn.put(getIds(column), columnProcess);
		rtn.put(getIds(view), viewProcess);
		rtn.put(getIds(bill), bdProcess);
		rtn.put(getIds(form), fmProcess);
		rtn.put(getIds(style), styleProcess);
		rtn.put(getIds(module), mdProcess);
		rtn.put(getIds(apps), appProcess);
		rtn.put(getIds(comps), cpProcess);
		rtn.put(getIds(res), resProcess);
		rtn.put(getIds(reps), repProcess);
		rtn.put(getIds(imagerep), imagerepProcess);

		//PersistenceUtils.getSessionSignal().sessionSignal--;
		return rtn;
	}

	private String[] getIds(Collection colls) {
		String[] rtn = new String[colls.size()];
		int count = 0;
		for (Iterator iter = colls.iterator(); iter.hasNext();) {
			rtn[count] = ((ValueObject) iter.next()).getId();
			count++;
		}
		return rtn;
	}

	private void deleteAll(Map values) throws Exception {
		for (Iterator iter = values.keySet().iterator(); iter.hasNext();) {
			//PersistenceUtils.getSessionSignal().sessionSignal++;
			String[] ids = (String[]) iter.next();
			BaseProcess process = (BaseProcess) values.get(ids);
			process.doRemove(ids);
			//PersistenceUtils.getSessionSignal().sessionSignal--;
		}
	}

	private void deleteAllDir(File file) {
		if (file.isDirectory() && file.getParentFile() != null) {
			File[] files = file.listFiles();
			for (int i = 0; i < files.length; i++) {
				files[i].delete();
			}
		}

		if (file.getParentFile() != null) {
			deleteAllDir(file.getParentFile());
		}
	}
}