hibernate 单元測试框架

hibernate在写数据库配置文件时很的不确定，必须进行必要的測试保证数据库结构的正确性。所以能够应用junit进行測试。

使用junit很easy，eclipse仅仅须要右键项目新建一个junit test case就可以（填写类名和包名）。然后在相应位置写相应的代码执行測试就可以。

以下给出一个经常使用的hibernate測试框架：

package com.atguigu.hibernate.entities;

import java.io.FileInputStream;
import java.io.IOException;
import java.io.InputStream;
import java.sql.Blob;
import java.sql.Connection;
import java.sql.SQLException;
import java.util.Date;

import org.hibernate.Hibernate;
import org.hibernate.Session;
import org.hibernate.SessionFactory;
import org.hibernate.Transaction;
import org.hibernate.cfg.Configuration;
import org.hibernate.jdbc.Work;
import org.hibernate.service.ServiceRegistry;
import org.hibernate.service.ServiceRegistryBuilder;
import org.junit.After;
import org.junit.Before;
import org.junit.Test;

public class HibernateTest {

	private SessionFactory sessionFactory;
	private Session session;
	private Transaction transaction;

	@Before
	public void init(){
		Configuration configuration = new Configuration().configure();
		ServiceRegistry serviceRegistry =
				new ServiceRegistryBuilder().applySettings(configuration.getProperties())
				                            .buildServiceRegistry();
		sessionFactory = configuration.buildSessionFactory(serviceRegistry);

		session = sessionFactory.openSession();
		transaction = session.beginTransaction();
	}

	@After
	public void destroy(){
		transaction.commit();
		session.close();
		sessionFactory.close();
	}

	@Test
	public void testComponent(){
		Worker worker = new Worker();
		Pay pay = new Pay();

		pay.setMonthlyPay(1000);
		pay.setYearPay(80000);
		pay.setVocationWithPay(5);

		worker.setName("ABCD");
		worker.setPay(pay);

		session.save(worker);
	}

	@Test
	public void testBlob() throws Exception{
//		News news = new News();
//		news.setAuthor("cc");
//		news.setContent("CONTENT");
//		news.setDate(new Date());
//		news.setDesc("DESC");
//		news.setTitle("CC");
//
//		InputStream stream = new FileInputStream("Hydrangeas.jpg");
//		Blob image = Hibernate.getLobCreator(session)
//				              .createBlob(stream, stream.available());
//		news.setImage(image);
//
//		session.save(news);

		News news = (News) session.get(News.class, 1);
		Blob image = news.getImage();

		InputStream in = image.getBinaryStream();
		System.out.println(in.available());
	}

	@Test
	public void testPropertyUpdate(){
		News news = (News) session.get(News.class, 1);
		news.setTitle("aaaa"); 

		System.out.println(news.getDesc());
		System.out.println(news.getDate().getClass());
	}

	@Test
	public void testIdGenerator() throws InterruptedException{
		News news = new News("AA", "aa", new java.sql.Date(new Date().getTime()));
		session.save(news); 

//		Thread.sleep(5000);
	}

	@Test
	public void testDynamicUpdate(){
		News news = (News) session.get(News.class, 1);
		news.setAuthor("AABCD");

	}

	@Test
	public void testDoWork(){
		session.doWork(new Work() {

			@Override
			public void execute(Connection connection) throws SQLException {
				System.out.println(connection); 

				//调用存储过程.
			}
		});
	}

	/**
	 * evict: 从 session 缓存中把指定的持久化对象移除
	 */
	@Test
	public void testEvict(){
		News news1 = (News) session.get(News.class, 1);
		News news2 = (News) session.get(News.class, 2);

		news1.setTitle("AA");
		news2.setTitle("BB");

		session.evict(news1);
	}

	/**
	 * delete: 运行删除操作. 仅仅要 OID 和数据表中一条记录相应, 就会准备运行 delete 操作
	 * 若 OID 在数据表中没有相应的记录, 则抛出异常
	 *
	 * 能够通过设置 hibernate 配置文件 hibernate.use_identifier_rollback 为 true,
	 * 使删除对象后, 把其 OID 置为  null
	 */
	@Test
	public void testDelete(){
//		News news = new News();
//		news.setId(11);

		News news = (News) session.get(News.class, 163840);
		session.delete(news); 

		System.out.println(news);
	}

	/**
	 * 注意:
	 * 1. 若 OID 不为 null, 但数据表中还没有和其相应的记录. 会抛出一个异常.
	 * 2. 了解: OID 值等于 id 的 unsaved-value 属性值的对象, 也被觉得是一个游离对象
	 */
	@Test
	public void testSaveOrUpdate(){
		News news = new News("FFF", "fff", new Date());
		news.setId(11);

		session.saveOrUpdate(news);
	}

