service/src/test/java/org/apache/griffin/core/job/JobServiceImplTest.java

package org.apache.griffin.core.job;

import org.apache.griffin.core.exception.GriffinException;
import org.apache.griffin.core.job.entity.AbstractJob;
import org.apache.griffin.core.job.repo.JobRepo;
import org.junit.Test;
import org.junit.runner.RunWith;
import org.mockito.InjectMocks;
import org.mockito.Mock;
import org.quartz.JobKey;
import org.quartz.Scheduler;
import org.quartz.SchedulerException;
import org.springframework.scheduling.quartz.SchedulerFactoryBean;
import org.springframework.test.context.junit4.SpringRunner;

import static org.apache.griffin.core.util.EntityMocksHelper.createGriffinJob;
import static org.mockito.BDDMockito.given;
import static org.mockito.Matchers.any;
import static org.mockito.Mockito.mock;
import static org.mockito.Mockito.verify;
import static org.mockito.internal.verification.VerificationModeFactory.times;

@RunWith(SpringRunner.class)
public class JobServiceImplTest {

    @Mock
    private JobRepo<AbstractJob> jobRepo;

    @Mock
    private SchedulerFactoryBean factory;

    @InjectMocks
    private JobServiceImpl jobService;


    @Test
    public void testTriggerJobById() throws SchedulerException {
        Long jobId = 1L;
        AbstractJob job = createGriffinJob();
        given(jobRepo.findByIdAndDeleted(jobId,false)).willReturn(job);
        Scheduler scheduler = mock(Scheduler.class);
        given(scheduler.checkExists(any(JobKey.class))).willReturn(true);
        given(factory.getScheduler()).willReturn(scheduler);
        jobService.triggerJobById(jobId);

        verify(scheduler, times(1)).scheduleJob(any());
    }


    @Test(expected = GriffinException.NotFoundException.class)
    public void testTriggerJobByIdFail() throws SchedulerException {
        Long jobId = 1L;
        given(jobRepo.findByIdAndDeleted(jobId,false)).willReturn(null);
        jobService.triggerJobById(jobId);
    }
}