记录一下我的测试类测接口的代码,尤其是通用代码,不用写第二遍了。 测接口,也可以用postman,但我这次想很多个接口一起测,刚好单元测试写的也很少,就锻炼下自己。 用的是RestTemplate来发请求,顺便封装了下成通用代码。代码
贴一下通用代码
import org.springframework.http.HttpEntity;
import org.springframework.http.HttpHeaders;
import org.springframework.http.HttpMethod;
import org.springframework.http.ResponseEntity;
import org.springframework.util.linkedMultiValueMap;
import org.springframework.util.MultiValueMap;
import org.springframework.web.client.RestTemplate;
import org.springframework.web.util.UriComponentsBuilder;
import java.lang.reflect.Array;
import java.util.Map;
import static org.springframework.http.MediaType.APPLICATION_JSON;
public abstract class CtrlbaseTest {
protected final static String baseUrl = "http://localhost:8701/";
public String baseId = "{{base_id_admin_test1}}";
protected HttpHeaders headers;
private static RestTemplate restTemplate = new RestTemplate();
static{
}
public CtrlbaseTest(String baseId) {
this.baseId = baseId;
headers = new HttpHeaders();
headers.set("base_id",this.baseId);
}
public String getResult(HttpMethod method, String restSuffix, Map params ){
String finalUrl = baseUrl + restSuffix; //最终发起调用的url
HttpEntity httpEntity = null;
if(method==HttpMethod.GET && !params.isEmpty()){
finalUrl = bulidUrl(finalUrl,params);
httpEntity = new HttpEntity(headers);
}
if(method==HttpMethod.POST){
headers.setContentType(APPLICATION_JSON);
httpEntity = new HttpEntity(params,headers);
}
ResponseEntity response = restTemplate.exchange(finalUrl ,
method,
httpEntity,
String.class,params);
System.out.println( String.format("do Http ==> methodType:%s url:%s ",method.name(),finalUrl));
return response.getBody();
}
private String bulidUrl(String originalUrl ,Map params ){
//url需要用占位符的方式
MultiValueMap urlParams = new linkedMultiValueMap<>();
params.forEach((key,value)->{
if(isArray(value)){
//
int length= Array.getLength(value) ;//如果是数组 那么获取数组的长度
for(int i=0;i
然后具体使用就是先启动项目,然后自己写个测试类,继承上面的通用代码,也可以不继承,随便你。
反正如果继承,注意重写构造方法,要修改baseUrl及header的逻辑,如果有要headers的话。我这边封装了一个默认上传1个base_id的请求头。
在这个测试类里写要测试的代码即可发起http请求。注意:
- 这个类只要打上@RunWith(SpringJUnit4ClassRunner.class)即可.相当于起到1个客户端的作用,不需要打@SpringBootTest(classes = {Run.class})这个注解



