big-data/hadoop
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

HDFS-14784. Add more methods to WebHdfsTestUtil to support tests outside of package. Contributed by Chen Zhang.

Browse Source
This commit is contained in:
Inigo Goiri 2019-09-05 21:15:17 -07:00
parent acbea8d976
commit 494d75eb2b
3 changed files with 25 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
hadoop-hdfs-project/hadoop-hdfs/src/test/java/org/apache/hadoop/hdfs/web/TestWebHdfsFileSystemContract.java
View File

@ -410,8 +410,9 @@ public void testResponseCode() throws IOException {
{//test GETHOMEDIRECTORY
final URL url = webhdfs.toUrl(GetOpParam.Op.GETHOMEDIRECTORY, root);
final HttpURLConnection conn = (HttpURLConnection) url.openConnection();
final Map<?, ?> m = WebHdfsTestUtil.connectAndGetJson(
conn, HttpServletResponse.SC_OK);
assertEquals(WebHdfsTestUtil.sendRequest(conn),
HttpServletResponse.SC_OK);
final Map<?, ?> m = WebHdfsTestUtil.getAndParseResponse(conn);
assertEquals(webhdfs.getHomeDirectory().toUri().getPath(),
m.get(Path.class.getSimpleName()));
conn.disconnect();

2
hadoop-hdfs-project/hadoop-hdfs/src/test/java/org/apache/hadoop/hdfs/web/TestWebHdfsTokens.java
View File

@ -348,7 +348,7 @@ op, null, new RenewerParam(null)) {
@Override
Token<DelegationTokenIdentifier> decodeResponse(Map<?, ?> json)
throws IOException {
return JsonUtilClient.toDelegationToken(json);
return WebHdfsTestUtil.convertJsonToDelegationToken(json);
}
}.run();

25
hadoop-hdfs-project/hadoop-hdfs/src/test/java/org/apache/hadoop/hdfs/web/WebHdfsTestUtil.java
View File

@ -25,6 +25,8 @@
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.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 static URL toUrl(final WebHdfsFileSystem webhdfs,
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);
}
}