获取token通过JWT拦截

第一种是自己编写一个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;
    }
}

发布了51 篇原创文章 · 获赞 0 · 访问量 728

猜你喜欢

转载自blog.csdn.net/lxy1740/article/details/104308369