); FinishTaskRequest finishRequest = new FinishTaskRequest("workerX", PROVISIONER_ID, tenantId, task.getTaskId(), null, null, 0, null, ipAddresses, result);
Assert.assertEquals(clusterTask.getTaskId(), task.getTaskId());
private void testCallbacks(boolean failJob) throws Exception { ClusterScheduler clusterScheduler = injector.getInstance(ClusterScheduler.class); String tenantId = cluster.getAccount().getTenantId(); clusterQueues.add(tenantId, new Element(cluster.getId(), ClusterAction.CLUSTER_CREATE.name())); clusterScheduler.run(); CallbackScheduler callbackScheduler = injector.getInstance(CallbackScheduler.class); // should be no job in the queue until the start callback runs Assert.assertEquals(0, jobQueues.size(tenantId)); waitForCallback(callbackScheduler); Assert.assertEquals(CallbackData.Type.START, mockClusterCallback.getReceivedCallbacks().get(0).getType()); JobScheduler jobScheduler = injector.getInstance(JobScheduler.class); jobScheduler.run(); // take tasks until there are no more TakeTaskRequest takeRequest = new TakeTaskRequest("consumer1", PROVISIONER_ID, tenantId); SchedulableTask task = TestHelper.takeTask(getInternalServerUrl(), takeRequest); while (task != null) { FinishTaskRequest finishRequest = new FinishTaskRequest("consumer1", PROVISIONER_ID, tenantId, task.getTaskId(), null, null, failJob ? 1 : 0, null, null, null); TestHelper.finishTask(getInternalServerUrl(), finishRequest); jobScheduler.run(); jobScheduler.run(); task = TestHelper.takeTask(getInternalServerUrl(), takeRequest); } jobScheduler.run(); waitForCallback(callbackScheduler); // at this point, the failure callback should have run Assert.assertEquals(failJob ? CallbackData.Type.FAILURE : CallbackData.Type.SUCCESS, mockClusterCallback.getReceivedCallbacks().get(1).getType()); }
Map<String, String> ipAddresses = ImmutableMap.of("access", "123.456.789.123"); FinishTaskRequest finishRequest = new FinishTaskRequest("consumer1", PROVISIONER_ID, tenantId, task.getTaskId(), null, null, 0, null, ipAddresses, result); TestHelper.finishTask(getInternalServerUrl(), finishRequest); ipAddresses = ImmutableMap.of("access", "456.789.123.123"); finishRequest = new FinishTaskRequest("consumer1", PROVISIONER_ID, tenantId, task.getTaskId(), null, null, 0, null, ipAddresses, result); TestHelper.finishTask(getInternalServerUrl(), finishRequest);
result.addProperty("ipaddress", "111.222.333." + i); FinishTaskRequest finishRequest = new FinishTaskRequest("workerX", PROVISIONER_ID, tenantId, task.getTaskId(), null, null, 0, null, null, result); task.getTaskId(), null, null, 1, null, null, result); TestHelper.finishTask(getBaseUrlInternalAPI(), finishRequest); assertResponseStatus(response, HttpResponseStatus.OK); result.addProperty("ipaddress", "111.222.333." + i); finishRequest = new FinishTaskRequest("workerX", PROVISIONER_ID, tenantId, task.getTaskId(), null, null, 0, null, null, result);
Map<String, String> ipAddresses = ImmutableMap.of("access", "123.456.789.123"); FinishTaskRequest finishRequest = new FinishTaskRequest("consumer1", PROVISIONER_ID, tenantId, task.getTaskId(), null, null, 0, null, ipAddresses, result); TestHelper.finishTask(getInternalServerUrl(), finishRequest); ipAddresses = ImmutableMap.of("access", "456.789.123.123"); finishRequest = new FinishTaskRequest("consumer1", PROVISIONER_ID, tenantId, task.getTaskId(), null, null, 0, null, ipAddresses, result); TestHelper.finishTask(getInternalServerUrl(), finishRequest); task = TestHelper.takeTask(getInternalServerUrl(), takeRequest); finishRequest = new FinishTaskRequest("consumer1", PROVISIONER_ID, tenantId, task.getTaskId(), null, null, 0, null, null, null); TestHelper.finishTask(getInternalServerUrl(), finishRequest); jobScheduler.run();
Assert.assertEquals(clusterTask.getTaskId(), task.getTaskId());
result.addProperty("ipaddress", "111.222.333." + i); FinishTaskRequest finishRequest = new FinishTaskRequest("workerX", PROVISIONER_ID, tenantId, task.getTaskId(), null, null, 1, null, null, result);
new FinishTaskRequest("workerX", PROVISIONER_ID, tenantId, task.getTaskId(), null, null, 0, null, null, null);
FinishTaskRequest finishRequest = new FinishTaskRequest("workerX", PROVISIONER_ID, tenantId, task.getTaskId(), null, null, 0, null, null, result);
FinishTaskRequest finishRequest = new FinishTaskRequest("workerX", PROVISIONER_ID, tenantId, task.getTaskId(), null, null, 0, null, null, result);