写在前面
记录一下我的测试类测接口的代码,尤其是通用代码,不用写第二遍了。
测接口,也可以用postman,但我这次想很多个接口一起测,刚好单元测试写的也很少,就锻炼下自己。
用的是RestTemplate来发请求,顺便封装了下成通用代码。
代码
这个类只要打上@RunWith(SpringJUnit4ClassRunner.class)即可.相当于起到1个客户端的作用,不需要打@SpringBootTest(classes={Ruclass})这个注解
贴一下通用代码
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);
}
/**
*
* @param method
* @param restSuffix 请求url的后缀。如完成请求url是http://localhost:8701/test。则这里是test
* @param params get的传参或post/put的 body中的参数
* @return
*/
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
文章为作者独立观点,不代表股票交易接口观点