封装单元测试的重复代码

　　　平时几乎每个单元测试开头都要写以下代码，神烦！

 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();
    }
}

封装单元测试的重复代码

相关推荐