SpringBoot(四)——http請求工具類
阿新 • • 發佈:2020-08-27
記錄一下單元測試中使用的http工具類,一般只需要留意路徑、請求體、請求頭。至少目前遇到的是這樣。
GetMapping沒有請求體,傳參的話直接用用?和&連線寫在請求路徑裡,例如"?id=10086&age=22"
PostMapping有請求體,以JSON格式傳遞的
1.路徑
例如,[ http://localhost:8080 ],這是基本路徑,一般在公共接口裡定義。後續的RequestMapping和GetMapping等具體路徑在測試方法裡自行拼接。
2.請求頭一般攜帶金鑰,一般在公共接口裡定義,其他測試類直接實現獲取請求頭。
import java.util.HashMap;import java.util.Map;
public interface TestBase { //String IP = "10.8.45.120"; String IP = "localhost"; int PORT = 8103; String BAST_PATH = "http://" + IP + ":" + (PORT); //預設的金鑰 String AUTHORIZATION = "Basic c2FiZXI6c2FV0"; String SCC_AUTH = "bearer eyJ0eXAiOiJKc29uV2ViVG9rZ";default Map<String, String> getHead() { Map<String, String> head = new HashMap<>(); head.put("authorization", AUTHORIZATION); head.put("scc-auth", SCC_AUTH); return head; } }
3.封裝get請求和post請求於一個工具類
import lombok.extern.slf4j.Slf4j; import org.apache.commons.collections4.MapUtils;import org.apache.http.HttpEntity; import org.apache.http.client.config.RequestConfig; import org.apache.http.client.methods.CloseableHttpResponse; import org.apache.http.client.methods.HttpGet; import org.apache.http.impl.client.CloseableHttpClient; import org.apache.http.impl.client.HttpClients; import org.apache.http.util.EntityUtils; import org.springframework.http.HttpHeaders; import org.springframework.http.MediaType; import org.springframework.http.ResponseEntity; import org.springframework.web.client.RestTemplate; import java.io.IOException; import java.util.Map; /** * http 請求工具類 */ @Slf4j public final class HttpRequestUtils { //設定字元編碼 private static final String CHARSET = "UTF-8"; private static RequestConfig defaultRequestConfig = RequestConfig .custom() //設定等待資料超時時間 .setSocketTimeout(30000) //設定連線超時時間 .setConnectTimeout(30000) //設定從連線池獲取連線的等待超時時間 .setConnectionRequestTimeout(30000) //.setStaleConnectionCheckEnabled(true) .build(); //釋放資源,httpResponse為響應流,httpClient為請求客戶端 private static void release(CloseableHttpResponse httpResponse, CloseableHttpClient httpClient) throws IOException { if (httpResponse != null) { httpResponse.close(); } if (httpClient != null) { httpClient.close(); } } //get請求帶引數、帶請求頭 public static String get(String urlWithParams, Map<String, String> header) throws IOException { CloseableHttpClient httpClient = HttpClients.createDefault(); HttpGet httpget = new HttpGet(urlWithParams); if (!MapUtils.isEmpty(header)) { header.forEach(httpget::addHeader); } CloseableHttpResponse response = httpClient.execute(httpget); HttpEntity entity = response.getEntity(); String result = EntityUtils.toString(entity, CHARSET); httpget.releaseConnection(); release(response, httpClient); return result; } public static String get(String urlWithParams) throws IOException { return get(urlWithParams, null); } //傳送post請求,帶json請求體和請求頭 public static String postJson(String url, String json, Map<String, String> headersMap) { RestTemplate restTemplate = new RestTemplate(); HttpHeaders headers = new HttpHeaders(); headers.setContentType(MediaType.APPLICATION_JSON); for (Map.Entry<String, String> entry : headersMap.entrySet()) { headers.add(entry.getKey(),entry.getValue()); } org.springframework.http.HttpEntity<String> request = new org.springframework.http.HttpEntity<>(json, headers); ResponseEntity<String> response = restTemplate.postForEntity(url, request, String.class); return response.getBody(); } }
4.單元測試
import org.junit.jupiter.api.Test; import java.io.IOException; public class GoodsCategoryMapperTest implements TestBase{ private static final String PRE = BAST_PATH + "/goodsCategory"; @Test public void test1() throws IOException { String url = "/getGoodsCategory?category_id=29"; String s = HttpRequestUtils.get(PRE + url, getHead()); System.out.println(s); } @Test public void test2() throws IOException { String url = "/getGoodsCategoryListInfo?page_index=1&page_size=10"; String params = "{\n" + " \"brandId\":\"1000000005\"\n" + "}"; String s = HttpRequestUtils.postJson(PRE + url, params, getHead()); System.out.println(s); } }
5.執行截圖