From 4c6ffa9dc5afe7b44a814b5b0f52abb0003972ae Mon Sep 17 00:00:00 2001
From: Christoph Strobl <christoph.strobl@broadcom.com>
Date: Wed, 31 Jan 2024 11:41:49 +0100
Subject: [PATCH] Add count query creation test.

See: #3324
---
 .../data/jpa/repository/query/QueryUtilsUnitTests.java      | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git a/spring-data-jpa/src/test/java/org/springframework/data/jpa/repository/query/QueryUtilsUnitTests.java b/spring-data-jpa/src/test/java/org/springframework/data/jpa/repository/query/QueryUtilsUnitTests.java
index 1bef84f9b..a550ad203 100644
--- a/spring-data-jpa/src/test/java/org/springframework/data/jpa/repository/query/QueryUtilsUnitTests.java
+++ b/spring-data-jpa/src/test/java/org/springframework/data/jpa/repository/query/QueryUtilsUnitTests.java
@@ -937,8 +937,8 @@ class QueryUtilsUnitTests {
 	}
 
 	@Test // GH-3324
-	void createCountQueryForSimpleQuery(){
-		String originalQuery = "select * from User";
-		assertCountQuery(originalQuery,"select count(*) from User");
+	void createCountQueryForSimpleQuery() {
+		assertCountQuery("select * from User","select count(*) from User");
+		assertCountQuery("select * from User u","select count(u) from User u");
 	}
 }