use of uk.gov.gchq.gaffer.jobtracker.Job in project Gaffer by gchq.
the class GraphTest method shouldAddSchemaGroupsIfNotIncludedInJob.
@Test
public void shouldAddSchemaGroupsIfNotIncludedInJob() throws OperationException {
// given
final Job job = new Job(null, new OperationChain.Builder().first(new GetAllElements()).build());
final Store store = mock(Store.class);
given(store.getSchema()).willReturn(new Schema.Builder().entity(TestGroups.ENTITY, new SchemaEntityDefinition()).edge(TestGroups.EDGE, new SchemaEdgeDefinition()).edge(TestGroups.EDGE_2, new SchemaEdgeDefinition()).build());
given(store.getProperties()).willReturn(new StoreProperties());
final Graph graph = new Graph.Builder().config(new GraphConfig.Builder().graphId(GRAPH_ID).build()).storeProperties(StreamUtil.storeProps(getClass())).store(store).build();
final ArgumentCaptor<Job> jobCaptor = ArgumentCaptor.forClass(Job.class);
final ArgumentCaptor<Context> contextCaptor = ArgumentCaptor.forClass(Context.class);
given(store.executeJob(jobCaptor.capture(), contextCaptor.capture())).willReturn(new JobDetail());
// when
graph.executeJob(job, context);
// then
final GetAllElements operation = (GetAllElements) ((OperationChain) jobCaptor.getValue().getOperation()).getOperations().get(0);
assertEquals(new View.Builder().entity(TestGroups.ENTITY, new ViewElementDefinition()).edge(TestGroups.EDGE, new ViewElementDefinition()).edge(TestGroups.EDGE_2, new ViewElementDefinition()).build(), operation.getView());
}
use of uk.gov.gchq.gaffer.jobtracker.Job in project Gaffer by gchq.
the class GraphTest method shouldThrowExceptionOnExecuteJobUsingJobWithANullJob.
@Test
public void shouldThrowExceptionOnExecuteJobUsingJobWithANullJob() throws OperationException {
// Given
final Context context = new Context();
final Graph graph = new Graph.Builder().config(new GraphConfig.Builder().graphId(GRAPH_ID).build()).storeProperties(StreamUtil.storeProps(getClass())).addSchemas(StreamUtil.schemas(getClass())).build();
final Job job = null;
// When / Then
assertThatIllegalArgumentException().isThrownBy(() -> graph.executeJob(job, context)).withMessage("A job is required");
}
use of uk.gov.gchq.gaffer.jobtracker.Job in project Gaffer by gchq.
the class StoreTest method shouldCorrectlySetUpScheduledJobDetail.
@Test
public void shouldCorrectlySetUpScheduledJobDetail() throws Exception {
// Given
final StoreProperties properties = mock(StoreProperties.class);
given(properties.getJobTrackerEnabled()).willReturn(true);
given(properties.getJobExecutorThreadCount()).willReturn(1);
StoreImpl2 store = new StoreImpl2();
store.initialise("graphId", schema, properties);
final Repeat repeat = new Repeat(0, 100, TimeUnit.SECONDS);
final OperationChain opChain = new OperationChain.Builder().first(new DiscardOutput()).build();
final Context context = new Context(user);
final String operationChainOverviewString = opChain.toOverviewString();
final String serialisedOperationChain = new String(JSONSerialiser.serialise(opChain), Charset.forName(CommonConstants.UTF_8));
// When - setup job
JobDetail parentJobDetail = store.executeJob(new Job(repeat, opChain), context);
ScheduledExecutorService service = store.getExecutorService();
// Then - assert scheduled
verify(service).scheduleAtFixedRate(any(Runnable.class), eq(repeat.getInitialDelay()), eq(repeat.getRepeatPeriod()), eq(repeat.getTimeUnit()));
// Then - assert job detail is as expected
assertEquals(JobStatus.SCHEDULED_PARENT, parentJobDetail.getStatus());
assertEquals(operationChainOverviewString, parentJobDetail.getOpChain());
assertEquals(serialisedOperationChain, parentJobDetail.getSerialisedOperationChain());
assertEquals(context.getUser(), parentJobDetail.getUser());
}
use of uk.gov.gchq.gaffer.jobtracker.Job in project Gaffer by gchq.
the class JobServiceV2IT method shouldCorrectlyDoAndThenCancelScheduledJob.
@Test
public void shouldCorrectlyDoAndThenCancelScheduledJob() throws IOException, InterruptedException {
// When
final Repeat repeat = new Repeat(1, 2, TimeUnit.SECONDS);
Job job = new Job(repeat, new OperationChain.Builder().first(new GetAllElements()).build());
final Response jobSchedulingResponse = client.scheduleJob(job);
JobDetail jobSchedulingDetail = jobSchedulingResponse.readEntity(new GenericType<JobDetail>() {
});
// Then
assertEquals(201, jobSchedulingResponse.getStatus());
String parentJobId = jobSchedulingDetail.getJobId();
// Wait for first scheduled to run
Thread.sleep(1500);
final Response getAllJobDetailsResponse = client.executeOperation(new GetAllJobDetails());
List<JobDetail> jobDetails = getAllJobDetailsResponse.readEntity(new GenericType<List<JobDetail>>() {
});
for (JobDetail jobDetail : jobDetails) {
if (null != jobDetail.getParentJobId() && jobDetail.getParentJobId().equals(parentJobId)) {
assertEquals(JobStatus.FINISHED, jobDetail.getStatus());
}
if (jobDetail.getJobId().equals(parentJobId)) {
assertEquals(JobStatus.SCHEDULED_PARENT, jobDetail.getStatus());
}
}
client.executeOperation(new CancelScheduledJob.Builder().jobId(parentJobId).build());
final Response getAllJobDetailsResponseAfterCancelled = client.executeOperation(new GetAllJobDetails());
List<JobDetail> jobDetailsAfterCancelled = getAllJobDetailsResponseAfterCancelled.readEntity(new GenericType<List<JobDetail>>() {
});
for (JobDetail jobDetail : jobDetailsAfterCancelled) {
if (parentJobId.equals(jobDetail.getJobId())) {
assertEquals(JobStatus.CANCELLED, jobDetail.getStatus());
}
}
}
use of uk.gov.gchq.gaffer.jobtracker.Job in project Gaffer by gchq.
the class JobServiceV2IT method shouldNotKeepScheduledJobsRunningAfterRestartWhenUsingInMemoryCache.
@Test
public void shouldNotKeepScheduledJobsRunningAfterRestartWhenUsingInMemoryCache() throws IOException {
// Given - schedule Job
final Repeat repeat = new Repeat(1, 2, TimeUnit.SECONDS);
Job job = new Job(repeat, new OperationChain.Builder().first(new GetAllElements()).build());
final Response scheduleResponse = client.scheduleJob(job);
String parentJobId = scheduleResponse.readEntity(new GenericType<JobDetail>() {
}).getJobId();
// When - get all JobDetails
final Response allJobDetailsResponse = client.executeOperation(new GetAllJobDetails());
List<JobDetail> allJobDetails = allJobDetailsResponse.readEntity(new GenericType<List<JobDetail>>() {
});
// then - assert parent is of Scheduled parent
assertEquals(JobStatus.SCHEDULED_PARENT, allJobDetails.stream().filter(jobDetail -> jobDetail.getJobId().equals(parentJobId)).findFirst().get().getStatus());
// Restart server to check Job still scheduled
client.stopServer();
client.reinitialiseGraph();
// When - get all JobDetails
final Response allJobDetailsResponse2 = client.executeOperation(new GetAllJobDetails());
List<JobDetail> allJobDetails2 = allJobDetailsResponse2.readEntity(new GenericType<List<JobDetail>>() {
});
// Then - assert parent job id is not present
assertTrue(allJobDetails2.stream().noneMatch(jobDetail -> jobDetail.getJobId().equals(parentJobId)));
}
Aggregations