Package org.apache.aurora.scheduler.storage.entities

Examples of org.apache.aurora.scheduler.storage.entities.IResourceAggregate

60
61
62
63
64
65
66
67
68
69
70
  @Test
  public void testGetQuotaInfo() {
    IScheduledTask prodTask = createTask("foo", "id1", 3, 3, 3, true);
    IScheduledTask nonProdTask = createTask("bar", "id1", 2, 2, 2, false);
    IResourceAggregate quota = IResourceAggregate.build(new ResourceAggregate(4, 4, 4));

    expectQuota(quota);
    expectTasks(prodTask, nonProdTask);
    storageUtil.expectOperations();
View Full Code Here
79
80
81
82
83
84
85
86
87
88
89
        quotaInfo.getNonProdConsumption());
  }

  @Test
  public void testGetQuotaInfoNoTasks() {
    IResourceAggregate quota = IResourceAggregate.build(new ResourceAggregate(4, 4, 4));

    expectQuota(quota);
    expectNoTasks();
    storageUtil.expectOperations();
View Full Code Here
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
public class QuotaCheckResultTest {

  @Test
  public void testGreaterOrEqualPass() {
    IResourceAggregate quota = IResourceAggregate.build(new ResourceAggregate()
        .setNumCpus(1.0)
        .setRamMb(256L)
        .setDiskMb(512L));
    IResourceAggregate request = IResourceAggregate.build(new ResourceAggregate()
        .setNumCpus(1.0)
        .setRamMb(256L)
        .setDiskMb(512L));
    assertEquals(SUFFICIENT_QUOTA, QuotaCheckResult.greaterOrEqual(quota, request).getResult());
  }
View Full Code Here
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
    assertEquals(SUFFICIENT_QUOTA, QuotaCheckResult.greaterOrEqual(quota, request).getResult());
  }

  @Test
  public void testGreaterOrEqualFailsCpu() {
    IResourceAggregate quota = IResourceAggregate.build(new ResourceAggregate()
        .setNumCpus(1.0)
        .setRamMb(256L)
        .setDiskMb(512L));
    IResourceAggregate request = IResourceAggregate.build(new ResourceAggregate()
        .setNumCpus(2.0)
        .setRamMb(256L)
        .setDiskMb(512L));
    QuotaCheckResult result = QuotaCheckResult.greaterOrEqual(quota, request);
    assertEquals(INSUFFICIENT_QUOTA, result.getResult());
View Full Code Here
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
    assertTrue(result.getDetails().get().contains("CPU"));
  }

  @Test
  public void testGreaterOrEqualFailsRam() {
    IResourceAggregate quota = IResourceAggregate.build(new ResourceAggregate()
        .setNumCpus(1.0)
        .setRamMb(256L)
        .setDiskMb(512L));
    IResourceAggregate request = IResourceAggregate.build(new ResourceAggregate()
        .setNumCpus(1.0)
        .setRamMb(512L)
        .setDiskMb(512L));
    QuotaCheckResult result = QuotaCheckResult.greaterOrEqual(quota, request);
    assertEquals(INSUFFICIENT_QUOTA, result.getResult());
View Full Code Here
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
    assertTrue(result.getDetails().get().contains("RAM"));
  }

  @Test
  public void testGreaterOrEqualFailsDisk() {
    IResourceAggregate quota = IResourceAggregate.build(new ResourceAggregate()
        .setNumCpus(1.0)
        .setRamMb(256L)
        .setDiskMb(512L));
    IResourceAggregate request = IResourceAggregate.build(new ResourceAggregate()
        .setNumCpus(1.0)
        .setRamMb(256L)
        .setDiskMb(1024L));
    QuotaCheckResult result = QuotaCheckResult.greaterOrEqual(quota, request);
    assertEquals(INSUFFICIENT_QUOTA, result.getResult());
View Full Code Here
591
592
593
594
595
596
597
598
599
600
601
602
603
604
605
606
607
608
609
  }

  @Test
  public void testSaveQuota() throws Exception {
    final String role = "role";
    final IResourceAggregate quota =
        IResourceAggregate.build(new ResourceAggregate(1.0, 128L, 1024L));

    new MutationFixture() {
      @Override
      protected void setupExpectations() throws Exception {
        storageUtil.expectWriteOperation();
        storageUtil.quotaStore.saveQuota(role, quota);
        streamMatcher.expectTransaction(Op.saveQuota(new SaveQuota(role, quota.newBuilder())))
            .andReturn(position);
      }

      @Override
      protected void performMutations(MutableStoreProvider storeProvider) {
View Full Code Here
1760
1761
1762
1763
1764
1765
1766
1767
1768
1769
1770
1771
1772
1773
1774
1775
1776
1777
  public void testGetQuota() throws Exception {
    QuotaInfo infoMock = createMock(QuotaInfo.class);
    expect(quotaManager.getQuotaInfo(ROLE)).andReturn(infoMock);
    expect(infoMock.guota()).andReturn(QUOTA);
    expect(infoMock.getProdConsumption()).andReturn(CONSUMED);
    IResourceAggregate nonProdConsumed = IResourceAggregate.build(new ResourceAggregate(1, 0, 0));
    expect(infoMock.getNonProdConsumption()).andReturn(nonProdConsumed);
    control.replay();

    Response response = assertOkResponse(thrift.getQuota(ROLE));
    assertEquals(QUOTA.newBuilder(), response.getResult().getGetQuotaResult().getQuota());
    assertEquals(
        CONSUMED.newBuilder(),
        response.getResult().getGetQuotaResult().getProdConsumption());
    assertEquals(
        nonProdConsumed.newBuilder(),
        response.getResult().getGetQuotaResult().getNonProdConsumption());
  }
View Full Code Here
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
        public QuotaInfo apply(StoreProvider storeProvider) {
          FluentIterable<ITaskConfig> tasks = FluentIterable
              .from(storeProvider.getTaskStore().fetchTasks(Query.roleScoped(role).active()))
              .transform(Tasks.SCHEDULED_TO_INFO);

          IResourceAggregate prodConsumed = fromTasks(tasks.filter(Tasks.IS_PRODUCTION));
          IResourceAggregate nonProdConsumed =
              fromTasks(tasks.filter(Predicates.not(Tasks.IS_PRODUCTION)));

          IResourceAggregate quota =
              storeProvider.getQuotaStore().fetchQuota(role).or(ResourceAggregates.none());

          return new QuotaInfo(quota, prodConsumed, nonProdConsumed);
        }
      });
View Full Code Here
131
132
133
134
135
136
137
138
139
140
141
        return new QuotaCheckResult(SUFFICIENT_QUOTA);
      }

      QuotaInfo quotaInfo = getQuotaInfo(JobKeys.from(template).getRole());

      IResourceAggregate additionalRequested =
          ResourceAggregates.scale(fromTasks(ImmutableSet.of(template)), instances);

      return QuotaCheckResult.greaterOrEqual(
          quotaInfo.guota(),
          add(quotaInfo.getProdConsumption(), additionalRequested));
View Full Code Here
TOP

Related Classes of org.apache.aurora.scheduler.storage.entities.IResourceAggregate

  • org.apache.aurora.scheduler.quota.QuotaCheckResultTest
  • org.apache.aurora.scheduler.quota.QuotaManager$QuotaManagerImpl
  • org.apache.aurora.scheduler.quota.QuotaManagerImplTest
  • org.apache.aurora.scheduler.storage.log.LogStorageTest
  • org.apache.aurora.scheduler.storage.mem.MemStorageTest
  • org.apache.aurora.scheduler.thrift.SchedulerThriftInterfaceTest
    • 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.