spring中使用mockito

1 mockito介紹和入門

官方：https://github.com/mockito/mockito

入門：

5分鐘瞭解Mockito http://liuzhijun.iteye.com/blog/1512780

Mockito：一個強大的用於 Java 開發的模擬測試框架 http://www.oschina.net/translate/mockito-a-great-mock-framework-for-java-development

2 spring中正常使用mockito

上demo代碼：

 1 @RunWith(SpringJUnit4ClassRunner.class)
 2 @ContextConfiguration(locations = { "classpath:testApplicationContext.xml" })
 3 public class SpringMockitoTest {
 4  
 5     @Mock
 6     private ApiService mockApiService;
 7  
 8     @Before
 9     public void initMocks() {
10         MockitoAnnotations.initMocks(this);
11         when(mockApiService.test()).thenReturn("ok");
12     }
13  
14     @Test
15     public void should_success_when_testApiService() {
16         String result = mockApiService.test();
17         Assert.assertEquals("ok", result);
18     }
19 }
20  
21 @Component
22 public class ApiService {
23  
24     @Autowired
25     private TestApiService testApiService;
26  
27     public String test() {
28         String connect = testApiService.connect();
29         connect += "test";//test本身的業務
30         return connect;
31     }
32 }
33  
34 @Component
35 public class TestApiService {
36     public String connect() {
37         return "error";
38     }
39  
40     public String  findFromDb() {
41         return "db_data";
42     }
43 }

正常使用spring和mockito中，咱們把須要的mock的ApiService給mock掉，可是咱們更想的是把TestApiService中的connect方法mock掉，這樣就能夠測試咱們本身的代碼，也就是ApiService中test方法本身的業務。

3 spring中mock任何容器內對象

上面的demo中，咱們如何mock掉TestApiService中的test方法？

由於TestApiService是spring容器管理的bean，而且ApiService中使用到TestApiService，因此咱們把ApiService中引用的TestApiService替換成咱們的mock對象便可。

Spring框架有個反射工具ReflectionTestUtils，能夠把一個對象中屬性設置爲新值，咱們能夠使用：

ReflectionTestUtils.setField(apiService, "testApiService", spyTestApiService);

把咱們mock的testApiService放到apiService中，這樣apiService調用就是咱們mock的對象了；可是默認spring中apiService對象是代理對象，不能直接把值設置到屬性上，因此咱們本身寫個小的工具類，在最後以下：

ReflectionTestUtils.setField(AopTargetUtils.getTarget(apiService), "testApiService", spyTestApiService);

完整demo：

 1 @RunWith(SpringJUnit4ClassRunner.class)
 2 @ContextConfiguration(locations = { "classpath:testApplicationContext.xml" })
 3 public class SpringMockitoTest {
 4  
 5     @Autowired
 6     private ApiService apiService;
 7     @Mock
 8     private TestApiService spyTestApiService;
 9     @Autowired
10     private TestApiService testApiService;
11  
12     @Before
13     public void initMocks() throws Exception {
14         MockitoAnnotations.initMocks(this);
15         ReflectionTestUtils.setField(AopTargetUtils.getTarget(apiService), "testApiService", spyTestApiService);
16         when(spyTestApiService.connect()).thenReturn("ok");
17     }
18  
19     @After
20     public void clearMocks() throws Exception {
21         ReflectionTestUtils.setField(AopTargetUtils.getTarget(apiService), "testApiService", testApiService);
22     }
23  
24     @Test
25     public void should_success_when_testApiService() {
26         String result = apiService.test();
27         Assert.assertEquals("oktest", result);
28     }
29 }
30  
31 @Component
32 public class ApiService {
33  
34     @Autowired
35     private TestApiService testApiService;
36  
37     public String test() {
38         String connect = testApiService.connect();
39         connect += "test";//test本身的業務
40         return connect;
41     }
42 }
43  
44 @Component
45 public class TestApiService {
46     public String connect() {
47         return "error";
48     }
49  
50     public String  findFromDb() {
51         return "db_data";
52     }
53 }
54  
55 public class AopTargetUtils {
56     /** 
57      * 獲取 目標對象 
58      * @param proxy 代理對象 
59      * @return  
60      * @throws Exception 
61      */
62     public static Object getTarget(Object proxy) throws Exception {  
63         if(!AopUtils.isAopProxy(proxy)) {  
64             return proxy;//不是代理對象  
65         }  
66         if(AopUtils.isJdkDynamicProxy(proxy)) {  
67             return getJdkDynamicProxyTargetObject(proxy);  
68         } else { //cglib  
69             return getCglibProxyTargetObject(proxy);  
70         }  
71     }  
72  
73     private static Object getCglibProxyTargetObject(Object proxy) throws Exception {  
74         Field h = proxy.getClass().getDeclaredField("CGLIB$CALLBACK_0");  
75         h.setAccessible(true);  
76         Object dynamicAdvisedInterceptor = h.get(proxy);  
77         Field advised = dynamicAdvisedInterceptor.getClass().getDeclaredField("advised");  
78         advised.setAccessible(true);  
79         Object target = ((AdvisedSupport)advised.get(dynamicAdvisedInterceptor)).getTargetSource().getTarget();  
80         return target;  
81     }  
82  
83  
84     private static Object getJdkDynamicProxyTargetObject(Object proxy) throws Exception {  
85         Field h = proxy.getClass().getSuperclass().getDeclaredField("h");  
86         h.setAccessible(true);  
87         AopProxy aopProxy = (AopProxy) h.get(proxy);  
88         Field advised = aopProxy.getClass().getDeclaredField("advised");  
89         advised.setAccessible(true);  
90         Object target = ((AdvisedSupport)advised.get(aopProxy)).getTargetSource().getTarget();  
91         return target;  
92     }  
93 }

最後就是注意測試以後要還原現場，把spring對象還原，尤爲在跑maven test的時候，不然可能會影響其餘人的測試。