	/**
	 * update:
	 * 1. 若更新一个持久化对象, 不须要显示的调用 update 方法. 由于在调用 Transaction
	 * 的 commit() 方法时, 会先运行 session 的 flush 方法.
	 * 2. 更新一个游离对象, 须要显式的调用 session 的 update 方法. 能够把一个游离对象
	 * 变为持久化对象
	 *
	 * 须要注意的:
	 * 1. 不管要更新的游离对象和数据表的记录是否一致, 都会发送 UPDATE 语句.
	 *    怎样能让 updat 方法不再盲目的出发 update 语句呢 ? 在 .hbm.xml 文件的 class 节点设置
	 *    select-before-update=true (默觉得 false). 但通常不须要设置该属性.
	 *
	 * 2. 若数据表中没有相应的记录, 但还调用了 update 方法, 会抛出异常
	 *
	 * 3. 当 update() 方法关联一个游离对象时,
	 * 假设在 Session 的缓存中已经存在同样 OID 的持久化对象, 会抛出异常. 由于在 Session 缓存中
	 * 不能有两个 OID 同样的对象!
	 *
	 */
	@Test
	public void testUpdate(){
		News news = (News) session.get(News.class, 1);

		transaction.commit();
		session.close();

//		news.setId(100);

		session = sessionFactory.openSession();
		transaction = session.beginTransaction();

//		news.setAuthor("SUN"); 

		News news2 = (News) session.get(News.class, 1);
		session.update(news);
	}

	/**
	 * get VS load:
	 *
	 * 1. 运行 get 方法: 会马上载入对象.
	 *    运行 load 方法, 若不适用该对象, 则不会马上运行查询操作, 而返回一个代理对象
	 *
	 *    get 是 马上检索, load 是延迟检索.
	 *
	 * 2. load 方法可能会抛出 LazyInitializationException 异常: 在须要初始化
	 * 代理对象之前已经关闭了 Session
	 *
	 * 3. 若数据表中没有相应的记录, Session 也没有被关闭.
	 *    get 返回 null
	 *    load 若不使用该对象的不论什么属性, 没问题; 若须要初始化了, 抛出异常.
	 */
	@Test
	public void testLoad(){

		News news = (News) session.load(News.class, 10);
		System.out.println(news.getClass().getName()); 

//		session.close();
//		System.out.println(news);
	}

	@Test
	public void testGet(){
		News news = (News) session.get(News.class, 1);
//		session.close();
		System.out.println(news);
	}

	/**
	 * persist(): 也会运行 INSERT 操作
	 *
	 * 和 save() 的差别 :
	 * 在调用 persist 方法之前, 若对象已经有 id 了, 则不会运行 INSERT, 而抛出异常
	 */
	@Test
	public void testPersist(){
		News news = new News();
		news.setTitle("EE");
		news.setAuthor("ee");
		news.setDate(new Date());
		news.setId(200); 

		session.persist(news);
	}

	/**
	 * 1. save() 方法
	 * 1). 使一个暂时对象变为持久化对象
	 * 2). 为对象分配 ID.
	 * 3). 在 flush 缓存时会发送一条 INSERT 语句.
	 * 4). 在 save 方法之前的 id 是无效的
	 * 5). 持久化对象的 ID 是不能被改动的!
	 */
	@Test
	public void testSave(){
		News news = new News();
		news.setTitle("CC");
		news.setAuthor("cc");
		news.setDate(new Date());
		news.setId(100); 

		System.out.println(news);

		session.save(news);

		System.out.println(news);
//		news.setId(101);
	}

	/**
	 * clear(): 清理缓存
	 */
	@Test
	public void testClear(){
		News news1 = (News) session.get(News.class, 1);

		session.clear();

		News news2 = (News) session.get(News.class, 1);
	}

	/**
	 * refresh(): 会强制发送 SELECT 语句, 以使 Session 缓存中对象的状态和数据表中相应的记录保持一致!
	 */
	@Test
	public void testRefresh(){
		News news = (News) session.get(News.class, 1);
		System.out.println(news);

		session.refresh(news);
		System.out.println(news);
	}

