testuserdao.java

来自「Hibernate (著译者: 陈天河等)项目开发宝典本书以Hibernate为」· Java 代码 · 共 69 行

package cn.hxex.message.test;

import junit.framework.TestCase;
import cn.hxex.message.dao.IUserDAO;
import cn.hxex.message.exception.MessageDAOException;
import cn.hxex.message.model.User;
import cn.hxex.message.util.DaoFactory;
import cn.hxex.message.util.HibernateUtil;

public class TestUserDAO extends TestCase 
{

	IUserDAO userDAO;
	static String username = "test" + (int)(Math.random() * 100000);
	static String userid;
	
	protected void setUp() throws Exception 
	{
		super.setUp();
		userDAO = (IUserDAO)DaoFactory.getDao( "userDao" );
		assertNotNull( "User DAO is NULL!", userDAO );
		HibernateUtil.beginTransaction();
	}

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

	public void testSaveUser()
	{
		User user = new User();
		user.setName( username );
		user.setPassword( "test" );
		userDAO.saveUser( user );

		assertNotNull( "Save User Success!", user.getId() );
		userid = user.getId();
		
		User u3 = new User();
		u3.setName( username );
		u3.setPassword( "ttt" );
		boolean result = false;
		try
		{
			userDAO.saveUser( u3 );
		}
		catch( MessageDAOException ex )
		{
			result = true;
		}
		assertTrue( "The user Not Exists!", result );
	}
	
	public void testGetUser()
	{
		User user = userDAO.getUser( username );
		assertNotNull( "Get User Success!", user );
	}
	
	public void testGetUserById()
	{
		User u = userDAO.getUserById( userid );
		assertNotNull( "Get(By Id) User Success!", u );		
	}
}