testmessagedao.java

ajax博客系统,该文件内容部分综合网上的资源,自己进行了改进,转载请注明 汪建伟

package cn.hxex.blog.test;

import java.sql.Timestamp;
import java.util.List;

import junit.framework.TestCase;
import cn.hxex.blog.dao.DaoFactory;
import cn.hxex.blog.dao.IMessageDAO;
import cn.hxex.blog.dao.IUserDAO;
import cn.hxex.blog.exception.BlogDAOException;
import cn.hxex.blog.hibernate.HibernateUtil;
import cn.hxex.blog.model.Message;
import cn.hxex.blog.model.User;


public class TestMessageDAO extends TestCase {
	
	IMessageDAO messageDAO;
	IUserDAO userDAO;
	public static String messageId;
	public static String replyMessageId;
	
	protected void setUp() throws Exception {
		super.setUp();
		messageDAO = (IMessageDAO) DaoFactory.getDao("messageDao");
		assertNotNull("messageDAO is Null!", messageDAO);
		userDAO = (IUserDAO) DaoFactory.getDao("userDao");
		assertNotNull("userDAO is Null!", userDAO);
		HibernateUtil.getSessionFactory().getCurrentSession().beginTransaction();
	}

	protected void tearDown() throws Exception {
		super.tearDown();
		HibernateUtil.getSessionFactory().getCurrentSession().getTransaction().commit();
	}

	public void testSaveMessage() {
		User user = userDAO.getUser("test31889");
		Message message = new Message();
		message.setTitle("title");
		message.setUser(user);
		message.setContent("content");
		message.setPubdate(new Timestamp(System.currentTimeMillis()));
		messageDAO.saveMessage(message);
		assertNotNull("add Message Failed!", message.getId());
		messageId = message.getId();
	}

	public void testGetMessages() {
		List list;
		list = messageDAO.getMessages();
		assertTrue( "Get Messages Success!", list.size()>0 );
	}

	public void testGetMessage() {
		Message m = messageDAO.getMessage(messageId);
		assertNotNull("Get Message Failed!", m);
	}

	public void testUpdateMessage() {
		Message m = messageDAO.getMessage(messageId);
		assertNotNull("Couldn't the message!", m);
		m.setTitle("GGG");
		m.setContent("NNN");
		messageDAO.updateMessage(m);
		Message m2 = messageDAO.getMessage(messageId);
		assertEquals("update Success!", m2.getTitle(), "GGG");
		assertEquals("update Success!", m2.getContent(), "NNN");
	}

	public void testDeleteMessage() {
		boolean result = false;
		Message m = messageDAO.getMessage(messageId);
		messageDAO.deleteMessage(m.getId(), m.getUser().getId());
		try {
			messageDAO.getMessage(messageId);
		} catch (Exception e) {
			result = true;
		}
		assertTrue("Delete Failed!", result);
	}

}