【发布时间】:2018-01-13 13:57:38
【问题描述】:
我是使用 Mockito 对 Spring Rest 控制器进行单元测试的新手。这是我的控制器和测试代码。
@RestController
@RequestMapping("/api/food/customer")
public class CustomerController {
@Autowired
private CustomerService service;
@RequestMapping(method=RequestMethod.POST, produces= MediaType.APPLICATION_JSON_VALUE)
public ResponseEntity<Customer> addCustomer(@RequestBody Customer c){
Logger log = LoggerFactory.getLogger(CustomerController.class.getName());
try {
service.addCustomer(c);
} catch (UserNameException e){
log.error("UserNameException", e);
return new ResponseEntity(HttpStatus.BAD_REQUEST);
} catch (Exception e){
log.error("", e);
return new ResponseEntity(HttpStatus.BAD_REQUEST);
}
log.trace("Customer added: " + c.toString());
return new ResponseEntity(c, HttpStatus.CREATED);
}
}
@RunWith(MockitoJUnitRunner.class)
@WebMvcTest
public class CustomerRestTest {
private MockMvc mockMvc;
@Mock
private CustomerService customerService;
@Mock
private CustomerDao customerDao;
@InjectMocks
private CustomerController customerController;
@Before
public void setup(){
this.mockMvc = MockMvcBuilders.standaloneSetup(customerController).build();
}
@Test
public void testAddDuplicateCustomer() throws Exception {
Customer myCustomer = mock(Customer.class);
when(customerService.addCustomer(myCustomer)).thenThrow(UserNameException.class);
String content = "{\"lastName\" : \"Orr\",\"firstName\" : \"Richard\",\"userName\" : \"Ricky\"}";
RequestBuilder requestBuilder = MockMvcRequestBuilders.post("/api/food/customer").accept(MediaType.APPLICATION_JSON).
content(content).contentType(MediaType.APPLICATION_JSON);
MvcResult result = mockMvc.perform(requestBuilder).andReturn();
MockHttpServletResponse response = result.getResponse();
assertEquals(HttpStatus.BAD_REQUEST.value(), response.getStatus());
}
}
我正在尝试模拟我的服务层并让它在调用 addCustomer 时抛出我的自定义异常。我要返回 HttpStatus.CREATED 而不是 BAD_REQUEST。对于可能正常工作的服务模拟行(带有 thenThrow 的行),我可以做些什么不同的事情?
【问题讨论】:
标签: java spring-mvc spring-boot junit mockito