Add test

querydsl · May 15, 2016 · b7f45b8 · b7f45b8
1 parent e98ac3a
commit b7f45b8
Show file tree

Hide file tree

Showing 2 changed files with 25 additions and 0 deletions.
diff --git a/querydsl-jpa/src/test/java/com/querydsl/jpa/AbstractJPATest.java b/querydsl-jpa/src/test/java/com/querydsl/jpa/AbstractJPATest.java
@@ -646,6 +646,14 @@ public void distinct_orderBy2() {
         ));
     }
 
+    @Test
+    @NoEclipseLink(HSQLDB)
+    public void count_distinct3() {
+        QCat kitten = new QCat("kitten");
+        assertEquals(4, query().from(cat).leftJoin(cat.kittens, kitten).select(kitten.countDistinct()).fetchOne().intValue());
+        assertEquals(6, query().from(cat).leftJoin(cat.kittens, kitten).select(kitten.countDistinct()).fetchCount());
+    }
+
     @Test
     public void distinctResults() {
         System.out.println("-- fetch results");

diff --git a/querydsl-jpa/src/test/java/com/querydsl/jpa/JPQLSerializerTest.java b/querydsl-jpa/src/test/java/com/querydsl/jpa/JPQLSerializerTest.java
@@ -93,6 +93,23 @@ public void case2_hibernate() {
         assertEquals("case when (cat.toes = ?1) then (cat.id * ?1) when (cat.toes = ?2) then (cat.id * ?2) else 4 end", serializer.toString());
     }
 
+    @Test
+    public void count() {
+        QCat cat = QCat.cat;
+        QueryMetadata md = new DefaultQueryMetadata();
+        md.addJoin(JoinType.DEFAULT, cat);
+        md.setProjection(cat.mate.countDistinct());
+        JPQLSerializer serializer1 = new JPQLSerializer(HQLTemplates.DEFAULT);
+        serializer1.serialize(md, true, null);
+        assertEquals("select count(count(distinct cat.mate))\n" +
+                "from Cat cat",  serializer1.toString());
+
+        JPQLSerializer serializer2 = new JPQLSerializer(HQLTemplates.DEFAULT);
+        serializer2.serialize(md, false, null);
+        assertEquals("select count(distinct cat.mate)\n" +
+                "from Cat cat", serializer2.toString());
+    }
+
     @Test
     public void fromWithCustomEntityName() {
         JPQLSerializer serializer = new JPQLSerializer(HQLTemplates.DEFAULT);