testdemo.java

企业内部邮件系统（初级开发）企业内部邮件系统（初级开发）企业内部邮件系统（初级开发）

package com.myemail.test;

import java.util.Date;
import java.util.HashSet;
import java.util.Iterator;
import java.util.Set;

import junit.framework.TestCase;

import org.hibernate.Session;
import org.hibernate.Transaction;

import com.hibernate.HibernateUtils;
import com.jopo.Message;
import com.jopo.User;

public class testdemo extends TestCase{
	@SuppressWarnings("unchecked")
	public void sendMessage() {
		Session session = HibernateUtils.getSession();

		Message message = new Message();
		message.setTitle("今晚没空");
		message.setContent("。。。。。。。。。。。。");
		message.setSentDate(new Date());

		User from = (User) session.load(User.class, 2);
		message.setSentFrom(from);

		User to1 = (User) session.load(User.class, 1);
		User to2 = (User) session.load(User.class, 3);

		Set tolist = new HashSet();

		tolist.add(to1);
		tolist.add(to2);

		message.setReceiveList( tolist);

		Transaction tx = session.beginTransaction();
		session.save(message);
		tx.commit();

		session.close();
	}

	public void myMessage() {
		Session session = HibernateUtils.getSession();

		User user = (User) session.load(User.class,3);

		Set messages = user.getMyMessages();
		Iterator it = messages.iterator();

		while (it.hasNext()) {
			Message m = (Message) it.next();
			System.out
					.println(m.getTitle() + "   " + m.getSentFrom().getUsername());
		}

		session.close();
	}

	public void receiveList() {
		Session session = HibernateUtils.getSession();

		Message message = (Message) session.load(Message.class, 2);

		Set receivelist = message.getReceiveList();
		Iterator it = receivelist.iterator();

		while (it.hasNext()) {
			User m = (User) it.next();
			System.out
					.println(m.getUsername());
		}

		session.close();
	}
	
	public void mySentMessage(){
		Session session = HibernateUtils.getSession();

		User user = (User) session.load(User.class, 2);

		Set messages = user.getSentMessages();
		Iterator it = messages.iterator();

		while (it.hasNext()) {
			Message m = (Message) it.next();
			System.out
					.println(m.getTitle() + "   " + m.getSentFrom().getUsername());
		}

		session.close();
	}
}