YARN-7199. Fix TestAMRMClientContainerRequest.testOpportunisticAndGuaranteedRequests. (Botong Huang via asuresh)

This commit is contained in:
Arun Suresh 2017-09-18 11:26:44 -07:00
parent 0adc0471d0
commit 29dd55153e

View File

@ -118,7 +118,7 @@ public static class ContainerRequest {
private String nodeLabelsExpression; private String nodeLabelsExpression;
private ExecutionTypeRequest executionTypeRequest = private ExecutionTypeRequest executionTypeRequest =
ExecutionTypeRequest.newInstance(); ExecutionTypeRequest.newInstance();
private String resourceProfile; private String resourceProfile = ProfileCapability.DEFAULT_PROFILE;
/** /**
* Instantiates a {@link ContainerRequest} with the given constraints and * Instantiates a {@link ContainerRequest} with the given constraints and
@ -540,6 +540,11 @@ public ContainerRequestBuilder executionTypeRequest(
return this; return this;
} }
public ContainerRequestBuilder resourceProfile(String resourceProfile) {
containerRequest.resourceProfile = resourceProfile;
return this;
}
public ContainerRequest build() { public ContainerRequest build() {
containerRequest.sanityCheck(); containerRequest.sanityCheck();
return containerRequest; return containerRequest;