slfan1989 commented on code in PR #4757:
URL: https://github.com/apache/hadoop/pull/4757#discussion_r952023587
##########
hadoop-yarn-project/hadoop-yarn/hadoop-yarn-server/hadoop-yarn-server-router/src/test/java/org/apache/hadoop/yarn/server/router/webapp/TestFederationInterceptorREST.java:
##########
@@ -949,4 +959,73 @@ public void testGetAppQueue() throws IOException,
InterruptedException, YarnExce
Assert.assertNotNull(queue);
Assert.assertEquals(queueName, queue.getQueue());
}
+
+ @Test
+ public void testGetAppStatistics() throws IOException, InterruptedException,
YarnException {
+ AppState appStateRUNNING = new
AppState(YarnApplicationState.RUNNING.name());
+
+ // Submit application to multiSubCluster
+ ApplicationId appId = ApplicationId.newInstance(Time.now(), 1);
+ ApplicationSubmissionContextInfo context = new
ApplicationSubmissionContextInfo();
+ context.setApplicationId(appId.toString());
+ context.setApplicationType("MapReduce");
+ context.setQueue("queue");
+
+ Assert.assertNotNull(interceptor.submitApplication(context, null));
+
+ GetApplicationHomeSubClusterRequest request =
+ GetApplicationHomeSubClusterRequest.newInstance(appId);
+ GetApplicationHomeSubClusterResponse response =
+ stateStore.getApplicationHomeSubCluster(request);
+
+ Assert.assertNotNull(response);
+ ApplicationHomeSubCluster homeSubCluster =
response.getApplicationHomeSubCluster();
+
+ DefaultRequestInterceptorREST interceptorREST =
+
interceptor.getInterceptorForSubCluster(homeSubCluster.getHomeSubCluster());
+
+ MockDefaultRequestInterceptorREST mockInterceptorREST =
+ (MockDefaultRequestInterceptorREST) interceptorREST;
+ mockInterceptorREST.updateApplicationState(YarnApplicationState.RUNNING,
+ appId.toString());
+
+ Set<String> stateQueries = new HashSet<>();
+ stateQueries.add(YarnApplicationState.RUNNING.name());
+
+ Set<String> typeQueries = new HashSet<>();
+ typeQueries.add("MapReduce");
+
+ ApplicationStatisticsInfo response2 =
+ interceptor.getAppStatistics(null, stateQueries, typeQueries);
+
+ Assert.assertNotNull(response2);
+ Assert.assertTrue(!response2.getStatItems().isEmpty());
Review Comment:
I will fix it.
--
This is an automated message from the Apache Git Service.
To respond to the message, please log on to GitHub and use the
URL above to go to the specific comment.
To unsubscribe, e-mail: [email protected]
For queries about this service, please contact Infrastructure at:
[email protected]
---------------------------------------------------------------------
To unsubscribe, e-mail: [email protected]
For additional commands, e-mail: [email protected]