	/**
	 * flush: 使数据表中的记录和 Session 缓存中的对象的状态保持一致. 为了保持一致, 则可能会发送相应的 SQL 语句.
	 * 1. 在 Transaction 的 commit() 方法中: 先调用 session 的 flush 方法, 再提交事务
	 * 2. flush() 方法会可能会发送 SQL 语句, 但不会提交事务.
	 * 3. 注意: 在未提交事务或显式的调用 session.flush() 方法之前, 也有可能会进行 flush() 操作.
	 * 1). 运行 HQL 或 QBC 查询, 会先进行 flush() 操作, 以得到数据表的最新的记录
	 * 2). 若记录的 ID 是由底层数据库使用自增的方式生成的, 则在调用 save() 方法时, 就会马上发送 INSERT 语句.
	 * 由于 save 方法后, 必须保证对象的 ID 是存在的!
	 */
	@Test
	public void testSessionFlush2(){
		News news = new News("Java", "SUN", new Date());
		session.save(news);
	}

	@Test
	public void testSessionFlush(){
		News news = (News) session.get(News.class, 1);
		news.setAuthor("Oracle");

//		session.flush();
//		System.out.println("flush");

		News news2 = (News) session.createCriteria(News.class).uniqueResult();
		System.out.println(news2);
	}

	@Test
	public void testSessionCache(){

		News news = (News) session.get(News.class, 1);
		System.out.println(news); 

		News news2 = (News) session.get(News.class, 1);
		System.out.println(news2);
	}

}

package com.atguigu.hibernate.entities.n21;

import java.io.FileInputStream;
import java.io.IOException;
import java.io.InputStream;
import java.sql.Blob;
import java.sql.Connection;
import java.sql.SQLException;
import java.util.Date;

import org.hibernate.Hibernate;
import org.hibernate.LazyInitializationException;
import org.hibernate.Session;
import org.hibernate.SessionFactory;
import org.hibernate.Transaction;
import org.hibernate.cfg.Configuration;
import org.hibernate.jdbc.Work;
import org.hibernate.service.ServiceRegistry;
import org.hibernate.service.ServiceRegistryBuilder;
import org.junit.After;
import org.junit.Before;
import org.junit.Test;

public class HibernateTest {

	private SessionFactory sessionFactory;
	private Session session;
	private Transaction transaction;

	@Before
	public void init(){
		Configuration configuration = new Configuration().configure();
		ServiceRegistry serviceRegistry =
				new ServiceRegistryBuilder().applySettings(configuration.getProperties())
				                            .buildServiceRegistry();
		sessionFactory = configuration.buildSessionFactory(serviceRegistry);

		session = sessionFactory.openSession();
		transaction = session.beginTransaction();
	}

	@After
	public void destroy(){
		transaction.commit();
		session.close();
		sessionFactory.close();
	}

	@Test
	public void testDelete(){
		//在不设定级联关系的情况下, 且 1 这一端的对象有 n 的对象在引用, 不能直接删除 1 这一端的对象
		Customer customer = (Customer) session.get(Customer.class, 1);
		session.delete(customer);
	}

	@Test
	public void testUpdate(){
		Order order = (Order) session.get(Order.class, 1);
		order.getCustomer().setCustomerName("AAA");
	}

	@Test
	public void testMany2OneGet(){
		//1. 若查询多的一端的一个对象, 则默认情况下, 仅仅查询了多的一端的对象. 而没有查询关联的
		//1 的那一端的对象!
		Order order = (Order) session.get(Order.class, 1);
		System.out.println(order.getOrderName()); 

		System.out.println(order.getCustomer().getClass().getName());

		session.close();

		//2. 在须要使用到关联的对象时, 才发送相应的 SQL 语句.
		Customer customer = order.getCustomer();
		System.out.println(customer.getCustomerName()); 

		//3. 在查询 Customer 对象时, 由多的一端导航到 1 的一端时,
		//若此时 session 已被关闭, 则默认情况下
		//会发生 LazyInitializationException 异常

		//4. 获取 Order 对象时, 默认情况下, 其关联的 Customer 对象是一个代理对象!

	}

	@Test
	public void testMany2OneSave(){
		Customer customer = new Customer();
		customer.setCustomerName("BB");

		Order order1 = new Order();
		order1.setOrderName("ORDER-3");

		Order order2 = new Order();
		order2.setOrderName("ORDER-4");

		//设定关联关系
		order1.setCustomer(customer);
		order2.setCustomer(customer);

		//运行  save 操作: 先插入 Customer, 再插入 Order, 3 条 INSERT
		//先插入 1 的一端, 再插入 n 的一端, 仅仅有 INSERT 语句.
//		session.save(customer);
//
//		session.save(order1);
//		session.save(order2);

		//先插入 Order, 再插入 Customer. 3 条 INSERT, 2 条 UPDATE
		//先插入 n 的一端, 再插入 1 的一端, 会多出 UPDATE 语句!
		//由于在插入多的一端时, 无法确定 1 的一端的外键值. 所以仅仅能等 1 的一端插入后, 再额外发送 UPDATE 语句.
		//推荐先插入 1 的一端, 后插入 n 的一端
		session.save(order1);
		session.save(order2);

		session.save(customer);
	}

}

