|
@@ -25,6 +25,8 @@ import java.net.URL;
|
|
|
import java.security.PrivilegedExceptionAction;
|
|
|
import java.util.Map;
|
|
|
|
|
|
+import org.apache.hadoop.hdfs.security.token.delegation.DelegationTokenIdentifier;
|
|
|
+import org.apache.hadoop.security.token.Token;
|
|
|
import org.slf4j.Logger;
|
|
|
import org.slf4j.LoggerFactory;
|
|
|
import org.apache.hadoop.conf.Configuration;
|
|
@@ -34,7 +36,6 @@ import org.apache.hadoop.hdfs.DFSConfigKeys;
|
|
|
import org.apache.hadoop.hdfs.web.resources.HttpOpParam;
|
|
|
import org.apache.hadoop.hdfs.web.resources.Param;
|
|
|
import org.apache.hadoop.security.UserGroupInformation;
|
|
|
-import org.junit.Assert;
|
|
|
|
|
|
public class WebHdfsTestUtil {
|
|
|
public static final Logger LOG =
|
|
@@ -87,10 +88,26 @@ public class WebHdfsTestUtil {
|
|
|
return url;
|
|
|
}
|
|
|
|
|
|
- public static Map<?, ?> connectAndGetJson(final HttpURLConnection conn,
|
|
|
- final int expectedResponseCode) throws IOException {
|
|
|
+ public static HttpURLConnection openConnection(URL url, Configuration conf)
|
|
|
+ throws IOException {
|
|
|
+ URLConnectionFactory connectionFactory =
|
|
|
+ URLConnectionFactory.newDefaultURLConnectionFactory(60000, 60000, conf);
|
|
|
+ return (HttpURLConnection) connectionFactory.openConnection(url);
|
|
|
+ }
|
|
|
+
|
|
|
+ public static int sendRequest(final HttpURLConnection conn)
|
|
|
+ throws IOException {
|
|
|
conn.connect();
|
|
|
- Assert.assertEquals(expectedResponseCode, conn.getResponseCode());
|
|
|
+ return conn.getResponseCode();
|
|
|
+ }
|
|
|
+
|
|
|
+ public static Map<?, ?> getAndParseResponse(final HttpURLConnection conn)
|
|
|
+ throws IOException {
|
|
|
return WebHdfsFileSystem.jsonParse(conn, false);
|
|
|
}
|
|
|
+
|
|
|
+ public static Token<DelegationTokenIdentifier> convertJsonToDelegationToken(
|
|
|
+ Map<?, ?> json) throws IOException {
|
|
|
+ return JsonUtilClient.toDelegationToken(json);
|
|
|
+ }
|
|
|
}
|