12

我正在使用 Spring MVC 编写一个 Rest 服务。以下是课程大纲：

 @Controller
 public class MyController{

     @RequestMapping(..)
     public void myMethod(...) throws NotAuthorizedException{...}

     @ExceptionHandler(NotAuthorizedException.class)
     @ResponseStatus(value=HttpStatus.UNAUTHORIZED, reason="blah")
     public void handler(...){...}
 }

我已经使用此处发布的设计编写了单元测试。测试基本如下：

@RunWith(SpringJUnit4ClassRunner.class)
@ContextConfiguration(....)
public class mytest{

    MockHttpServletRequest requestMock;
    MockHttpServletResponse responseMock;
    AnnotationMethodHandlerAdapter handlerAdapter;

@Before
public void setUp() {
    requestMock = new MockHttpServletRequest();
    requestMock.setContentType(MediaType.APPLICATION_JSON_VALUE);
    requestMock.addHeader(HttpHeaders.ACCEPT, MediaType.APPLICATION_JSON_VALUE);

    responseMock = new MockHttpServletResponse();

    handlerAdapter = new AnnotationMethodHandlerAdapter();
}

@Test
public void testExceptionHandler(){
    // setup ....
    handlerAdapter.handle(...);

    // verify
    // I would like to do the following
    assertThat(responseMock.getStatus(), is(HttpStatus.UNAUTHORIZED.value()));
}

}

但是，调用 tohandle正在抛出NotAuthorizedException. 我已经读到这是设计使然能够对方法抛出适当的异常进行单元测试，但是我想编写一个自动化测试框架正在适当地处理这个异常并且被测试的类已经适当地实现了处理程序. 有没有办法做到这一点？

请注意，我无法在可以发布的地方访问实际代码。

此外，我仅限于（由于不幸的原因）Spring 3.0.5 或 3.1.2。

4

3 回答 3

8

考虑使用 Spring 3.2 及其mvc-test-framework

import static org.springframework.test.web.servlet.setup.MockMvcBuilders.*;
import static org.springframework.test.web.servlet.request.MockMvcRequestBuilders.*;
import static org.springframework.test.web.servlet.result.MockMvcResultMatchers.*;

@RunWith(SpringJUnit4ClassRunner.class)
@WebAppConfiguration
@ContextConfiguration("file:src/main/webapp/WEB-INF/spring/appServlet/servlet-context.xml")
public class WebMvcTest {

    @Autowired
    private WebApplicationContext wac;

    private MockMvc mockMvc;

    @Before
    public void setup() {
        this.mockMvc = MockMvcBuilders.webAppContextSetup(this.wac).build();
    }

    @Test
    public void getFoo() throws Exception {
        this.mockMvc.perform(
            get("/testx")
            .contentType(MediaType.APPLICATION_JSON)
            .accept(MediaType.APPLICATION_JSON)
            )
            .andExpect(status().isUnauthorized());
    }
}

控制器代码

@Controller
public class MyController {

    public class MyException extends RuntimeException {
    };

    @RequestMapping("/testx")
    public void myMethod() {
        throw new MyException();

    }

    @ExceptionHandler(MyException.class)
    @ResponseStatus(value = HttpStatus.UNAUTHORIZED, reason = "blah")
    public void handler() {
        System.out.println("handler processed");
    }
}

这个“测试”通过得很好。

免责声明：目前我是 Spring MVC 测试的菜鸟，实际上这是我的第一次测试。
更新：感谢德雷克的更正。

于 2013-01-30T20:54:54.210 回答
2

@ControllerAdvice使用而不是注释您的异常处理控制器@Controller。

正如 Boris Treukhov 在将@ExceptionHandler注释添加到引发异常的控制器中的方法时所指出的那样，它将使其工作，但只能从该特定控制器工作。

@ControllerAdvice将允许您的异常处理方法适用于您的整个应用程序，而不仅仅是一个特定的控制器。

于 2014-12-03T22:08:47.920 回答
-2

您可以将@Test 更改为

@Test(expected=NotAuthorizedException.class)

如果内部抛出该异常，这将返回 true，否则返回 false。

这也会使 assertThat() 变得不必要。您可以编写第二个测试来捕获 NotAuthorizedException，然后您可以在该条件下检查 responseMock。

于 2013-01-30T14:31:57.420 回答