How to use

ClusterScheduler

in

co.cask.coopr.scheduler

List<Set<ClusterTask>> clusterTasks = createClusterTasks(linearizedTasks, cluster, job,
                             jobPlanner.getServiceMap(),
                             clusterAction, jobPlanner.getNodeMap());

clusterScheduler.run();

@Test(timeout = 20000)
public void testFalseOnStartStopsJob() throws Exception {
 String tenantId = "q";
 ClusterScheduler clusterScheduler = injector.getInstance(ClusterScheduler.class);
 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));
 // tell mock callback to return false for onStart callback
 mockClusterCallback.setReturnOnStart(false);
 // wait for start callback to finish
 waitForCallback(callbackScheduler);
 Assert.assertEquals(CallbackData.Type.START, mockClusterCallback.getReceivedCallbacks().get(0).getType());
 // wait for fail callback to finish
 if (mockClusterCallback.getReceivedCallbacks().size() < 2) {
  waitForCallback(callbackScheduler);
 }
 Assert.assertEquals(CallbackData.Type.FAILURE, mockClusterCallback.getReceivedCallbacks().get(1).getType());
 // there also should not be any jobs in the queue
 Assert.assertEquals(0, jobQueues.size(tenantId));
}

           ClusterAction.CLUSTER_CREATE, 0, 0);
clusterScheduler.run();
callbackScheduler.run();

           ClusterAction.CLUSTER_CREATE, 0, 0);
clusterScheduler.run();

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());
}

clusterScheduler.run();

           ClusterAction.CLUSTER_CREATE, 0, 0);
clusterScheduler.run();

           ClusterAction.CLUSTER_CREATE, 0, 0);
clusterScheduler.run();
callbackScheduler.run();

           ClusterAction.SOLVE_LAYOUT, 0, 0);
clusterScheduler.run();
assertResponseStatus(response, HttpResponseStatus.OK);
clusterScheduler.run();

clusterScheduler.run();
waitForCallback(callbackScheduler);
Assert.assertEquals(1, jobQueues.size(tenantId));

clusterScheduler.run();
callbackScheduler.run();

assertResponseStatus(response, HttpResponseStatus.OK);
clusterScheduler.run();
callbackScheduler.run();

clusterScheduler.run();

           ClusterAction.CLUSTER_CREATE, 0, 0);
clusterScheduler.run();
callbackScheduler.run();

           ClusterAction.CLUSTER_CREATE, 0, 0);
clusterScheduler.run();
callbackScheduler.run();

clusterScheduler.run();
callbackScheduler.run();