Skip to content

Commit 3065a2f

Browse files
committed
modify unit test.
1 parent e61cfd1 commit 3065a2f

File tree

1 file changed

+7
-0
lines changed

1 file changed

+7
-0
lines changed

hadoop-hdfs-project/hadoop-hdfs-rbf/src/test/java/org/apache/hadoop/hdfs/server/federation/router/async/TestRouterAsyncRpcMultiDestination.java

Lines changed: 7 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -20,6 +20,8 @@
2020
import org.apache.hadoop.conf.Configuration;
2121
import org.apache.hadoop.hdfs.server.federation.MiniRouterDFSCluster;
2222
import org.apache.hadoop.hdfs.server.federation.RouterConfigBuilder;
23+
import org.apache.hadoop.hdfs.server.federation.fairness.RouterAsyncRpcFairnessPolicyController;
24+
import org.apache.hadoop.hdfs.server.federation.fairness.RouterRpcFairnessPolicyController;
2325
import org.apache.hadoop.hdfs.server.federation.router.RBFConfigKeys;
2426
import org.apache.hadoop.hdfs.server.federation.router.TestRouterRpcMultiDestination;
2527
import org.apache.hadoop.security.UserGroupInformation;
@@ -29,6 +31,7 @@
2931
import java.util.concurrent.TimeUnit;
3032

3133
import static org.apache.hadoop.hdfs.server.federation.router.RBFConfigKeys.DFS_ROUTER_ASYNC_RPC_ENABLE_KEY;
34+
import static org.apache.hadoop.hdfs.server.federation.router.RBFConfigKeys.DFS_ROUTER_FAIRNESS_POLICY_CONTROLLER_CLASS;
3235
import static org.apache.hadoop.hdfs.server.federation.router.async.utils.AsyncUtil.syncReturn;
3336
import static org.junit.Assert.assertArrayEquals;
3437

@@ -49,6 +52,10 @@ public static void globalSetUp() throws Exception {
4952
RBFConfigKeys.DN_REPORT_CACHE_EXPIRE, 1, TimeUnit.SECONDS);
5053
// use async router.
5154
routerConf.setBoolean(DFS_ROUTER_ASYNC_RPC_ENABLE_KEY, true);
55+
// Use RouterAsyncRpcFairnessPolicyController as the fairness controller.
56+
routerConf.setClass(DFS_ROUTER_FAIRNESS_POLICY_CONTROLLER_CLASS,
57+
RouterAsyncRpcFairnessPolicyController.class,
58+
RouterRpcFairnessPolicyController.class);
5259
setUp(routerConf);
5360
}
5461

0 commit comments

Comments
 (0)