Package org.apache.aurora.gen

Examples of org.apache.aurora.gen.ResourceAggregate


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


    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

    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

    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

  @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();
View Full Code Here

    snapshot.setJobs(ImmutableSet.of(
        new StoredJob(
            "jobManager",
            new JobConfiguration().setKey(new JobKey("owner", "env", "jobA")))));
    snapshot.setQuotaConfigurations(
        ImmutableSet.of(new QuotaConfiguration("roleA", new ResourceAggregate(10, 1024, 1024))));
    snapshot.setSchedulerMetadata(new SchedulerMetadata().setFrameworkId("frameworkId"));
    snapshot.setTasks(ImmutableSet.of(new ScheduledTask()));
    return snapshot;
  }
View Full Code Here

    assertOkResponse(thrift.killTasks(query.get(), DEFAULT_LOCK, SESSION));
  }

  @Test
  public void testSetQuota() throws Exception {
    ResourceAggregate resourceAggregate = new ResourceAggregate()
        .setNumCpus(10)
        .setDiskMb(100)
        .setRamMb(200);
    expectAuth(ROOT, true);
    quotaManager.saveQuota(ROLE, IResourceAggregate.build(resourceAggregate));
View Full Code Here

    assertOkResponse(thrift.setQuota(ROLE, resourceAggregate, SESSION));
  }

  @Test
  public void testSetQuotaFails() throws Exception {
    ResourceAggregate resourceAggregate = new ResourceAggregate()
        .setNumCpus(10)
        .setDiskMb(100)
        .setRamMb(200);
    expectAuth(ROOT, true);
    quotaManager.saveQuota(ROLE, IResourceAggregate.build(resourceAggregate));
View Full Code Here

    assertResponse(INVALID_REQUEST, thrift.setQuota(ROLE, resourceAggregate, SESSION));
  }

  @Test
  public void testProvisionerSetQuota() throws Exception {
    ResourceAggregate resourceAggregate = new ResourceAggregate()
        .setNumCpus(10)
        .setDiskMb(100)
        .setRamMb(200);
    expectAuth(ROOT, false);
    expectAuth(PROVISIONER, true);
View Full Code Here

    assertResponse(AUTH_FAILED, thrift.endMaintenance(hosts, SESSION));
  }

  @Test
  public void testSetQuotaAuthFailure() throws Exception {
    ResourceAggregate resourceAggregate = new ResourceAggregate()
        .setNumCpus(10)
        .setDiskMb(100)
        .setRamMb(200);
    expectAuth(ROOT, false);
    expectAuth(PROVISIONER, false);
View Full Code Here

TOP

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

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.