예제 #1
0
 @Test(priority = 3)
 public void testUpdateJobInstance() {
   SchedulerJobInstanceHandle handle = instances.keySet().iterator().next();
   SchedulerJobInstanceInfo info = instances.get(handle);
   SchedulerJobInstanceRun run = info.getInstanceRunList().get(0);
   run.setInstanceState(SchedulerJobInstanceState.LAUNCHED);
   schedulerDAO.updateJobInstanceRun(run);
   // Get the instance
   Assert.assertEquals(schedulerDAO.getSchedulerJobInstanceInfo(handle), info);
 }
예제 #2
0
  @Test(priority = 2)
  public void testStoreInstance() throws Exception {
    long currentTime = System.currentTimeMillis();
    SchedulerJobInstanceHandle instanceHandle = new SchedulerJobInstanceHandle(UUID.randomUUID());
    SchedulerJobInstanceInfo firstInstance =
        new SchedulerJobInstanceInfo(
            instanceHandle, jobHandle, currentTime, new ArrayList<SchedulerJobInstanceRun>());
    SchedulerJobInstanceRun run1 =
        new SchedulerJobInstanceRun(
            instanceHandle,
            1,
            new LensSessionHandle(UUID.randomUUID(), UUID.randomUUID()),
            currentTime,
            currentTime,
            "/tmp/",
            QueryHandle.fromString(UUID.randomUUID().toString()),
            SchedulerJobInstanceState.WAITING);
    instances.put(firstInstance.getId(), firstInstance);
    schedulerDAO.storeJobInstance(firstInstance);
    schedulerDAO.storeJobInstanceRun(run1);
    // Put run in the instance
    firstInstance.getInstanceRunList().add(run1);

    currentTime = System.currentTimeMillis();
    instanceHandle = new SchedulerJobInstanceHandle(UUID.randomUUID());
    SchedulerJobInstanceInfo secondInstance =
        new SchedulerJobInstanceInfo(
            instanceHandle, jobHandle, currentTime, new ArrayList<SchedulerJobInstanceRun>());
    SchedulerJobInstanceRun run2 =
        new SchedulerJobInstanceRun(
            instanceHandle,
            1,
            new LensSessionHandle(UUID.randomUUID(), UUID.randomUUID()),
            currentTime,
            currentTime,
            "/tmp/",
            QueryHandle.fromString(UUID.randomUUID().toString()),
            SchedulerJobInstanceState.WAITING);
    instances.put(secondInstance.getId(), secondInstance);
    schedulerDAO.storeJobInstance(secondInstance);
    schedulerDAO.storeJobInstanceRun(run2);
    secondInstance.getInstanceRunList().add(run2);

    List<SchedulerJobInstanceInfo> handleList = schedulerDAO.getJobInstances(jobHandle);
    // Size should be 2
    Assert.assertEquals(handleList.size(), 2);
    // Get the definition of instance from the store.
    SchedulerJobInstanceInfo instance1 = handleList.get(0);
    Assert.assertEquals(instances.get(handleList.get(0).getId()), instance1);

    SchedulerJobInstanceInfo instance2 = handleList.get(1);
    Assert.assertEquals(instances.get(handleList.get(1).getId()), instance2);
  }