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

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

View File

@ -196,7 +196,15 @@ public class MockServiceAM extends ServiceMaster {
@Override @Override
public RegisterApplicationMasterResponse registerApplicationMaster( 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 response = mock(
RegisterApplicationMasterResponse.class); RegisterApplicationMasterResponse.class);
when(response.getResourceTypes()).thenReturn( when(response.getResourceTypes()).thenReturn(