您还未登录! 登录 | 注册 | 帮助  

您的位置: 首页 > 软件测试技术 > 单元测试 > 正文

封装单元测试的重复代码

发表于:2017-01-09 作者:网络转载 来源:

  平时几乎每个单元测试开头都要写以下代码,神烦!
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 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();
}
}