Examples with AggregateQuery org.molgenis.data.aggregation.AggregateQuery used on opensource projects

Example 16 with AggregateQuery

use of org.molgenis.data.aggregation.AggregateQuery in project molgenis by molgenis.

the class IndexedRepositoryDecoratorTest method aggregateUnknownIndexExceptionUnrecoverable.

@Test(expectedExceptions = MolgenisDataException.class, expectedExceptionsMessageRegExp = "Error executing query, index for entity type 'My entity type' with id 'entity' does not exist")
public void aggregateUnknownIndexExceptionUnrecoverable() {
    AggregateQuery aggregateQuery = mock(AggregateQuery.class);
    when(searchService.aggregate(repositoryEntityType, aggregateQuery)).thenThrow(new UnknownIndexException("msg"));
    indexedRepositoryDecorator.aggregate(aggregateQuery);
}

Example 17 with AggregateQuery

use of org.molgenis.data.aggregation.AggregateQuery in project molgenis by molgenis.

the class DataServiceIT method testAggregateOneDimensionalDistinct.

@WithMockUser(username = USERNAME_READ)
@Test(groups = "readtest")
public void testAggregateOneDimensionalDistinct() {
    AggregateQuery aggregateQuery = new AggregateQueryImpl().query(new QueryImpl<>()).attrX(entityType.getAttribute(ATTR_BOOL)).attrDistinct(entityType.getAttribute(ATTR_ENUM));
    AggregateResult result = dataService.aggregate(entityType.getId(), aggregateQuery);
    AggregateResult expectedResult = new AggregateResult(asList(singletonList(1L), singletonList(1L)), asList(0L, 1L), emptyList());
    assertEquals(result, expectedResult);
}

Example 18 with AggregateQuery

use of org.molgenis.data.aggregation.AggregateQuery in project molgenis by molgenis.

the class DataServiceIT method testAggregateTwoDimensional.

@WithMockUser(username = USERNAME_READ)
@Test(groups = "readtest")
public void testAggregateTwoDimensional() {
    AggregateQuery aggregateQuery = new AggregateQueryImpl().query(new QueryImpl<>()).attrX(entityType.getAttribute(ATTR_BOOL)).attrY(entityType.getAttribute(ATTR_ENUM));
    AggregateResult result = dataService.aggregate(entityType.getId(), aggregateQuery);
    AggregateResult expectedResult = new AggregateResult(asList(asList(0L, 1L), asList(2L, 0L)), asList(0L, 1L), asList("option1", "option2"));
    assertEquals(result, expectedResult);
}

Example 19 with AggregateQuery

use of org.molgenis.data.aggregation.AggregateQuery in project molgenis by molgenis.

the class DataServiceIT method testAggregateOneDimensional.

@WithMockUser(username = USERNAME_READ)
@Test(groups = "readtest")
public void testAggregateOneDimensional() {
    AggregateQuery aggregateQuery = new AggregateQueryImpl().query(new QueryImpl<>()).attrX(entityType.getAttribute(ATTR_BOOL));
    AggregateResult result = dataService.aggregate(entityType.getId(), aggregateQuery);
    AggregateResult expectedResult = new AggregateResult(asList(singletonList(1L), singletonList(2L)), asList(0L, 1L), emptyList());
    assertEquals(result, expectedResult);
}

Example 20 with AggregateQuery

use of org.molgenis.data.aggregation.AggregateQuery in project molgenis by molgenis.

the class IndexedRepositoryDecoratorTest method aggregate.

@Test
public void aggregate() {
    when(indexedRepositoryDecorator.getName()).thenReturn("entity");
    Attribute xAttr = when(mock(Attribute.class).getName()).thenReturn("xAttr").getMock();
    Attribute yAttr = when(mock(Attribute.class).getName()).thenReturn("yAttr").getMock();
    Attribute distinctAttr = when(mock(Attribute.class).getName()).thenReturn("distinctAttr").getMock();
    @SuppressWarnings("unchecked") Query<Entity> q = mock(Query.class);
    AggregateQuery aggregateQuery = new AggregateQueryImpl().attrX(xAttr).attrY(yAttr).attrDistinct(distinctAttr).query(q);
    indexedRepositoryDecorator.aggregate(aggregateQuery);
    verify(searchService).aggregate(repositoryEntityType, aggregateQuery);
}