Package org.apache.aurora.gen

Examples of org.apache.aurora.gen.Response

2216
2217
2218
2219
2220
2221
2222
2223
2224
2225
  public void testGetLocks() throws Exception {
    expect(lockManager.getLocks()).andReturn(ImmutableSet.of(LOCK));

    control.replay();

    Response response = thrift.getLocks();
    assertEquals(
        LOCK.newBuilder(),
        Iterables.getOnlyElement(response.getResult().getGetLocksResult().getLocks()));
  }
View Full Code Here
2232
2233
2234
2235
2236
2237
2238
2239
2240
2241
2242
2243
2244
2245
    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
2251
2252
2253
2254
2255
2256
2257
2258
2259
2260
    expect(storageUtil.jobUpdateStore.fetchJobUpdateSummaries(IJobUpdateQuery.build(query)))
        .andReturn(IJobUpdateSummary.listFromBuilders(summaries));

    control.replay();

    Response response = assertOkResponse(thrift.getJobUpdateSummaries(query));
    assertEquals(
        summaries,
        response.getResult().getGetJobUpdateSummariesResult().getUpdateSummaries());
  }
View Full Code Here
2266
2267
2268
2269
2270
2271
2272
2273
2274
2275
    expect(storageUtil.jobUpdateStore.fetchJobUpdateDetails(id))
        .andReturn(Optional.of(IJobUpdateDetails.build(details)));

    control.replay();

    Response response = assertOkResponse(thrift.getJobUpdateDetails(id));
    assertEquals(
        details,
        response.getResult().getGetJobUpdateDetailsResult().getDetails());
  }
View Full Code Here
2341
2342
2343
2344
2345
2346
2347
2348
    // Validate key is populated during sanitizing.
    JobUpdateRequest request = buildJobUpdateRequest(update);
    request.getTaskConfig().unsetJob();

    Response response = assertOkResponse(thrift.startJobUpdate(request, SESSION));
    assertEquals(UPDATE_ID, response.getResult().getStartJobUpdateResult().getUpdateId());
  }
View Full Code Here
2379
2380
2381
2382
2383
2384
2385
2386
2387
    jobUpdateController.start(IJobUpdate.build(expected), USER);

    control.replay();

    Response response =
        assertOkResponse(thrift.startJobUpdate(buildJobUpdateRequest(update), SESSION));
    assertEquals(UPDATE_ID, response.getResult().getStartJobUpdateResult().getUpdateId());
  }
View Full Code Here
2501
2502
2503
2504
2505
2506
2507
2508
2509
2510
2511
        newTask,
        ImmutableMap.of(oldTask.getAssignedTask().getTask(), ImmutableSet.of(new Range(0, 0))));

    control.replay();
    JobUpdateRequest request = buildJobUpdateRequest(update);
    Response response = thrift.startJobUpdate(request, SESSION);
    assertResponse(OK, response);
    assertEquals(
        NOOP_JOB_UPDATE_MESSAGE,
        Iterables.getOnlyElement(response.getDetails()).getMessage());
  }
View Full Code Here
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
        .setKey(LockKey.job(new JobKey("role", "env", "name")))
        .setToken("token");
    SessionKey session = new SessionKey()
        .setData("session data".getBytes(StandardCharsets.UTF_8))
        .setMechanism("fake");
    Response response = new Response()
        .setResponseCode(OK);

    JobConfiguration job = JOB_CONFIG.newBuilder();
    expect(thrift.createJob(job, lock, session)).andReturn(response);

    replayAndStart();

    Response actualResponse = getRequestBuilder("/apibeta/createJob")
        .entity(
            ImmutableMap.of("description", job, "lock", lock, "session", session),
            MediaType.APPLICATION_JSON)
        .post(Response.class);
    assertEquals(response, actualResponse);
View Full Code Here
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
    assertEquals(response, actualResponse);
  }

  @Test
  public void testGetRoleSummary() throws Exception {
    Response response = new Response()
        .setResponseCode(OK)
        .setResult(Result.roleSummaryResult(new RoleSummaryResult()
            .setSummaries(ImmutableSet.of(new RoleSummary()
                .setCronJobCount(1)
                .setJobCount(2)
                .setRole("role")))));

    expect(thrift.getRoleSummary()).andReturn(response);

    replayAndStart();

    Response actualResponse = getRequestBuilder("/apibeta/getRoleSummary")
        .post(Response.class);
    assertEquals(response, actualResponse);
  }
View Full Code Here
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
    assertEquals(response, actualResponse);
  }

  @Test
  public void testGetJobSummary() throws Exception {
    Response response = new Response()
        .setResponseCode(OK)
        .setResult(Result.jobSummaryResult(new JobSummaryResult()
            .setSummaries(ImmutableSet.of(new JobSummary()
                .setJob(JOB_CONFIG.newBuilder())))));

    expect(thrift.getJobSummary("roleA")).andReturn(response);

    replayAndStart();

    Response actualResponse = getRequestBuilder("/apibeta/getJobSummary")
        .entity(ImmutableMap.of("role", "roleA"), MediaType.APPLICATION_JSON)
        .post(Response.class);
    assertEquals(response, actualResponse);
  }
View Full Code Here
TOP

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

  • org.apache.aurora.scheduler.http.api.ApiBetaTest
  • org.apache.aurora.scheduler.http.ServletFilterTest
  • org.apache.aurora.scheduler.thrift.aop.AopModuleTest
  • org.apache.aurora.scheduler.thrift.aop.FeatureToggleInterceptorTest
  • org.apache.aurora.scheduler.thrift.aop.ServerInfoInterceptor
  • org.apache.aurora.scheduler.thrift.aop.ServerInfoInterceptorTest
  • org.apache.aurora.scheduler.thrift.aop.ThriftStatsExporterInterceptorTest
  • org.apache.aurora.scheduler.thrift.SchedulerThriftInterface
  • org.apache.aurora.scheduler.thrift.SchedulerThriftInterfaceTest
  • org.apache.aurora.scheduler.thrift.Util
    • 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.