Examples with Store uk.gov.gchq.gaffer.store.Store used on opensource projects

Example 26 with Store

use of uk.gov.gchq.gaffer.store.Store in project Gaffer by gchq.

the class GraphTest method shouldExposeGetTraitsMethod.

@Test
public void shouldExposeGetTraitsMethod() throws OperationException {
    // Given
    final Store store = mock(Store.class);
    final View view = mock(View.class);
    final Graph graph = new Graph.Builder().store(store).view(view).build();
    // When
    final Set<StoreTrait> storeTraits = new HashSet<>(Arrays.asList(StoreTrait.STORE_AGGREGATION, StoreTrait.TRANSFORMATION));
    given(store.getTraits()).willReturn(storeTraits);
    final Collection<StoreTrait> returnedTraits = graph.getStoreTraits();
    // Then
    assertEquals(returnedTraits, storeTraits);
}

Example 27 with Store

use of uk.gov.gchq.gaffer.store.Store in project Gaffer by gchq.

the class ExportToGafferResultCacheHandlerTest method shouldHandleOperationByDelegatingToAnExistingExporter.

@Test
public void shouldHandleOperationByDelegatingToAnExistingExporter() throws OperationException {
    // Given
    final List<?> results = Arrays.asList(1, 2, 3);
    final ExportToGafferResultCache export = new ExportToGafferResultCache.Builder().key("key").input(results).build();
    final Context context = new Context();
    final Store store = mock(Store.class);
    final JSONSerialiser jsonSerialiser = mock(JSONSerialiser.class);
    final Long timeToLive = 10000L;
    final String visibility = "visibility value";
    final GafferResultCacheExporter exporter = mock(GafferResultCacheExporter.class);
    context.addExporter(exporter);
    final ExportToGafferResultCacheHandler handler = new ExportToGafferResultCacheHandler();
    handler.setStorePropertiesPath(StreamUtil.STORE_PROPERTIES);
    handler.setJsonSerialiser(jsonSerialiser);
    handler.setTimeToLive(timeToLive);
    handler.setVisibility(visibility);
    // When
    final Object handlerResult = handler.doOperation(export, context, store);
    // Then
    verify(exporter).add("key", results);
    assertSame(handlerResult, results);
}

Example 28 with Store

use of uk.gov.gchq.gaffer.store.Store in project Gaffer by gchq.

the class GenerateElementsHandlerTest method shouldReturnElements.

@Test
public void shouldReturnElements() throws OperationException {
    // Given
    final GenerateElementsHandler<String> handler = new GenerateElementsHandler<>();
    final Store store = mock(Store.class);
    final GenerateElements<String> operation = mock(GenerateElements.class);
    final CloseableIterable<Element> elements = mock(CloseableIterable.class);
    final ElementGenerator<String> elementGenerator = mock(ElementGenerator.class);
    final CloseableIterable<String> objs = mock(CloseableIterable.class);
    final Context context = new Context();
    final CloseableIterator<Element> elementsIter = mock(CloseableIterator.class);
    given(elements.iterator()).willReturn(elementsIter);
    given(elementGenerator.getElements(objs)).willReturn(elements);
    given(operation.getObjects()).willReturn(objs);
    given(operation.getElementGenerator()).willReturn(elementGenerator);
    // When
    final CloseableIterable<Element> result = handler.doOperation(operation, context, store);
    // Then
    assertSame(elementsIter, result.iterator());
}

Example 29 with Store

use of uk.gov.gchq.gaffer.store.Store in project Gaffer by gchq.

the class GenerateObjectsHandlerTest method shouldReturnObjects.

@Test
public void shouldReturnObjects() throws OperationException {
    // Given
    final GenerateObjectsHandler<String> handler = new GenerateObjectsHandler<>();
    final Store store = mock(Store.class);
    final GenerateObjects<Element, String> operation = mock(GenerateObjects.class);
    final CloseableIterable<Element> elements = mock(CloseableIterable.class);
    final ElementGenerator<String> elementGenerator = mock(ElementGenerator.class);
    final CloseableIterable<String> objs = mock(CloseableIterable.class);
    final Context context = new Context();
    final CloseableIterator<String> objsIter = mock(CloseableIterator.class);
    given(objs.iterator()).willReturn(objsIter);
    given(elementGenerator.getObjects(elements)).willReturn(objs);
    given(operation.getElements()).willReturn(elements);
    given(operation.getElementGenerator()).willReturn(elementGenerator);
    // When
    final CloseableIterable<String> result = handler.doOperation(operation, context, store);
    // Then
    assertSame(objsIter, result.iterator());
}

Example 30 with Store

use of uk.gov.gchq.gaffer.store.Store 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);
}