OGM-1542 Distinct count tests

NathanQingyangXu · Dec 14, 2018 · a8567d5 · a8567d5
1 parent 896f5f8
commit a8567d5
Showing 1 changed file with 24 additions and 0 deletions.
diff --git a/core/src/test/java/org/hibernate/ogm/backendtck/queries/AggregateOperationQueryTest.java b/core/src/test/java/org/hibernate/ogm/backendtck/queries/AggregateOperationQueryTest.java
@@ -52,6 +52,30 @@ public void shouldCountEntities() {
 		} );
 	}
 
+	@Test
+	public void shouldCountByName() {
+		inTransaction( em -> {
+			Number result = (Number) em.createQuery( "SELECT COUNT(author.name) FROM Author author" ).getSingleResult();
+			assertThat( result.intValue() ).isEqualTo( 4 );
+		} );
+	}
+
+	@Test
+	public void shouldCountDistinctByName() {
+		inTransaction( em -> {
+			Number result = (Number) em.createQuery( "SELECT COUNT(DISTINCT author.name) FROM Author author" ).getSingleResult();
+			assertThat( result.intValue() ).isEqualTo( 2 );
+		} );
+	}
+
+	@Test
+	public void shouldCountDistinctWithNullValues() {
+		inTransaction( em -> {
+			Number result = (Number) em.createQuery( "SELECT COUNT(DISTINCT author.age) FROM Author author" ).getSingleResult();
+			assertThat( result.intValue() ).isEqualTo( 3 );
+		} );
+	}
+
 	@Test
 	public void shouldCountEntitiesWithCondition() {
 		inTransaction( em -> {