使用MockMvcBuilders的SpringBootTest独立设置不会加载我的ControllerAdvice尽管设置它

时间:2017-09-17 22:55:56

标签: java spring spring-mvc spring-boot spring-boot-test

我正在创建我的控制器和控制器建议:

测试类:

@RunWith(SpringRunner.class)
@SpringBootTest
public class TestController {

    private MockMvc mockMvc;

    @Mock
    private MyService myService;

    @Autowired
    @InjectMocks
    private MyController myController;

    @Before
    public void setup() {

        MockitoAnnotations.initMocks(this);

        //Build the controller mock handler
        mockMvc = MockMvcBuilders
            .standaloneSetup(MyController.class)
            .setControllerAdvice(new MyControllerAdvice())

            //This also doesn't work
            //.setHandlerExceptionResolvers(createExceptionResolver())
            .build();
    }

    //This also did not work
    private ExceptionHandlerExceptionResolver createExceptionResolver() {
        ExceptionHandlerExceptionResolver exceptionResolver = new ExceptionHandlerExceptionResolver() {
            protected ServletInvocableHandlerMethod getExceptionHandlerMethod(HandlerMethod handlerMethod, Exception exception) {
                Method method = new ExceptionHandlerMethodResolver(MyControllerAdvice.class).resolveMethod(exception);
                return new ServletInvocableHandlerMethod(new MyControllerAdvice(), method);
            }
        };
        exceptionResolver.afterPropertiesSet();
        return exceptionResolver;
    }

    /**
     * Tests passing bad input to see if our exception handler is called.
     */
    @Test
    public void testBadRequest()
    {
        //Make a request object that has a bad input (e.g. bad date string)
        MyRequest request = new MyRequest();

        //Set the request values
        request.setDate( "a" );

        try
        {
            myController.getSomething( request );
        }
        catch (Exception e)
        {
            //It reaches here without ever reaching my controller advice in debugging
            e.printStackTrace();
        }
    }
}

控制器建议:

@EnableWebMvc
@ControllerAdvice
@Component
public class MyControllerAdvice {

    @ExceptionHandler(value = Exception.class)
    public ResponseEntity<String> handleException(HttpServletRequest request, Exception exception) throws Exception
    {
        //This is never called (I'm using a debugger and have a breakpoint here)
        return new ResponseEntity<String>(
            "test",
            HttpStatus.INTERNAL_SERVER_ERROR
        );
    }
}

1 个答案:

答案 0 :(得分:4)

您的示例中有两个问题:

  1. MockMvcBuilders#standaloneSetup()接收Controller个对象作为参数,而不是Class个对象。所以它应该是:

    mockMvc = MockMvcBuilders
            .standaloneSetup(new MyController())
            .setControllerAdvice(new MyControllerAdvice())
            .build();
    
  2. 您正在直接呼叫myController.getSomething( request ),而您应该使用之前构建的mockMvc。直接呼叫未经过修改,因为它未使用TestDispatcherServlet进行处理。以下是mockMvc请求的几个示例:

  3. GET

        mockMvc.perform(get("/testSomething"))
                .andExpect(status().is5xxServerError())
                .andReturn();
    

    POST

        mockMvc.perform(post("/testSomething")
                        .contentType(MediaType.APPLICATION_JSON)
                        .content(json)) //it's JSON string
                .andExpect(status().is5xxServerError())
                .andReturn();