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

YARN-7963. Updated MockServiceAM unit test to prevent test hang.

Browse Source 
Contributed by Chandni Singh
This commit is contained in:
Eric Yang 2018-02-26 18:49:01 -05:00
parent 6ce9f79cc9
commit b4f1ba1413
1 changed files with 9 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
hadoop-yarn-project/hadoop-yarn/hadoop-yarn-applications/hadoop-yarn-services/hadoop-yarn-services-core/src/test/java/org/apache/hadoop/yarn/service/MockServiceAM.java
View File

@ -196,7 +196,15 @@ protected AMRMClientAsync<AMRMClient.ContainerRequest> createAMRMClient() {
@Override
public RegisterApplicationMasterResponse registerApplicationMaster(
String appHostName, int appHostPort, String appTrackingUrl) {
String appHostName, int appHostPort, String appTrackingUrl,
Map placementConstraintsMap) throws YarnException, IOException {
return this.registerApplicationMaster(appHostName, appHostPort,
appTrackingUrl);
}
@Override
public RegisterApplicationMasterResponse registerApplicationMaster(
String appHostName, int appHostPort, String appTrackingUrl) {
RegisterApplicationMasterResponse response = mock(
RegisterApplicationMasterResponse.class);
when(response.getResourceTypes()).thenReturn(