第一种是自己编写一个security接口,在这个接口中模拟发出http请求,将响应请求的token返回即可,在测试类中直接调用该接口返回的token,则可以通过jwt验证,完成测试类的编写。
以下为security api接口的代码
package com.siid.webapi.device;
import org.springframework.http.HttpEntity;
import org.springframework.http.HttpHeaders;
import org.springframework.http.MediaType;
import org.springframework.web.client.RestTemplate;
public class SecurityApi {
public String login(String userName, String password) {
String url = "http://test1.siid.com.cn/security/login";
RestTemplate restTemplate = new RestTemplate();
String params = String.format("{ \"userName\":\"%s\", \"password\":\"%s\" }", userName, password);
HttpHeaders headers = new HttpHeaders();
MediaType type = MediaType.parseMediaType("application/json; charset=UTF-8");
headers.setContentType(type);
HttpEntity<String> entity = new HttpEntity<String>(params,headers);
String token = restTemplate.postForObject(url, entity, String.class);
return token;
}
}
在测试类中可以这样获取token,通过jwt验证
....
private String token = "";
@Before
public void setUp() {
List<DevicePositionEntity> positionEntities =positionRepository.findAll();
positionEntity=positionEntities.get(0);
SecurityApi securityApi = new SecurityApi();
token = securityApi.login("admin", "siid0531");
}
@Test
public void getAllByModel() throws Exception{
String url=URI_PREFIX;
MvcResult mvcResult=mvc.perform(MockMvcRequestBuilders.get(url)
.header("Authorization", "Bearer "+token)
.accept(MediaType.APPLICATION_JSON))
.andReturn();
MockHttpServletResponse response = mvcResult.getResponse();
int status=response.getStatus();
String content=response.getContentAsString();
logger.info("测试getAllByModel接口,获取指定型号设备——————");
logger.info("状态码:"+status);
logger.info("测试结果:"+content);
}
每新写一个Controller的接口,定义了一个新的路径后,要在JwtConfig.java中注册该路径,才能通过jwt的验证,如
package com.siid.webapi.device;
import com.siid.webapi.device.filter.JwtFilter;
import org.springframework.boot.web.servlet.FilterRegistrationBean;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;
//import javax.servlet.FilterRegistration;
@Configuration
public class JwtConfig {
@Bean
public FilterRegistrationBean jwtFilter() {
final FilterRegistrationBean registrationBean = new FilterRegistrationBean();
registrationBean.setFilter(new JwtFilter());
registrationBean.addUrlPatterns("/api/camera/*");
registrationBean.addUrlPatterns("/api/device/*");
registrationBean.addUrlPatterns("/api/issue/*");
registrationBean.addUrlPatterns("/api/manhole/*");
registrationBean.addUrlPatterns("/api/position/*");
registrationBean.addUrlPatterns("/api/streetlight/*");
registrationBean.addUrlPatterns("/api/zone/*");
registrationBean.addUrlPatterns("/api/disaster/*");//就是在这注册路径
return registrationBean;
}
}