use of com.google.cloud.spring.data.datastore.core.DatastoreResultsIterable in project spring-cloud-gcp by GoogleCloudPlatform.
the class GqlDatastoreQueryTests method streamResultTest.
@Test
void streamResultTest() {
Mockito.<Class>when(this.queryMethod.getReturnedObjectType()).thenReturn(Trade.class);
Parameters parameters = mock(Parameters.class);
when(this.queryMethod.getParameters()).thenReturn(parameters);
when(parameters.getNumberOfParameters()).thenReturn(0);
when(this.queryMethod.isStreamQuery()).thenReturn(true);
Trade tradeA = new Trade();
tradeA.id = "a";
Trade tradeB = new Trade();
tradeB.id = "b";
doAnswer(invocation -> {
GqlQuery statement = invocation.getArgument(0);
assertThat(statement.getQueryString()).isEqualTo("unusedGqlString");
Cursor cursor = Cursor.copyFrom("abc".getBytes());
DatastoreResultsIterable datastoreResultsIterable = new DatastoreResultsIterable(Arrays.asList(tradeA, tradeB), cursor);
return datastoreResultsIterable;
}).when(this.datastoreTemplate).queryKeysOrEntities(any(), eq(Trade.class));
GqlDatastoreQuery gqlDatastoreQuery = createQuery("unusedGqlString", false, false);
Object result = gqlDatastoreQuery.execute(new Parameters[0]);
assertThat(result).isInstanceOf(Stream.class);
assertThat((Stream) result).hasSize(2).containsExactly(tradeA, tradeB);
}
use of com.google.cloud.spring.data.datastore.core.DatastoreResultsIterable in project spring-cloud-gcp by GoogleCloudPlatform.
the class GqlDatastoreQueryTests method pageableTestPage.
@Test
void pageableTestPage() {
String gql = "SELECT * FROM trades WHERE price=@price";
String expected = "SELECT * FROM trades WHERE price=@price LIMIT @limit OFFSET @offset";
Object[] paramVals = new Object[] { 1, PageRequest.of(0, 2) };
String[] paramNames = new String[] { "price", null };
Parameters parameters = buildParameters(paramVals, paramNames);
Mockito.<Class>when(this.queryMethod.getReturnedObjectType()).thenReturn(Trade.class);
when(parameters.hasPageableParameter()).thenReturn(true);
when(parameters.getPageableIndex()).thenReturn(1);
GqlDatastoreQuery gqlDatastoreQuery = createQuery(gql, true, true);
Cursor cursor = Cursor.copyFrom("abc".getBytes());
doAnswer(invocation -> {
GqlQuery statement = invocation.getArgument(0);
assertThat(statement.getQueryString().equals(gql) || statement.getQueryString().equals(expected)).isTrue();
Map<String, Value> paramMap = statement.getNamedBindings();
if (statement.getQueryString().equals(expected)) {
assertThat(paramMap).hasSize(3);
assertThat(paramMap.get("price").get()).isEqualTo(1L);
assertThat(paramMap.get("limit").get()).isEqualTo(2L);
assertThat(paramMap.get("offset").get()).isEqualTo(0L);
return new DatastoreResultsIterable(Collections.emptyList(), cursor);
} else if (statement.getQueryString().equals(gql)) {
assertThat(paramMap).hasSize(1);
assertThat(paramMap.get("price").get()).isEqualTo(1L);
return new DatastoreResultsIterable(Arrays.asList(1L, 2L), cursor);
}
return null;
}).when(this.datastoreTemplate).queryKeysOrEntities(any(), eq(Trade.class));
doReturn(false).when(gqlDatastoreQuery).isNonEntityReturnedType(any());
doAnswer(invocation -> invocation.getArgument(0)).when(gqlDatastoreQuery).processRawObjectForProjection(any());
Slice result = (Page) gqlDatastoreQuery.execute(paramVals);
assertThat(((DatastorePageable) result.getPageable()).toCursor()).isEqualTo(cursor);
assertThat(((DatastorePageable) result.getPageable()).getTotalCount()).isEqualTo(2L);
assertThat(((Page) result).getTotalElements()).isEqualTo(2L);
verify(this.datastoreTemplate, times(2)).queryKeysOrEntities(any(), eq(Trade.class));
}
use of com.google.cloud.spring.data.datastore.core.DatastoreResultsIterable in project spring-cloud-gcp by GoogleCloudPlatform.
the class PartTreeDatastoreQueryTests method prepareSliceResults.
private void prepareSliceResults(int offset, Integer queryLimit, Boolean hasNext) {
Cursor cursor = Cursor.copyFrom("abc".getBytes());
List<Integer> datastoreMatchingRecords = Arrays.asList(3, 4, 5);
when(this.datastoreTemplate.queryEntitiesSlice(isA(EntityQuery.class), any(), any())).thenAnswer(invocation -> {
EntityQuery statement = invocation.getArgument(0);
EntityQuery expected = StructuredQuery.newEntityQueryBuilder().setFilter(FILTER).setKind("trades").setOffset(offset).setOrderBy(OrderBy.desc("__key__")).setLimit(queryLimit).build();
assertThat(statement).isEqualTo(expected);
return new SliceImpl(new DatastoreResultsIterable(datastoreMatchingRecords.iterator(), cursor).toList(), Pageable.unpaged(), hasNext);
});
}
use of com.google.cloud.spring.data.datastore.core.DatastoreResultsIterable in project spring-cloud-gcp by GoogleCloudPlatform.
the class PartTreeDatastoreQueryTests method nonCollectionReturnType.
@Test
void nonCollectionReturnType() throws NoSuchMethodException {
Trade trade = new Trade();
queryWithMockResult("findByAction", null, getClass().getMethod("findByAction", String.class), true, null);
Object[] params = new Object[] { "BUY" };
when(this.datastoreTemplate.queryKeysOrEntities(any(), any())).thenAnswer(invocation -> {
EntityQuery statement = invocation.getArgument(0);
EntityQuery expected = StructuredQuery.newEntityQueryBuilder().setFilter(PropertyFilter.eq("action", "BUY")).setKind("trades").setLimit(1).build();
assertThat(statement).isEqualTo(expected);
List<Trade> results = Collections.singletonList(trade);
return new DatastoreResultsIterable(results.iterator(), null);
});
assertThat(this.partTreeDatastoreQuery.execute(params)).isEqualTo(trade);
}
use of com.google.cloud.spring.data.datastore.core.DatastoreResultsIterable in project spring-cloud-gcp by GoogleCloudPlatform.
the class SimpleDatastoreRepositoryTests method findByExampleFluentQueryOneValue.
@Test
void findByExampleFluentQueryOneValue() {
Example<Object> example = Example.of(new Object());
Iterable entities = Arrays.asList();
doAnswer(invocationOnMock -> new DatastoreResultsIterable(entities, null)).when(this.datastoreTemplate).queryByExample(same(example), any());
this.spyRepo.findBy(example, FetchableFluentQuery::oneValue);
verify(this.spyRepo).findOne(same(example));
}
Aggregations