Spring中的Junit

Spring中的Junit

 package com.imooc.test.base;

 import org.junit.After;
 import org.junit.Before;
 import org.springframework.beans.BeansException;
 import org.springframework.context.support.ClassPathXmlApplicationContext;
 import org.springframework.util.StringUtils;

 public class UnitTestBase {

     private ClassPathXmlApplicationContext context;

     private String springXmlpath;

     public UnitTestBase() {}

     public UnitTestBase(String springXmlpath) {
         this.springXmlpath = springXmlpath;
     }

     @Before
     public void before() {
         if (StringUtils.isEmpty(springXmlpath)) {
             springXmlpath = "classpath*:spring-*.xml";
         }
         try {
             context = new ClassPathXmlApplicationContext(springXmlpath.split("[,\\s]+"));
             context.start();
         } catch (BeansException e) {
             e.printStackTrace();
         }
     }

     @After
     public void after() {
         context.destroy();
     }

     @SuppressWarnings("unchecked")
     protected <T extends Object> T getBean(String beanId) {
         try {
             return (T)context.getBean(beanId);
         } catch (BeansException e) {
             e.printStackTrace();
             return null;
         }
     }

     protected <T extends Object> T getBean(Class<T> clazz) {
         try {
             return context.getBean(clazz);
         } catch (BeansException e) {
             e.printStackTrace();
             return null;
         }
     }

 }