use of uk.gov.gchq.gaffer.operation.impl.job.GetJobDetails in project Gaffer by gchq.
the class GetJobDetailsHandlerTest method shouldGetJobDetailsByDelegatingToJobTrackerWithOperationJobId.
@Test
public void shouldGetJobDetailsByDelegatingToJobTrackerWithOperationJobId() throws OperationException {
// Given
final String jobId = "jobId";
final GetJobDetailsHandler handler = new GetJobDetailsHandler();
final GetJobDetails operation = new GetJobDetails.Builder().jobId(jobId).build();
final Store store = mock(Store.class);
final JobTracker jobTracker = mock(JobTracker.class);
final User user = mock(User.class);
final JobDetail jobsDetail = mock(JobDetail.class);
final Context context = new Context(user);
given(store.getJobTracker()).willReturn(jobTracker);
given(jobTracker.getJob(jobId, user)).willReturn(jobsDetail);
// When
final JobDetail result = handler.doOperation(operation, context, store);
// Then
assertSame(jobsDetail, result);
}
use of uk.gov.gchq.gaffer.operation.impl.job.GetJobDetails in project Gaffer by gchq.
the class OperationChainTest method shouldBuildOperationChain.
@Test
public void shouldBuildOperationChain() {
// Given
final AddElements addElements1 = mock(AddElements.class);
final AddElements addElements2 = mock(AddElements.class);
final GetAdjacentIds getAdj1 = mock(GetAdjacentIds.class);
final GetAdjacentIds getAdj2 = mock(GetAdjacentIds.class);
final GetAdjacentIds getAdj3 = mock(GetAdjacentIds.class);
final GetElements getElements1 = mock(GetElements.class);
final GetElements getElements2 = mock(GetElements.class);
final GetAllElements getAllElements = mock(GetAllElements.class);
final DiscardOutput discardOutput = mock(DiscardOutput.class);
final GetJobDetails getJobDetails = mock(GetJobDetails.class);
final GenerateObjects<EntityId> generateEntitySeeds = mock(GenerateObjects.class);
final Limit<Element> limit = mock(Limit.class);
final ToSet<Element> deduplicate = mock(ToSet.class);
final CountGroups countGroups = mock(CountGroups.class);
final ExportToSet<GroupCounts> exportToSet = mock(ExportToSet.class);
final ExportToGafferResultCache<CloseableIterable<? extends Element>> exportToGafferCache = mock(ExportToGafferResultCache.class);
final If<Iterable<? extends EntityId>, Iterable<? extends EntityId>> ifOp = mock(If.class);
// When
final OperationChain<JobDetail> opChain = new Builder().first(addElements1).then(getAdj1).then(getAdj2).then(getElements1).then(generateEntitySeeds).then(getAdj3).then(ifOp).then(getElements2).then(deduplicate).then(limit).then(countGroups).then(exportToSet).then(discardOutput).then(getAllElements).then(exportToGafferCache).then(addElements2).then(getJobDetails).build();
// Then
final Operation[] expecteds = { addElements1, getAdj1, getAdj2, getElements1, generateEntitySeeds, getAdj3, ifOp, getElements2, deduplicate, limit, countGroups, exportToSet, discardOutput, getAllElements, exportToGafferCache, addElements2, getJobDetails };
assertArrayEquals(expecteds, opChain.getOperationArray());
}
Aggregations