Skip to content

Commit 68d799f

Browse files
committed
Add tests for getting queryProvider result names
1 parent d305dde commit 68d799f

File tree

1 file changed

+19
-1
lines changed

1 file changed

+19
-1
lines changed

spring-batch-infrastructure/src/test/java/org/springframework/batch/item/database/support/AbstractSqlPagingQueryProviderTests.java

Lines changed: 19 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -103,7 +103,25 @@ public void testGenerateJumpToItemQueryForFirstPageWithMultipleSortKeys() {
103103
String s = pagingQueryProvider.generateJumpToItemQuery(45, pageSize);
104104
assertEquals(getJumpToItemQueryForFirstPageWithMultipleSortKeys(), s);
105105
}
106-
106+
107+
@Test
108+
public void testGetSortKeyResultNameWithoutMap() {
109+
assertEquals("id", pagingQueryProvider.getSortKeyResultName("id"));
110+
assertEquals("column", pagingQueryProvider.getSortKeyResultName("table.column"));
111+
assertEquals("column", pagingQueryProvider.getSortKeyResultName("schema.table.column"));
112+
}
113+
114+
@Test
115+
public void testGetSortKeyResultNameWithMap() {
116+
Map<String, String> sortKeyResultNames = new LinkedHashMap<>();
117+
sortKeyResultNames.put("foo.id", "foo_id");
118+
pagingQueryProvider.setSortKeyResultNames(sortKeyResultNames);
119+
120+
assertEquals("foo_id", pagingQueryProvider.getSortKeyResultName("foo.id"));
121+
assertEquals("column", pagingQueryProvider.getSortKeyResultName("column"));
122+
assertEquals("id", pagingQueryProvider.getSortKeyResultName("bar.id"));
123+
}
124+
107125
@Test
108126
public abstract void testGenerateFirstPageQuery();
109127

0 commit comments

Comments
 (0)