diff --git a/persistence-modules/spring-jooq/src/test/java/com/baeldung/jooq/introduction/CountQueryIntegrationTest.java b/persistence-modules/spring-jooq/src/test/java/com/baeldung/jooq/introduction/CountQueryIntegrationTest.java index 7dfa35eb73..4a21459d19 100644 --- a/persistence-modules/spring-jooq/src/test/java/com/baeldung/jooq/introduction/CountQueryIntegrationTest.java +++ b/persistence-modules/spring-jooq/src/test/java/com/baeldung/jooq/introduction/CountQueryIntegrationTest.java @@ -2,7 +2,13 @@ package com.baeldung.jooq.introduction; import static com.baeldung.jooq.introduction.db.public_.tables.Author.AUTHOR; +import java.util.ArrayList; +import java.util.List; + +import org.jooq.Condition; import org.jooq.DSLContext; +import org.jooq.Record2; +import org.jooq.Result; import org.jooq.impl.DSL; import org.junit.Assert; import org.junit.Test; @@ -22,8 +28,7 @@ public class CountQueryIntegrationTest { @Test public void givenValidData_whenSimpleSelect_thenSucceed() { - int count = dsl.select().from(AUTHOR) - .execute(); + int count = dsl.select().from(AUTHOR).execute(); Assert.assertEquals(3, count); } @@ -32,20 +37,62 @@ public class CountQueryIntegrationTest { int count = dsl.selectCount().from(AUTHOR) .where(AUTHOR.FIRST_NAME.equalIgnoreCase("Bryan")) .fetchOne(0, int.class); - Assert.assertEquals(3, count); + Assert.assertEquals(1, count); } @Test public void givenValidData_whenCount_thenSucceed() { - int count = dsl.select(DSL.count()).from(AUTHOR) - .fetchOne(0, int.class); - Assert.assertEquals(1, count); + int count = dsl.select(DSL.count()) + .from(AUTHOR).fetchOne(0, int.class); + Assert.assertEquals(3, count); } @Test public void givenValidData_whenFetchCount_thenSucceed() { - int count = dsl.fetchCount(DSL.selectFrom(AUTHOR) + int count = dsl.fetchCount( + DSL.selectFrom(AUTHOR) .where(AUTHOR.FIRST_NAME.equalIgnoreCase("Bryan"))); Assert.assertEquals(1, count); } + + @Test + public void givenValidData_whenFetchCountWithoutCondition_thenSucceed() { + int count = dsl.fetchCount( + DSL.selectFrom(AUTHOR)); + Assert.assertEquals(3, count); + } + + @Test + public void givenValidData_whenFetchCountWithSingleCondition_thenSucceed() { + int count = dsl.fetchCount(AUTHOR, AUTHOR.FIRST_NAME.equalIgnoreCase("Bryan")); + Assert.assertEquals(1, count); + } + + @Test + public void givenValidData_whenFetchCountWithMultipleConditions_thenSucceed() { + Condition firstCond = AUTHOR.FIRST_NAME.equalIgnoreCase("Bryan"); + Condition secondCond = AUTHOR.ID.notEqual(1); + List conditions = new ArrayList<>(); + conditions.add(firstCond); + conditions.add(secondCond); + int count = dsl.fetchCount(AUTHOR, conditions); + Assert.assertEquals(1, count); + } + + @Test + public void givenValidData_whenFetchCountWithConditionsInVarargs_thenSucceed() { + Condition firstCond = AUTHOR.FIRST_NAME.equalIgnoreCase("Bryan"); + Condition secondCond = AUTHOR.ID.notEqual(1); + int count = dsl.fetchCount(AUTHOR, firstCond, secondCond); + Assert.assertEquals(1, count); + } + + @Test + public void givenValidData_whenCountwithGroupBy_thenSucceed() { + final Result> result = dsl.select(AUTHOR.FIRST_NAME, DSL.count()) + .from(AUTHOR).groupBy(AUTHOR.FIRST_NAME).fetch(); + Assert.assertEquals(3, result.size()); + Assert.assertEquals(result.get(0).get(0), "Bert"); + Assert.assertEquals(result.get(0).get(1), 1); + } } \ No newline at end of file