Fix count query creation for HQL when using CTE.

Closes #3504
Original pull request #3508

Author: christophstrobl

spring-data-jpa/src/main/java/org/springframework/data/jpa/repository/query/HqlQueryTransformer.java

@@ -93,7 +93,7 @@ public boolean hasConstructorExpression() {
 	 */
 	private static boolean isSubquery(ParserRuleContext ctx) {
 
-		if (ctx instanceof HqlParser.SubqueryContext) {
+		if (ctx instanceof HqlParser.SubqueryContext || ctx instanceof HqlParser.CteContext) {
 			return true;
 		} else if (ctx instanceof HqlParser.SelectStatementContext) {
 			return false;

spring-data-jpa/src/test/java/org/springframework/data/jpa/repository/query/HqlQueryTransformerTests.java

@@ -500,6 +500,36 @@ select count(user) from User user
 				""");
 	}
 
+	@Test // GH-3504
+	void createCountWithCteShouldWork() {
+
+		String countQuery = createCountQueryFor("""
+				WITH maxId AS(select max(sr.snapshot.id) snapshotId from SnapshotReference sr
+					where sr.id.selectionId = ?1 and sr.enabled
+					group by sr.userId)
+				select sr from maxId m join SnapshotReference sr on sr.snapshot.id = m.snapshotId
+				""");
+
+		assertThat(countQuery).startsWith("WITH maxId AS(select max(sr.snapshot.id) snapshotId from SnapshotReference sr")
+				.endsWith("select count(m) from maxId m join SnapshotReference sr on sr.snapshot.id = m.snapshotId");
+	}
+
+	@Test // GH-3504
+	void createSortedQueryWithCteShouldWork() {
+
+		String sortedQuery = createQueryFor("""
+				WITH maxId AS(select max(sr.snapshot.id) snapshotId from SnapshotReference sr
+					where sr.id.selectionId = ?1 and sr.enabled
+					group by sr.userId)
+				select sr from maxId m join SnapshotReference sr on sr.snapshot.id = m.snapshotId
+				""", Sort.by("sr.snapshot"));
+
+		assertThat(sortedQuery).startsWith(
+				"WITH maxId AS(select max(sr.snapshot.id) snapshotId from SnapshotReference sr where sr.id.selectionId = ?1 and sr.enabled group by sr.userId )")
+				.endsWith(
+						"select sr from maxId m join SnapshotReference sr on sr.snapshot.id = m.snapshotId order by sr.snapshot asc");
+	}
+
 	@Test
 	void createCountQuerySupportsLineBreaksInSelectClause() {