Render entire statement after applying sort.

This commit makes sure to render the entire statement after applying the sort expression via the QueryEnhancer. Previously only parts, the actual statement body, had been considered.

Closes: #3263
Original Pull Request: #3264

Author: christophstrobl

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

@@ -141,7 +141,7 @@ public String applySorting(Sort sort, @Nullable String alias) {
 
 		selectBody.getOrderByElements().addAll(orderByElements);
 
-		return selectBody.toString();
+		return selectStatement.toString();
 
 	}
 

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

@@ -179,6 +179,17 @@ void findsExistingOrderByIndependentOfCase() {
 		assertThat(query).endsWithIgnoringCase("ORDER BY p.firstname, p.lastname asc");
 	}
 
+	@Test // GH-3263
+	void preserveSourceQueryWhenAddingSort() {
+
+		StringQuery query = new StringQuery("WITH all_projects AS (SELECT * FROM projects) SELECT * FROM all_projects p",
+				true);
+
+		assertThat(getEnhancer(query).applySorting(Sort.by("name"), "p")) //
+				.startsWithIgnoringCase(query.getQueryString())
+				.endsWithIgnoringCase("ORDER BY p.name ASC");
+	}
+
 	@Test // GH-2812
 	void createCountQueryFromDeleteQuery() {