Examples with Job com.google.cloud.dataproc.v1.Job used on opensource projects

Example 61 with Job

use of com.google.cloud.dataproc.v1.Job in project pentaho-platform by pentaho.

the class QuartzSchedulerIT method getJobTest.

@Test
public void getJobTest() throws Exception {
    final CronTrigger cronTrigger = mock(CronTrigger.class);
    when(cronTrigger.getCronExpression()).thenReturn(CRON_EXPRESSION);
    when(quartzScheduler.getTriggersOfJob(eq(JOB_ID), eq(USER_NAME))).thenReturn(new Trigger[] { cronTrigger });
    setJobDataMap(USER_NAME);
    final Job job = scheduler.getJob(JOB_ID);
    assertEquals(JOB_ID, job.getJobId());
    assertEquals(jobDetails, job.getJobParams());
    assertEquals(USER_NAME, job.getUserName());
    assertEquals(JOB_NAME, job.getJobName());
    assertEquals(Job.JobState.NORMAL, job.getState());
}

Example 62 with Job

use of com.google.cloud.dataproc.v1.Job in project pentaho-platform by pentaho.

the class QuartzSchedulerIT method createJobTest_ForUser.

@Test
public void createJobTest_ForUser() throws Exception {
    String actionId = "actionId";
    ComplexJobTrigger trigger = getComplexJobTrigger();
    IBackgroundExecutionStreamProvider outputStreamProvider = mock(IBackgroundExecutionStreamProvider.class);
    Map<String, Serializable> paramMap = new HashMap<>();
    paramMap.put(QuartzScheduler.RESERVEDMAPKEY_ACTIONUSER, "ninja");
    final Job job = scheduler.createJob(JOB_NAME, paramMap, trigger, outputStreamProvider);
    assertNotNull(job);
    assertEquals("ninja", job.getUserName());
    assertEquals(Job.JobState.NORMAL, job.getState());
}

Example 63 with Job

use of com.google.cloud.dataproc.v1.Job in project pentaho-platform by pentaho.

the class QuartzSchedulerIT method createJobTest.

@Test
public void createJobTest() throws Exception {
    String actionId = "actionId";
    ComplexJobTrigger trigger = getComplexJobTrigger();
    IBackgroundExecutionStreamProvider outputStreamProvider = mock(IBackgroundExecutionStreamProvider.class);
    final Job job = scheduler.createJob(JOB_NAME, actionId, null, trigger, outputStreamProvider);
    assertNotNull(job);
    assertEquals(Job.JobState.NORMAL, job.getState());
    assertTrue(job.getJobParams().containsKey(QuartzScheduler.RESERVEDMAPKEY_ACTIONID));
    assertEquals(actionId, job.getJobParams().get(QuartzScheduler.RESERVEDMAPKEY_ACTIONID));
    assertTrue(job.getJobParams().containsKey(QuartzScheduler.RESERVEDMAPKEY_STREAMPROVIDER));
    assertEquals(outputStreamProvider, job.getJobParams().get(QuartzScheduler.RESERVEDMAPKEY_STREAMPROVIDER));
    assertTrue(job.getJobParams().containsKey(QuartzScheduler.RESERVEDMAPKEY_LINEAGE_ID));
    assertNotNull(job.getJobParams().get(QuartzScheduler.RESERVEDMAPKEY_LINEAGE_ID));
    assertTrue(job.getJobParams().containsKey(QuartzScheduler.RESERVEDMAPKEY_ACTIONUSER));
    assertEquals(USER_NAME, job.getJobParams().get(QuartzScheduler.RESERVEDMAPKEY_ACTIONUSER));
}

Example 64 with Job

use of com.google.cloud.dataproc.v1.Job in project pentaho-platform by pentaho.

the class QuartzSchedulerIT method testPauseAndResumeJob.

@Test
public void testPauseAndResumeJob() throws SchedulerException {
    String jobName = "complexJob1";
    int counter = TestAction.counter;
    Calendar calendar = Calendar.getInstance();
    int startingMin = calendar.get(Calendar.MINUTE);
    int startingSec = calendar.get(Calendar.SECOND) + 1;
    if (startingSec > 59) {
        startingSec = startingSec % 60;
        startingMin++;
        if (startingMin > 59) {
            startingMin = startingMin % 60;
        }
    }
    ComplexJobTrigger complexJobTrigger = new ComplexJobTrigger();
    complexJobTrigger.setHourlyRecurrence((ITimeRecurrence) null);
    complexJobTrigger.setMinuteRecurrence(startingMin);
    complexJobTrigger.setSecondRecurrence(startingSec);
    Job job = scheduler.createJob(jobName, TestAction.class, jobParams, complexJobTrigger);
    scheduler.pauseJob(job.getJobId());
    sleep(2);
    Assert.assertEquals(counter, TestAction.counter);
    Assert.assertEquals(1, scheduler.getJobs(null).size());
    scheduler.resumeJob(job.getJobId());
    sleep(2);
    Assert.assertTrue(counter != TestAction.counter);
    scheduler.removeJob(job.getJobId());
    Assert.assertEquals(0, scheduler.getJobs(null).size());
}

Example 65 with Job

use of com.google.cloud.dataproc.v1.Job in project pentaho-platform by pentaho.

the class QuartzSchedulerIT method editJobTest.

@Test
public void editJobTest() throws SchedulerException {
    SimpleJobTrigger repeater = new SimpleJobTrigger();
    repeater.setStartTime(new Date());
    repeater.setRepeatInterval(2);
    repeater.setRepeatCount(20);
    Job job = scheduler.createJob("testName", TestAction.class, new HashMap<String, Serializable>(), repeater);
    sleep(3);
    Assert.assertTrue("Job did not run the correct number of times", TestAction.counter >= 2);
    repeater = new SimpleJobTrigger();
    repeater.setStartTime(new Date());
    repeater.setRepeatInterval(5);
    repeater.setRepeatCount(3);
    int count = TestAction.counter;
    System.out.println("updating job! " + new Date());
    scheduler.updateJob(job.getJobId(), new HashMap<String, Serializable>(), repeater);
    List<Job> jobs = scheduler.getJobs(null);
    Assert.assertEquals("Unexpected number of scheduled jobs", 1, jobs.size());
    SimpleJobTrigger simpleJobTrigger = (SimpleJobTrigger) jobs.get(0).getJobTrigger();
    Assert.assertEquals(5, simpleJobTrigger.getRepeatInterval());
    Assert.assertEquals(3, simpleJobTrigger.getRepeatCount());
    sleep(1);
    Assert.assertEquals("Job did not run the correct number of times", count + 1, TestAction.counter);
    count = TestAction.counter;
    sleep(3);
    Assert.assertEquals("Job ran unexpectedly", count, TestAction.counter);
    sleep(3);
    Assert.assertTrue("Job did not run the correct number of times", count < TestAction.counter);
}