单元测试 mock String authToken = request.getHeader(AUTH_TOKEN)
代码示例
String authToken = request.getHeader(AUTH_TOKEN)
@TEST示例
@Mock
MockHttpServletRequest request = new MockHttpServletRequest();
@Before
public void setUp() {
initMocks(this);
ServletRequestAttributes attributes = (ServletRequestAttributes) RequestContextHolder.getRequestAttributes();
if (attributes == null) {
return;
}
request = (MockHttpServletRequest) attributes.getRequest();
request.addHeader("Auth-Token", "user");
}
@Test
public void test() {
when(request.getHeader(eq("Auth-Token"))).thenReturn("user");
}