LOGGER.info(" {}", TextFormat.shortDebugString(badResource)); return offer.toBuilder() .clearResources() .addAllResources(goodResources)
offersToSend.add(serviceOfferResources.getOffer().toBuilder() .clearResources() .addAllResources(serviceOfferResources.getResources())
@Test public void regionAndZoneInjected() { offer = offer.toBuilder() .setDomain(TestConstants.LOCAL_DOMAIN_INFO) .build(); stage.evaluate(new MesosResourcePool(offer, Optional.of(Constants.ANY_ROLE)), podInfoBuilder); Protos.TaskInfo.Builder taskBuilder = podInfoBuilder.getTaskBuilder(TestConstants.TASK_NAME); Map<String, Protos.Environment.Variable> env = EnvUtils.toMap(taskBuilder.getCommand().getEnvironment()); Assert.assertEquals(TestConstants.LOCAL_REGION, env.get(EnvConstants.REGION_TASKENV).getValue()); Assert.assertEquals(TestConstants.ZONE, env.get(EnvConstants.ZONE_TASKENV).getValue()); }
ResourceTestUtils.getPersistenceId(formatTaskToReoffer.getResources(1))))); Protos.Offer offer = OfferTestUtils.getOffer(expectedResources).toBuilder() .addExecutorIds(formatTaskToReoffer.getExecutor().getExecutorId()) .build();
Protos.Offer offer = OfferTestUtils.getCompleteOffer(Arrays.asList(expectedTaskCpu)).toBuilder() .addExecutorIds(taskInfo.getExecutor().getExecutorId()) .build();
expectedResources.add(ResourceTestUtils.getReservedCpus(1.0, resourceId)); Protos.Offer.Builder offerBuilder = OfferTestUtils.getOffer(expectedResources).toBuilder(); Protos.Attribute.Builder attrBuilder = offerBuilder.addAttributesBuilder().setName("rack").setType(Protos.Value.Type.TEXT);
private static Offer offerWithAttribute(String name, String value) { Protos.Resource resource = ResourceBuilder.fromUnreservedValue( "cpus", Protos.Value.newBuilder() .setType(Protos.Value.Type.SCALAR) .setScalar(Protos.Value.Scalar.newBuilder().setValue(1.0)) .build()) .build(); Offer.Builder offerBuilder = OfferTestUtils.getCompleteOffer(resource).toBuilder(); offerBuilder.addAttributesBuilder() .setName(name) .setType(Value.Type.TEXT) .getTextBuilder().setValue(value); return offerBuilder.build(); }
private static Offer offerWithHost(String host) { return OfferTestUtils.getCompleteOffer( ResourceBuilder.fromUnreservedValue( "cpus", Protos.Value.newBuilder() .setType(Protos.Value.Type.SCALAR) .setScalar(Protos.Value.Scalar.newBuilder().setValue(1.0)) .build()) .build()) .toBuilder() .setHostname(host) .build(); }
@SuppressWarnings("deprecation") @Test public void testFilteredResources() { scheduler.setApiServerStarted(); String resourceId = "unexpected-volume-id-1"; Protos.Resource resourceMatchingRole = ResourceTestUtils.getReservedRootVolume(1000.0, resourceId, resourceId); Protos.Resource resourceOtherRole = resourceMatchingRole.toBuilder().setRole("other-role").build(); Protos.Offer offer = getOffer("foo").toBuilder().addResources(resourceMatchingRole).addResources(resourceOtherRole).build(); Protos.Offer filteredOffer = getOffer("foo").toBuilder().addResources(resourceMatchingRole).build(); // Resource with other role should be filtered out: scheduler.resourceOffers(mockSchedulerDriver, Collections.singletonList(offer)); verify(mockOfferProcessor).enqueue(Collections.singletonList(filteredOffer)); }
private static Protos.Offer offerWithAgent(String agentId, Protos.Resource resource) { Protos.Offer.Builder o = OfferTestUtils.getCompleteOffer(resource).toBuilder(); o.getSlaveIdBuilder().setValue(agentId); return o.build(); } }
private static void verifyDomainIsSet(Protos.DomainInfo expectedDomain) { // Infer the configured domain via a placement rule invocation Protos.Offer offerWithExpectedDomain = getOffer().toBuilder().setDomain(expectedDomain).build(); EvaluationOutcome outcome = new IsLocalRegionRule().filter(offerWithExpectedDomain, null, null); Assert.assertTrue(outcome.getReason(), outcome.isPassing()); }
private Protos.Offer getOffer(Protos.DomainInfo domainInfo) { return getOffer().toBuilder() .setDomain(domainInfo) .build(); } }