use of com.mesosphere.sdk.specification.ResourceSpec in project dcos-commons by mesosphere.
the class OfferEvaluationUtilsTest method testResourceIncreaseInsufficient.
private void testResourceIncreaseInsufficient(Optional<String> namespace) {
String resourceId = UUID.randomUUID().toString();
Protos.Value current = getValue(4);
Protos.Value desired = getValue(5);
Protos.Value toAdd = ValueUtils.subtract(desired, current);
ResourceSpec resourceSpec = getResourceSpec(desired);
when(mockPool.consumeReserved(RESOURCE_NAME, desired, resourceId)).thenReturn(Optional.of(getMesosResource(getResourceSpec(current), resourceId, namespace)));
when(mockPool.consumeReservableMerged(RESOURCE_NAME, toAdd, Constants.ANY_ROLE)).thenReturn(Optional.empty());
ReserveEvaluationOutcome outcome = OfferEvaluationUtils.evaluateSimpleResource(mockStage, resourceSpec, Optional.of(resourceId), namespace, mockPool);
Assert.assertFalse(outcome.getEvaluationOutcome().isPassing());
Assert.assertTrue(outcome.getEvaluationOutcome().getOfferRecommendations().isEmpty());
Assert.assertFalse(outcome.getResourceId().isPresent());
}
use of com.mesosphere.sdk.specification.ResourceSpec in project dcos-commons by mesosphere.
the class OfferEvaluationUtilsTest method testResourceSufficient.
private void testResourceSufficient(Optional<String> resourceId, Optional<String> namespace) {
Protos.Value desired = getValue(5);
ResourceSpec resourceSpec = getResourceSpec(desired);
if (resourceId.isPresent()) {
when(mockPool.consumeReserved(RESOURCE_NAME, desired, resourceId.get())).thenReturn(Optional.of(getMesosResource(resourceSpec, resourceId.get(), namespace)));
} else {
when(mockPool.consumeReservableMerged(RESOURCE_NAME, desired, Constants.ANY_ROLE)).thenReturn(Optional.of(getMesosResource(desired)));
}
ReserveEvaluationOutcome outcome = OfferEvaluationUtils.evaluateSimpleResource(mockStage, resourceSpec, resourceId, namespace, mockPool);
Assert.assertTrue(outcome.getEvaluationOutcome().isPassing());
if (resourceId.isPresent()) {
Assert.assertTrue(outcome.getEvaluationOutcome().getOfferRecommendations().isEmpty());
Assert.assertEquals(resourceId.get(), outcome.getResourceId().get());
} else {
OfferRecommendation recommendation = outcome.getEvaluationOutcome().getOfferRecommendations().get(0);
Assert.assertTrue(recommendation instanceof ReserveOfferRecommendation);
Assert.assertTrue(outcome.getResourceId().isPresent());
Protos.Resource resource = recommendation.getOperation().getReserve().getResources(0);
Assert.assertEquals(desired.getScalar(), resource.getScalar());
if (namespace.isPresent()) {
Assert.assertEquals(namespace.get(), ResourceUtils.getNamespace(resource).get());
} else {
Assert.assertFalse(ResourceUtils.getNamespace(resource).isPresent());
}
}
}
use of com.mesosphere.sdk.specification.ResourceSpec in project dcos-commons by mesosphere.
the class ResourceBuilderTest method testExistingFromResourceSpec.
private static void testExistingFromResourceSpec(Optional<String> namespace) {
Optional<String> resourceId = Optional.of(UUID.randomUUID().toString());
ResourceSpec resourceSpec = new DefaultResourceSpec("cpus", VALUE, TestConstants.ROLE, Constants.ANY_ROLE, TestConstants.PRINCIPAL);
validateScalarResource(ResourceBuilder.fromSpec(resourceSpec, resourceId, namespace).build(), resourceId, namespace);
}
use of com.mesosphere.sdk.specification.ResourceSpec in project dcos-commons by mesosphere.
the class ResourceBuilderTest method testNewFromResourceSpec.
private static void testNewFromResourceSpec(Optional<String> namespace) {
ResourceSpec resourceSpec = new DefaultResourceSpec("cpus", VALUE, TestConstants.ROLE, Constants.ANY_ROLE, TestConstants.PRINCIPAL);
ResourceBuilder resourceBuilder = ResourceBuilder.fromSpec(resourceSpec, Optional.empty(), namespace);
Protos.Resource resource = resourceBuilder.build();
validateScalarResource(resource, Optional.empty(), namespace);
}
use of com.mesosphere.sdk.specification.ResourceSpec in project dcos-commons by mesosphere.
the class ResourceBuilderTest method testRefineStaticResource.
/*
name: "cpus"
type: SCALAR
scalar {
value: 1.0
}
reservations {
role: "base-role"
type: STATIC
}
reservations {
principal: "test-principal"
labels {
labels {
key: "resource_id"
value: "a395f14b-3cc8-4009-9dc4-51838b423aed"
}
}
role: "test-role"
type: DYNAMIC
}
*/
private static void testRefineStaticResource(Optional<String> namespace) {
ResourceRefinementCapabilityContext context = new ResourceRefinementCapabilityContext(Capabilities.getInstance());
try {
ResourceSpec resourceSpec = new DefaultResourceSpec("cpus", VALUE, TestConstants.ROLE, TestConstants.PRE_RESERVED_ROLE, TestConstants.PRINCIPAL);
ResourceBuilder resourceBuilder = ResourceBuilder.fromSpec(resourceSpec, Optional.empty(), namespace);
Protos.Resource resource = resourceBuilder.build();
Assert.assertEquals(2, resource.getReservationsCount());
validateScalarResourceRefined(resource, Optional.empty(), namespace);
Assert.assertEquals(Protos.Resource.ReservationInfo.Type.STATIC, resource.getReservations(0).getType());
Assert.assertEquals(TestConstants.PRE_RESERVED_ROLE, resource.getReservations(0).getRole());
} finally {
context.reset();
}
}
Aggregations