CodeCow/querydsl
1
0
Fork 0
mirror of https://github.com/querydsl/querydsl.git synced 2026-07-03 21:07:49 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

test: added test fetchResults with factory expression and groupBy & having clause

Browse Source
This commit is contained in:
heesuk 2020-06-26 13:32:33 +09:00
parent 90b6fe90f4
commit 2d3a844120
1 changed files with 70 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

75
querydsl-jpa/src/test/java/com/querydsl/jpa/JPASQLBase.java
View File

@ -13,7 +13,9 @@
*/
package com.querydsl.jpa;
import static org.junit.Assert.assertEquals;
import java.util.Arrays;
import java.util.HashMap;
import java.util.List;
import javax.persistence.EntityManager;
import javax.persistence.Query;
@ -27,6 +29,8 @@ import org.junit.runner.RunWith;
import com.querydsl.core.Target;
import com.querydsl.core.testutil.ExcludeIn;
import com.querydsl.core.types.Expression;
import com.querydsl.core.types.Projections;
import com.querydsl.jpa.domain.Cat;
import com.querydsl.jpa.domain.Color;
import com.querydsl.jpa.domain.QCat;
@ -35,6 +39,8 @@ import com.querydsl.jpa.sql.JPASQLQuery;
import com.querydsl.jpa.testutil.JPATestRunner;
import com.querydsl.sql.SQLTemplates;
import static org.junit.Assert.assertEquals;
@RunWith(JPATestRunner.class)
public class JPASQLBase extends AbstractSQLTest implements JPATest {
@ -50,6 +56,9 @@ public class JPASQLBase extends AbstractSQLTest implements JPATest {
private EntityManager entityManager;
private final SAnimal cat = new SAnimal("cat");
private final QCat catEntity = QCat.cat;
@Override
protected JPASQLQuery<?> query() {
return new JPASQLQuery<Void>(entityManager, templates);
@ -73,11 +82,22 @@ public class JPASQLBase extends AbstractSQLTest implements JPATest {
}
}
private <T> void insertEntitiesForTest(final List<T> entities) {
for (T entity: entities) {
entityManager.persist(entity);
}
entityManager.flush();
}
private <T> void removeEntitiesForTest(final List<T> entities) {
for (T entity: entities) {
entityManager.remove(entity);
}
entityManager.flush();
}
@Test
public void entityQueries_createQuery() {
SAnimal cat = new SAnimal("cat");
QCat catEntity = QCat.cat;
Query query = query().from(cat).select(catEntity).createQuery();
assertEquals(6, query.getResultList().size());
}
@ -86,10 +106,55 @@ public class JPASQLBase extends AbstractSQLTest implements JPATest {
@ExcludeIn(Target.MYSQL)
public void entityQueries_createQuery2() {
SAnimal cat = new SAnimal("CAT");
QCat catEntity = QCat.cat;
Query query = query().from(cat).select(catEntity).createQuery();
assertEquals(6, query.getResultList().size());
}
@Test
public void should_fetch_results_with_factory_expression() {
final long expectedTotalResultCount = 6;
final HashMap<String, Expression<?>> bindings = new HashMap<String, Expression<?>>();
bindings.put("name", cat.name);
final long actualTotalResultCount = query().from(cat)
.select(Projections.bean(Cat.class, bindings))
.fetchResults()
.getTotal();
assertEquals(expectedTotalResultCount, actualTotalResultCount);
}
@Test
public void should_get_grouped_list_by_using_fetch_results() {
final long expectedCatColorKindCount = 1;
final long actualCatColorKindCount = query().from(cat)
.select(catEntity.color)
.groupBy(catEntity.color)
.fetchResults()
.getTotal();
assertEquals(expectedCatColorKindCount, actualCatColorKindCount);
}
@Test
public void should_get_black_cat_count_by_using_group_by_and_having() {
final long expectedTabbyCatCount = 2L;
final Cat tabbyColorCatFoo = new Cat("Foo", 7, Color.TABBY);
final Cat tabbyColorCatBar = new Cat("Bar", 8, Color.TABBY);
insertEntitiesForTest(Arrays.asList(tabbyColorCatFoo, tabbyColorCatBar));
final long actualTabbyCatCount = query().from(cat)
.select(catEntity.name.count())
.groupBy(catEntity.color)
.having(catEntity.name.count().eq(2L))
.fetchOne();
removeEntitiesForTest(Arrays.asList(tabbyColorCatFoo, tabbyColorCatBar));
assertEquals(expectedTabbyCatCount, actualTabbyCatCount);
}
}