menutreepowertest.java

来自「一个自娱自乐的demo 开发环境 apache-tomcat-6.0.16」· Java 代码 · 共 113 行

package com.hb.base.test.dao.ui;

import java.text.SimpleDateFormat;
import java.util.Date;
import java.util.List;

import com.hb.base.dao.organization.EmployeeDao;
import com.hb.base.dao.organization.EmploymentDao;
import com.hb.base.dao.ui.MenuTreeNodeDao;
import com.hb.base.dao.ui.MenuTreePowerDao;
import com.hb.base.domain.organization.Employee;
import com.hb.base.domain.organization.Employment;
import com.hb.base.domain.ui.MenuTreeNode;
import com.hb.base.domain.ui.MenuTreePower;
import com.hb.core.test.dao.HibernateDaoTestCase;

/**
 * 2008/01/15
 * 
 * @author 何 貝
 * 
 */
public class MenuTreePowerTest extends HibernateDaoTestCase {
	private MenuTreePowerDao menuTreePowerDao;
	private MenuTreeNodeDao menuTreeNodeDao;
	private EmployeeDao employeeDao;
	private EmploymentDao employmentDao;

	public void testInsertMenuTreePower() {
		MenuTreeNode root = menuTreeNodeDao.getTreeRootNode();

		Employee employee = employeeDao.getEmployeeByID("00000001");

		String nowDate = new SimpleDateFormat("yyyyMMdd").format(new Date());
		Employment employment = employmentDao.getEmploymentByEmpIDAndValidateDate("00000001", nowDate);

		if (root != null && employee != null && employment != null) {

			MenuTreePower p = new MenuTreePower(employment, root, employee);

			menuTreePowerDao.save(p);

			addMenuTreePowerOnSubNode(root.getSubNodeList(), employment, employee);
		}

	}

	private void addMenuTreePowerOnSubNode(List<MenuTreeNode> subNodeList, Employment employment,
			Employee employee) {
		if (subNodeList != null) {
			for (MenuTreeNode subNode : subNodeList) {
				MenuTreePower p = new MenuTreePower(employment, subNode, employee);

				menuTreePowerDao.save(p);

				addMenuTreePowerOnSubNode(subNode.getSubNodeList(), employment, employee);
			}
		}
	}

	/**
	 * 取得 menuTreePowerDao
	 * 
	 * @return the menuTreePowerDao
	 */
	public MenuTreePowerDao getMenuTreePowerDao() {
		return menuTreePowerDao;
	}

	/**
	 * 设置 menuTreePowerDao
	 * 
	 * @param menuTreePowerDao the menuTreePowerDao to set
	 */
	public void setMenuTreePowerDao(MenuTreePowerDao menuTreePowerDao) {
		this.menuTreePowerDao = menuTreePowerDao;
	}

	/**
	 * 取得 menuTreeNodeDao
	 * 
	 * @return the menuTreeNodeDao
	 */
	public MenuTreeNodeDao getMenuTreeNodeDao() {
		return menuTreeNodeDao;
	}

	/**
	 * 设置 menuTreeNodeDao
	 * 
	 * @param menuTreeNodeDao the menuTreeNodeDao to set
	 */
	public void setMenuTreeNodeDao(MenuTreeNodeDao menuTreeNodeDao) {
		this.menuTreeNodeDao = menuTreeNodeDao;
	}

	public EmployeeDao getEmployeeDao() {
		return employeeDao;
	}

	public void setEmployeeDao(EmployeeDao employeeDao) {
		this.employeeDao = employeeDao;
	}

	public EmploymentDao getEmploymentDao() {
		return employmentDao;
	}

	public void setEmploymentDao(EmploymentDao employmentDao) {
		this.employmentDao = employmentDao;
	}
}