寫在前面

記錄一下我的測驗類測介面的代碼，尤其是通用代碼，不用寫第二遍了，
測介面，也可以用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);
    }
    
    /**
     *
     * @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<String,Object> 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<String> 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<String,Object> params ){
        //url需要用占位符的方式
        MultiValueMap<String, String> urlParams = new LinkedMultiValueMap<>();
        params.forEach((key,value)->{
            if(isArray(value)){
                //
                int length= Array.getLength(value) ;//如果是陣列 那么獲取陣列的長度
                for(int i=0;i<length;i++)   //  進行回圈
                {
                    Object o = Array.get(value, i);
                    urlParams.add(key,String.valueOf(o));
                }
            } else{
                urlParams.add(key,String.valueOf(value));
            }
        });
        UriComponentsBuilder builder = UriComponentsBuilder.fromHttpUrl(originalUrl);
        return  builder.queryParams(urlParams).build().encode().toUri().toString();
    }

    /**
     * @return 是否為陣列物件，如果為{@code null} 回傳false
     */
    public static boolean isArray(Object obj) {
        if (null == obj) {
            return false;
        }
        return obj.getClass().isArray();
    }
}

然后具體使用就是先啟動專案，然后自己寫個測驗類，繼承上面的通用代碼，也可以不繼承，隨便你，
反正如果繼承，注意重寫構造方法，要修改baseUrl及header的邏輯，如果有要headers的話，我這邊封裝了一個默認上傳1個base_id的請求頭，
在這個測驗類里寫要測驗的代碼即可發起http請求，注意:

• 這個類只要打上@RunWith(SpringJUnit4ClassRunner.class)即可.相當于起到1個客戶端的作用，不需要打@SpringBootTest(classes = {Run.class})這個注解