平时几乎每个单元测试开头都要写以下代码，神烦！
1 private Session session; 2 private Transaction transaction; 3 4 @Before 5 public void init(){ 6 session = getSession(); 7 transaction = session.beginTransaction(); 8 } 9 10 @After 11 public void destroy(){ 12 transaction.commit(); 13 session.close(); 14 } |
今天要把它封装一下，做了一个父类：BaseTest，让所有需要上面代码的单元测试都及继承这个父类，具体内容如下：
1 class BaseTest { 2 static Session session; 3 private static Transaction transaction; 4 5 @Before 6 public void init() { 7 session = getSession(); 8 transaction = session.beginTransaction(); 9 System.out.println("[时间]"+new Date()+" session创建成功！事务已开启！"); 10 } 11 12 @After 13 public void destroy() { 14 transaction.commit(); 15 session.close(); 16 System.out.println("[时间]"+new Date()+" 事务已提交！session已销毁！"); 17 } 18 } |
然后子类就可以这么写：
public class QueryByCriteria extends BaseTest { @Test public void QueryTest(){ Criteria criteria = session.createCriteria(ClassRoom.class); List<ClassRoom> list = criteria.list(); list.stream() .forEach(item->System.out.println(item.getClassName())); } } |
最后附上HibernateUtil：
public class HibernateUtil { private static final SessionFactory sessionFactory = buildSessionFactory(); private static SessionFactory buildSessionFactory() { try { Configuration configuration = new Configuration(); configuration.configure("Hibernate.cfg.xml"); // 在下面添加映射，不要写在配置文件中 configuration.addAnnotatedClass(ClassRoom.class); StandardServiceRegistryBuilder ssrb = new StandardServiceRegistryBuilder().applySettings(configuration.getProperties()); return configuration.buildSessionFactory(ssrb.build()); } catch (Throwable ex) { System.err.println("Initial SessionFactory creation failed !" + ex); throw new ExceptionInInitializerError(ex); } } public static SessionFactory getSessionFactory() { return sessionFactory; } public static Session getSession() throws HibernateException { return sessionFactory.openSession(); } } |
