94
Junit模拟http请求
package cn.trasen.tpmc.tca.controller;
import javax.net.ssl.*;
import java.io.*;
import java.net.HttpURLConnection;
import java.net.MalformedURLException;
import java.net.URL;
import java.security.KeyManagementException;
import java.security.NoSuchAlgorithmException;
import java.security.SecureRandom;
import java.security.cert.X509Certificate;
import java.util.Iterator;
import java.util.Map;
import java.util.logging.Level;
import java.util.logging.Logger;
/**
* Created by yinx on 2017/9/19 0019.
*/
public class HttpUtils {
public static final String UTF8 = "utf-8";
public static final String GBK = "gbk";
public static final String GB2312 = "gb2312";
public static final String ISO88591 = "ISO-8859-1";
public static int READ_TIMEOUT = 30000;
public static int CONNECT_TIMEOUT = 30000;
/**
* post请求数据
*
* @param connectURL
* @param param
* @param charset
* @return
*/
public static String doPost(String connectURL, String param, String charset) {
byte[] bytes = null;
ByteArrayOutputStream byteArrayOut = null;
URL url = null;
HttpURLConnection httpPost = null;
OutputStream out = null;
InputStream in = null;
try {
url = new URL(connectURL);
httpPost = (HttpURLConnection) url.openConnection();
httpPost.setRequestMethod("POST");
httpPost.setDoInput(true);
httpPost.setDoOutput(true);
httpPost.setUseCaches(false);
httpPost.setConnectTimeout(CONNECT_TIMEOUT);
httpPost.setReadTimeout(READ_TIMEOUT);
httpPost.setRequestProperty("Content-Type", "application/x-www-form-urlencoded");
httpPost.connect();
out = httpPost.getOutputStream();
out.write(param.getBytes(charset));
out.flush();
in = httpPost.getInputStream();
byteArrayOut = new ByteArrayOutputStream();
byte[] buf = new byte[512];
int l = 0;
while ((l = in.read(buf)) != -1) {
byteArrayOut.write(buf, 0, l);
}
bytes = byteArrayOut.toByteArray();
return new String(bytes, charset);
} catch (Exception e) {
e.printStackTrace();
} finally {
close(byteArrayOut);
close(out);
close(in);
close(httpPost);
}
return null;
}
public static String doPostSSL(String connectURL, Map<String, String> params, String charset) throws MalformedURLException, IOException, UnsupportedEncodingException {
_ignoreSSL();
return doPost(connectURL, params, charset);
}
/**
* post请求数据
*
* @param connectURL
* @param params
* @param charset
* @return
*/
public static String doPost(String connectURL, Map<String, String> params, String charset) {
String param = "";
if (params != null && !params.isEmpty()) {
StringBuffer paramBuf = new StringBuffer();
for (Iterator<String> it = params.keySet().iterator(); it.hasNext();) {
String key = it.next();
String value = params.get(key);
paramBuf.append("&").append(key).append("=").append(value);
}
param = paramBuf.substring(1);
}
System.out.println("post url:" + connectURL);
System.out.println("post data:" + param);
byte[] bytes = null;
ByteArrayOutputStream byteArrayOut = null;
URL url = null;
HttpURLConnection httpPost = null;
OutputStream out = null;
BufferedInputStream in = null;
try {
url = new URL(connectURL);
httpPost = (HttpURLConnection) url.openConnection();
httpPost.setRequestMethod("POST");
httpPost.setDoInput(true);
httpPost.setDoOutput(true);
httpPost.setUseCaches(false);
httpPost.setConnectTimeout(CONNECT_TIMEOUT);
httpPost.setReadTimeout(READ_TIMEOUT);
httpPost.setRequestProperty("Content-Type", "application/x-www-form-urlencoded");
httpPost.connect();
out = httpPost.getOutputStream();
out.write(param.getBytes(charset));
out.flush();
try {
in = new BufferedInputStream(httpPost.getInputStream());
} catch (IOException e) {
in = new BufferedInputStream(httpPost.getErrorStream());
}
byteArrayOut = new ByteArrayOutputStream();
byte[] buf = new byte[512];
int l = 0;
while ((l = in.read(buf)) != -1) {
byteArrayOut.write(buf, 0, l);
}
bytes = byteArrayOut.toByteArray();
return new String(bytes, charset);
} catch (Exception e) {
e.printStackTrace();
} finally {
close(byteArrayOut);
close(in);
close(out);
close(httpPost);
}
return null;
}
/**
* Get请求数据
*
* @param connectURL
* @param charset
* @return
*/
public static String doGet(String connectURL, String charset) {
byte[] bytes = null;
ByteArrayOutputStream byteArrayOut = null;
URL url = null;
HttpURLConnection httpGet = null;
InputStream in = null;
try {
url = new URL(connectURL);
httpGet = (HttpURLConnection) url.openConnection();
httpGet.setConnectTimeout(CONNECT_TIMEOUT);
httpGet.setReadTimeout(READ_TIMEOUT);
httpGet.connect();
in = httpGet.getInputStream();
byteArrayOut = new ByteArrayOutputStream();
byte[] buf = new byte[512];
int l = 0;
while ((l = in.read(buf)) != -1) {
byteArrayOut.write(buf, 0, l);
}
bytes = byteArrayOut.toByteArray();
return bytes != null ? new String(bytes, charset) : null;
} catch (Exception e) {
e.printStackTrace();
} finally {
close(byteArrayOut);
close(in);
close(httpGet);
}
return null;
}
private static void close(Closeable stream) {
if (stream != null) {
try {
stream.close();
stream = null;
} catch (IOException e) {
e.printStackTrace();
}
}
}
private static void close(HttpURLConnection httpConn){
if(httpConn != null){
httpConn.disconnect();
}
}
private static HostnameVerifier ignoreHostnameVerifier = new HostnameVerifier() {
@Override
public boolean verify(String s, SSLSession sslsession) {
return true;
}
};
/**
* 忽略SSL
*/
private static void _ignoreSSL() {
try {
// Create a trust manager that does not validate certificate chains
TrustManager[] trustAllCerts = new TrustManager[] { new X509TrustManager() {
@Override
public X509Certificate[] getAcceptedIssuers() {
return null;
}
@Override
public void checkClientTrusted(X509Certificate[] certs, String authType) {
}
@Override
public void checkServerTrusted(X509Certificate[] certs, String authType) {
}
} };
// Install the all-trusting trust manager
SSLContext sc = SSLContext.getInstance("TLS");
sc.init(null, trustAllCerts, new SecureRandom());
HttpsURLConnection.setDefaultSSLSocketFactory(sc.getSocketFactory());
HttpsURLConnection.setDefaultHostnameVerifier(ignoreHostnameVerifier);
} catch (KeyManagementException ex) {
Logger.getLogger(HttpUtils.class.getName()).log(Level.SEVERE, null, ex);
} catch (NoSuchAlgorithmException ex) {
Logger.getLogger(HttpUtils.class.getName()).log(Level.SEVERE, null, ex);
}
}
}
最后更新：2017-09-19 10:03:17