Spring中的Junit

 1 package com.imooc.test.base;
 2 
 3 import org.junit.After;
 4 import org.junit.Before;
 5 import org.springframework.beans.BeansException;
 6 import org.springframework.context.support.ClassPathXmlApplicationContext;
 7 import org.springframework.util.StringUtils;
 8 
 9 public class UnitTestBase {
10     
11     private ClassPathXmlApplicationContext context;
12     
13     private String springXmlpath;
14     
15     public UnitTestBase() {}
16     
17     public UnitTestBase(String springXmlpath) {
18         this.springXmlpath = springXmlpath;
19     }
20     
21     @Before
22     public void before() {
23         if (StringUtils.isEmpty(springXmlpath)) {
24             springXmlpath = "classpath*:spring-*.xml";
25         }
26         try {
27             context = new ClassPathXmlApplicationContext(springXmlpath.split("[,\\s]+"));
28             context.start();
29         } catch (BeansException e) {
30             e.printStackTrace();
31         }
32     }
33     
34     @After
35     public void after() {
36         context.destroy();
37     }
38     
39     @SuppressWarnings("unchecked")
40     protected <T extends Object> T getBean(String beanId) {
41         try {
42             return (T)context.getBean(beanId);
43         } catch (BeansException e) {
44             e.printStackTrace();
45             return null;
46         }
47     }
48     
49     protected <T extends Object> T getBean(Class<T> clazz) {
50         try {
51             return context.getBean(clazz);
52         } catch (BeansException e) {
53             e.printStackTrace();
54             return null;
55         }
56     }
57 
58 }