Examples with Value com.google.cloud.datastore.Value used on opensource projects

Example 6 with Value

use of com.google.cloud.datastore.Value in project spring-cloud-gcp by spring-cloud.

the class GqlDatastoreQueryTests method pageableTestPage.

@Test
public 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)).isEqualTo(true);
        Map<String, Value> paramMap = statement.getNamedBindings();
        if (statement.getQueryString().equals(expected)) {
            assertThat(paramMap.size()).isEqualTo(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.size()).isEqualTo(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));
}

Example 7 with Value

use of com.google.cloud.datastore.Value in project spring-cloud-gcp by spring-cloud.

the class GqlDatastoreQueryTests method pageableTest.

@Test
public void pageableTest() {
    String gql = "SELECT * FROM trades WHERE price=@price";
    Object[] paramVals = new Object[] { 1, PageRequest.of(0, 2) };
    String[] paramNames = new String[] { "price", null };
    Parameters parameters = buildParameters(paramVals, paramNames);
    when(parameters.hasPageableParameter()).thenReturn(true);
    when(parameters.getPageableIndex()).thenReturn(1);
    GqlDatastoreQuery gqlDatastoreQuery = createQuery(gql, false, false);
    doAnswer((invocation) -> {
        GqlQuery statement = invocation.getArgument(0);
        assertThat(statement.getQueryString()).isEqualTo("SELECT * FROM trades WHERE price=@price LIMIT @limit OFFSET @offset");
        Map<String, Value> paramMap = statement.getNamedBindings();
        assertThat(paramMap.get("price").get()).isEqualTo(1L);
        assertThat(paramMap.get("limit").get()).isEqualTo(2L);
        assertThat(paramMap.get("offset").get()).isEqualTo(0L);
        return null;
    }).when(this.datastoreTemplate).queryKeysOrEntities(any(), eq(Trade.class));
    doReturn(false).when(gqlDatastoreQuery).isNonEntityReturnedType(any());
    gqlDatastoreQuery.execute(paramVals);
    verify(this.datastoreTemplate, times(1)).queryKeysOrEntities(any(), eq(Trade.class));
}

Example 8 with Value

use of com.google.cloud.datastore.Value in project spring-cloud-gcp by spring-cloud.

the class GqlDatastoreQueryTests method pageableTestPageCursor.

@Test
public void pageableTestPageCursor() {
    String gql = "SELECT * FROM trades WHERE price=@price";
    String expected = "SELECT * FROM trades WHERE price=@price LIMIT @limit OFFSET @offset";
    Cursor cursorInPageable = Cursor.copyFrom("cde".getBytes());
    long countInPageable = 123L;
    Object[] paramVals = new Object[] { 1, new DatastorePageable(PageRequest.of(0, 2), cursorInPageable, countInPageable) };
    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()).isEqualTo(expected);
        Map<String, Object> paramMap = statement.getNamedBindings();
        assertThat(paramMap.size()).isEqualTo(3);
        assertThat(((Value) paramMap.get("price")).get()).isEqualTo(1L);
        assertThat(((Value) paramMap.get("limit")).get()).isEqualTo(2L);
        assertThat(paramMap.get("offset")).isEqualTo(cursorInPageable);
        return new DatastoreResultsIterable(Collections.emptyList(), cursor);
    }).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(countInPageable);
    assertThat(((Page) result).getTotalElements()).isEqualTo(countInPageable);
    verify(this.datastoreTemplate, times(1)).queryKeysOrEntities(any(), eq(Trade.class));
}

Example 9 with Value

use of com.google.cloud.datastore.Value in project spring-cloud-gcp by spring-cloud.

the class GqlDatastoreQueryTests method pageableTestSort.

@Test
public void pageableTestSort() {
    String gql = "SELECT * FROM trades WHERE price=@price";
    Object[] paramVals = new Object[] { 1, Sort.by(Sort.Order.asc("p1"), Sort.Order.desc("p2")) };
    String[] paramNames = new String[] { "price", null };
    Parameters parameters = buildParameters(paramVals, paramNames);
    when(parameters.hasSortParameter()).thenReturn(true);
    when(parameters.getSortIndex()).thenReturn(1);
    GqlDatastoreQuery gqlDatastoreQuery = createQuery(gql, false, false);
    doAnswer((invocation) -> {
        GqlQuery statement = invocation.getArgument(0);
        assertThat(statement.getQueryString()).isEqualTo("SELECT * FROM trades WHERE price=@price ORDER BY p1 ASC, p2 DESC");
        Map<String, Value> paramMap = statement.getNamedBindings();
        assertThat(paramMap.get("price").get()).isEqualTo(1L);
        return null;
    }).when(this.datastoreTemplate).queryKeysOrEntities(any(), eq(Trade.class));
    doReturn(false).when(gqlDatastoreQuery).isNonEntityReturnedType(any());
    gqlDatastoreQuery.execute(paramVals);
    verify(this.datastoreTemplate, times(1)).queryKeysOrEntities(any(), eq(Trade.class));
}

Example 10 with Value

use of com.google.cloud.datastore.Value in project spring-cloud-gcp by spring-cloud.

the class GqlDatastoreQueryTests method pageableTestSlice.

@Test
public void pageableTestSlice() {
    String gql = "SELECT * FROM trades WHERE price=@price";
    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, false, true);
    Cursor cursor = Cursor.copyFrom("abc".getBytes());
    List<Map> params = new ArrayList<>();
    doAnswer((invocation) -> {
        GqlQuery statement = invocation.getArgument(0);
        assertThat(statement.getQueryString()).isEqualTo("SELECT * FROM trades WHERE price=@price LIMIT @limit OFFSET @offset");
        Map paramMap = statement.getNamedBindings();
        params.add(paramMap);
        return new DatastoreResultsIterable(Collections.emptyList(), cursor);
    }).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 = (Slice) gqlDatastoreQuery.execute(paramVals);
    assertThat(((DatastorePageable) result.getPageable()).toCursor()).isEqualTo(cursor);
    verify(this.datastoreTemplate, times(2)).queryKeysOrEntities(any(), eq(Trade.class));
    assertThat(((Value) params.get(0).get("price")).get()).isEqualTo(1L);
    assertThat(((Value) params.get(0).get("limit")).get()).isEqualTo(2L);
    assertThat(((Value) params.get(0).get("offset")).get()).isEqualTo(0L);
    assertThat(((Value) params.get(1).get("price")).get()).isEqualTo(1L);
    assertThat(((Value) params.get(1).get("limit")).get()).isEqualTo(1L);
    assertThat(params.get(1).get("offset")).isEqualTo(cursor);
}