package com.atguigu.hibernate.entities.n21.both;

import org.hibernate.Session;
import org.hibernate.SessionFactory;
import org.hibernate.Transaction;
import org.hibernate.cfg.Configuration;
import org.hibernate.service.ServiceRegistry;
import org.hibernate.service.ServiceRegistryBuilder;
import org.junit.After;
import org.junit.Before;
import org.junit.Test;

public class HibernateTest {

	private SessionFactory sessionFactory;
	private Session session;
	private Transaction transaction;

	@Before
	public void init(){
		Configuration configuration = new Configuration().configure();
		ServiceRegistry serviceRegistry =
				new ServiceRegistryBuilder().applySettings(configuration.getProperties())
				                            .buildServiceRegistry();
		sessionFactory = configuration.buildSessionFactory(serviceRegistry);

		session = sessionFactory.openSession();
		transaction = session.beginTransaction();
	}

	@After
	public void destroy(){
		transaction.commit();
		session.close();
		sessionFactory.close();
	}

	@Test
	public void testCascade(){
		Customer customer = (Customer) session.get(Customer.class, 3);
		customer.getOrders().clear();
	}

	@Test
	public void testDelete(){
		//在不设定级联关系的情况下, 且 1 这一端的对象有 n 的对象在引用, 不能直接删除 1 这一端的对象
		Customer customer = (Customer) session.get(Customer.class, 1);
		session.delete(customer);
	}

	@Test
	public void testUpdat2(){
		Customer customer = (Customer) session.get(Customer.class, 1);
		customer.getOrders().iterator().next().setOrderName("GGG");
	}

	@Test
	public void testUpdate(){
		Order order = (Order) session.get(Order.class, 1);
		order.getCustomer().setCustomerName("AAA");
	}

	@Test
	public void testOne2ManyGet(){
		//1. 对 n 的一端的集合使用延迟载入
		Customer customer = (Customer) session.get(Customer.class, 7);
		System.out.println(customer.getCustomerName());
		//2. 返回的多的一端的集合时 Hibernate 内置的集合类型.
		//该类型具有延迟载入和存放代理对象的功能.
		System.out.println(customer.getOrders().getClass()); 

		//session.close();
		//3. 可能会抛出 LazyInitializationException 异常 

		System.out.println(customer.getOrders().size()); 

		//4. 再须要使用集合中元素的时候进行初始化.
	}

	@Test
	public void testMany2OneGet(){
		//1. 若查询多的一端的一个对象, 则默认情况下, 仅仅查询了多的一端的对象. 而没有查询关联的
		//1 的那一端的对象!
		Order order = (Order) session.get(Order.class, 1);
		System.out.println(order.getOrderName()); 

		System.out.println(order.getCustomer().getClass().getName());

		session.close();

		//2. 在须要使用到关联的对象时, 才发送相应的 SQL 语句.
		Customer customer = order.getCustomer();
		System.out.println(customer.getCustomerName()); 

		//3. 在查询 Customer 对象时, 由多的一端导航到 1 的一端时,
		//若此时 session 已被关闭, 则默认情况下
		//会发生 LazyInitializationException 异常

		//4. 获取 Order 对象时, 默认情况下, 其关联的 Customer 对象是一个代理对象!

	}

	@Test
	public void testMany2OneSave(){
		Customer customer = new Customer();
		customer.setCustomerName("AA");

		Order order1 = new Order();
		order1.setOrderName("ORDER-1");

		Order order2 = new Order();
		order2.setOrderName("ORDER-2");

		//设定关联关系
		order1.setCustomer(customer);
		order2.setCustomer(customer);

		customer.getOrders().add(order1);
		customer.getOrders().add(order2);

		//运行  save 操作: 先插入 Customer, 再插入 Order, 3 条 INSERT, 2 条 UPDATE
		//由于 1 的一端和 n 的一端都维护关联关系. 所以会多出 UPDATE
		//能够在 1 的一端的 set 节点指定 inverse=true, 来使 1 的一端放弃维护关联关系!
		//建议设定 set 的 inverse=true, 建议先插入 1 的一端, 后插入多的一端
		//优点是不会多出 UPDATE 语句
		session.save(customer);

//		session.save(order1);
//		session.save(order2);

		//先插入 Order, 再插入 Cusomer, 3 条 INSERT, 4 条 UPDATE
//		session.save(order1);
//		session.save(order2);
//
//		session.save(customer);
	}

}

省略測试时用到的，entity类和相应的配置文件，由于这些文件依据业务的不同要进行相应的变换。