Package org.apache.aurora.gen

Examples of org.apache.aurora.gen.HostAttributes

162
163
164
165
166
167
168
169
170
171
172
  private Snapshot makeSnapshot() {
    Snapshot snapshot = new Snapshot();
    snapshot.setTimestamp(clock.nowMillis());
    snapshot.setHostAttributes(ImmutableSet.of(
        new HostAttributes(
            "hostA",
            ImmutableSet.of(new Attribute("attr", ImmutableSet.of("value"))))));
    snapshot.setJobs(ImmutableSet.of(
        new StoredJob(
            "jobManager",
View Full Code Here
557
558
559
560
561
562
563
564
565
566
        .transform(new Function<Offer, HostOffer>() {
          @Override
          public HostOffer apply(Offer offer) {
            return new HostOffer(
                offer,
                IHostAttributes.build(new HostAttributes().setMode(MaintenanceMode.NONE)));
          }
        });
    expect(offerQueue.getOffers()).andReturn(hostOffers);
  }
View Full Code Here
655
656
657
658
659
660
661
662
663
664
665
  }

  // Sets up a normal host, no dedicated hosts and no maintenance.
  private void setUpHost(String host, String rack) {
    IHostAttributes hostAttrs = IHostAttributes.build(
        new HostAttributes().setHost(host).setSlaveId(host + "_id")
            .setMode(NONE).setAttributes(ImmutableSet.of(rack(rack), host(host))));

    expect(this.storageUtil.attributeStore.getHostAttributes(host))
        .andReturn(Optional.of(hostAttrs)).anyTimes();
  }
View Full Code Here
175
176
177
178
179
180
181
182
        .clearResources()
        .addAllResources(resources)
        .build();
    assertEquals(0, insufficientOffers.get());
    assertFalse(gcExecutorLauncher.willUse(
        new HostOffer(smallOffer, IHostAttributes.build(new HostAttributes()))));
    assertEquals(1, insufficientOffers.get());
  }
View Full Code Here
419
420
421
422
423
424
425
426
427
428
429
    expectAnyMaintenanceCalls();
    Capture<Runnable> offerExpirationCapture = expectOfferDeclineIn(10);

    HostOffer offerAB = new HostOffer(
        Offers.makeOffer("OFFER_B").toBuilder().setSlaveId(OFFER_A.getOffer().getSlaveId()).build(),
        IHostAttributes.build(new HostAttributes()));

    driver.declineOffer(OFFER_A.getOffer().getId());
    driver.declineOffer(offerAB.getOffer().getId());

    replayAndCreateScheduler();
View Full Code Here
638
639
640
641
642
643
644
645
646
647
648
  private static HostOffer makeOffer(String offerId, String hostName, MaintenanceMode mode) {
    Offer offer = Offers.makeOffer(offerId, hostName);
    return new HostOffer(
        offer,
        IHostAttributes.build(new HostAttributes()
            .setHost(hostName)
            .setSlaveId(offer.getSlaveId().getValue())
            .setAttributes(ImmutableSet.<Attribute>of())
            .setMode(mode)));
  }
View Full Code Here
27
28
29
30
31
32
33
34
35
36
37
  public void testObjectDetachment() {
    // AURORA-729: Immutable wrapper objects are not always immutable.
    Attribute attribute = new Attribute()
        .setName("attr")
        .setValues(ImmutableSet.of("a", "b", "c"));
    HostAttributes mutable = new HostAttributes()
        .setHost("a")
        .setAttributes(ImmutableSet.of(attribute));
    IHostAttributes immutable1 = IHostAttributes.FROM_BUILDER.apply(mutable);
    IHostAttributes immutable2 = IHostAttributes.build(mutable);
    assertEquals(immutable1, immutable2);
View Full Code Here
185
186
187
188
189
190
191
192
193
194
195
    expect(storageUtil.taskStore.fetchTasks(Query.slaveScoped(hostName).active()))
        .andReturn(IScheduledTask.setFromBuilders(tasks));
  }

  private void expectMaintenanceModeChange(String hostName, MaintenanceMode mode) {
    IHostAttributes attributes = IHostAttributes.build(new HostAttributes().setHost(hostName));

    expect(storageUtil.attributeStore.getHostAttributes(hostName))
        .andReturn(Optional.of(attributes));
    IHostAttributes updated = IHostAttributes.build(attributes.newBuilder().setMode(mode));
    expect(storageUtil.attributeStore.saveHostAttributes(updated)).andReturn(true);
View Full Code Here
106
107
108
109
110
111
112
113
114
115
    assertEquals(Optional.of(attributes), read(HOST_A));
  }

  @Test(expected = IllegalArgumentException.class)
  public void testNoMode() {
    HostAttributes noMode = HOST_A_ATTRS.newBuilder();
    noMode.unsetMode();

    insert(IHostAttributes.build(noMode));
  }
View Full Code Here
114
115
116
117
118
119
120
121
122
123
    insert(IHostAttributes.build(noMode));
  }

  @Test(expected = IllegalArgumentException.class)
  public void testSaveAttributesNotSet() {
    HostAttributes attributes = HOST_A_ATTRS.newBuilder();
    attributes.unsetAttributes();

    insert(IHostAttributes.build(attributes));
  }
View Full Code Here
TOP

Related Classes of org.apache.aurora.gen.HostAttributes

  • org.apache.aurora.scheduler.async.GcExecutorLauncherTest
  • org.apache.aurora.scheduler.async.PreemptorImplTest
  • org.apache.aurora.scheduler.async.TaskSchedulerTest
  • org.apache.aurora.scheduler.base.Conversions
  • org.apache.aurora.scheduler.filter.AttributeAggregateTest
  • org.apache.aurora.scheduler.filter.SchedulingFilterImplTest
  • org.apache.aurora.scheduler.state.MaintenanceControllerImplTest
  • org.apache.aurora.scheduler.storage.backup.StorageBackupTest
  • org.apache.aurora.scheduler.storage.db.DbAttributeStore
  • org.apache.aurora.scheduler.storage.db.DbAttributeStoreTest
    • Copyright © 2018 www.massapicom. All rights reserved.
    All source code are property of their respective owners. Java is a trademark of Sun Microsystems, Inc and owned by ORACLE Inc. Contact coftware#gmail.com.