Polishing.

See #2514.
This commit is contained in:
Greg L. Turnquist
2022-05-04 10:31:48 -05:00
parent e4e153d273
commit 1f2e4a179a
2 changed files with 6 additions and 6 deletions

View File

@@ -712,26 +712,25 @@ class QueryEnhancerUnitTests {
assertThat(result).containsIgnoringCase("order by dd.institutesIds");
}
@Test //GH-2511
@Test // GH-2511
void countQueryUsesCorrectVariable() {
StringQuery nativeQuery = new StringQuery("SELECT * FROM User WHERE created_at > $1", true);
QueryEnhancer queryEnhancer = getEnhancer(nativeQuery);
String countQueryFor = queryEnhancer.createCountQueryFor();
assertThat(countQueryFor).isEqualTo("SELECT count(*) FROM User WHERE created_at > $1");
nativeQuery = new StringQuery("SELECT * FROM (select * from test) ",true);
nativeQuery = new StringQuery("SELECT * FROM (select * from test) ", true);
queryEnhancer = getEnhancer(nativeQuery);
countQueryFor = queryEnhancer.createCountQueryFor();
assertThat(countQueryFor).isEqualTo("SELECT count(*) FROM (SELECT * FROM test)");
nativeQuery = new StringQuery("SELECT * FROM (select * from test) as test",true);
nativeQuery = new StringQuery("SELECT * FROM (select * from test) as test", true);
queryEnhancer = getEnhancer(nativeQuery);
countQueryFor = queryEnhancer.createCountQueryFor();
assertThat(countQueryFor).isEqualTo("SELECT count(test) FROM (SELECT * FROM test) AS test");
}
public static Stream<Arguments> detectsJoinAliasesCorrectlySource() {
return Stream.of( //

View File

@@ -638,8 +638,9 @@ class QueryUtilsUnitTests {
"select * from (select * from user order by 1, 2, 3 desc limit 10) u order by u.active asc, age desc");
}
@Test //GH-2511
@Test // GH-2511
void countQueryUsesCorrectVariable() {
String countQueryFor = createCountQueryFor("SELECT * FROM User WHERE created_at > $1");
assertThat(countQueryFor).isEqualTo("select count(*) FROM User WHERE created_at > $1");