基于Mockito的多层模拟单元测试 小结

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
package com.bas.controller;
import com.bas.TestService;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.beans.factory.annotation.Value;
import org.springframework.web.bind.annotation.RequestMapping;
import org.springframework.web.bind.annotation.RequestMethod;
import org.springframework.web.bind.annotation.RestController;
@RestController
public class TestController {
@Autowired
private TestService service;
@RequestMapping(value = "/test/abc", method = RequestMethod.GET)
public Boolean cfginstFileParse() {
return service.parse("abc.xml");
}
}

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
package com.bas.serviceImpl;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.beans.factory.annotation.Value;
import org.springframework.stereotype.Service;
@Service
public class TestServiceImpl implements TestService {
@Autowired
private TestDao handler;
@Value("${size}")
private Integer size;
@Override
public
Boolean parse(String xpath){
return
handler.parseXml();
}
}

1
2
3
4
5
6
7
8
9
10
11
12
13
14
package com.bas.dao;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.beans.factory.annotation.Value;
import org.springframework.stereotype.Service;
@Service
public class TestDaoImpl implements TestDao {
@Override
public
Boolean parseXml(String xpath){
return
true;
}
}

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
package com.bas.common.util;
import com.bas.dao;
import org.springframework.test.web.servlet.MockMvc;
import org.springframework.test.web.servlet.MvcResult;
import org.springframework.test.web.servlet.request.MockMvcRequestBuilders;
import org.springframework.test.web.servlet.setup.MockMvcBuilders;
import org.springframework.web.context.WebApplicationContext;
import java.lang.reflect.Type;
import java.util.HashMap;
import java.util.List;
import static org.springframework.test.web.servlet.result.MockMvcResultMatchers.status;
import static org.mockito.Mockito.doNothing;
public class MockUtil {
private volatile static MockUtil mockUtil;
private static MockMvc mvc; // process request controller
public MockUtil() {}
public static MockUtil builder(WebApplicationContext wac) {
if(mockUtil == null) {
synchronized (MockUtil.class){
if(mockUtil == null) {
mockUtil = new MockUtil();
mvc = MockMvcBuilders.webAppContextSetup(wac).build();
}
}
}
return mockUtil;
}
/**
* Process the get request.
* @param url controller restful api
* @return return the result from controller
* @throws Exception
*/
public static MvcResult getPerformResult(String url) throws Exception {
return mvc.perform(MockMvcRequestBuilders
.get(url))
.andExpect(status().isOk())
.andReturn();
}
/**
* Process the get request for parsing.
* @param controller the app entrance we want to test
* @param handler the object what is mocked , so it would not run
* @param url controller restful api
* @return return the result from controller
* @throws Exception
*/
public static MvcResult parsePerformResult(Object controller, Dao handler, String url) throws Exception {
// Setup Spring test in standalone mode
MockMvc mockMvc = MockMvcBuilders.standaloneSetup(controller).build();
// stub the method handle , to do nothing when is's called.
doNothing().when(handler).handle(new HashMap<String, List<Object>>());
return mockMvc.perform(MockMvcRequestBuilders
.get(url))
.andExpect(status().isOk())
.andReturn();
}
}

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
package com.bas.test;
import org.junit.After;
import org.junit.Before;
import org.junit.Test;
import org.junit.runner.RunWith;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.boot.test.context.SpringBootTest;
import org.springframework.test.context.junit4.SpringRunner;
import org.springframework.test.web.servlet.MockMvc;
import org.springframework.test.web.servlet.MvcResult;
import org.springframework.test.web.servlet.setup.MockMvcBuilders;
import org.springframework.web.context.WebApplicationContext;
@RunWith(SpringRunner.class)
@SpringBootTest(webEnvironment = SpringBootTest.WebEnvironment.RANDOM_PORT)
public class Test {
@Autowired
private WebApplicationContext wac;
private MockMvc mvc;
@Before
public void before() {
mvc = MockMvcBuilders.webAppContextSetup(wac).build();
}
@Test
public void unitTest() {
try {
MvcResult result = MockUtil.builder(wac).getPerformResult("/test/abc");
System.out.println(result.getResponse().getContentAsString());
} catch (Exception e) {
}
}
}

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
package com.bas.test;
import com.bas.TestController;
import com.bas.TestServiceImpl;
import com.bas.TestDaoImpl;
import org.junit.After;
import org.junit.Before;
import org.junit.Test;
import org.junit.runner.RunWith;
import org.mockito.InjectMocks;
import org.mockito.Mock;
import org.mockito.MockitoAnnotations;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.beans.factory.annotation.Value;
import org.springframework.boot.test.context.SpringBootTest;
import org.springframework.test.context.junit4.SpringRunner;
import org.springframework.test.web.servlet.MvcResult;
@RunWith(SpringRunner.class)
@SpringBootTest(webEnvironment = SpringBootTest.WebEnvironment.RANDOM_PORT)
public class Test {
/* start to mock parse from controller api */
@Before
public void before() {
// initialize mock object , without it would cause in failed mock
MockitoAnnotations.initMocks(this);
}
@Autowired
private TestController testController;
@InjectMocks
@Autowired
TestServiceImpl mockService;
@Mock
TestDaoImpl mockHandler;
@Test
public void mockControllerTest() {
try {
// ReflectionTestUtils.setField(mockService, "size", 1);
// run controller , mock service and handler
MvcResult result = MockUtil.parsePerformResult(testController, mockHandler, "/test/abc");
System.out.println(result.getResponse().getContentAsString());
}catch (Exception e) {
}
}
}