|
21 | 21 | import static org.mockito.Mockito.*;
|
22 | 22 | import static org.springframework.data.jdbc.core.convert.DefaultDataAccessStrategyUnitTests.*;
|
23 | 23 |
|
| 24 | +import java.sql.JDBCType; |
24 | 25 | import java.util.ArrayList;
|
25 | 26 | import java.util.HashMap;
|
26 | 27 | import java.util.List;
|
|
32 | 33 | import org.springframework.data.convert.ReadingConverter;
|
33 | 34 | import org.springframework.data.convert.WritingConverter;
|
34 | 35 | import org.springframework.data.jdbc.core.mapping.JdbcMappingContext;
|
| 36 | +import org.springframework.data.jdbc.core.mapping.JdbcValue; |
35 | 37 | import org.springframework.data.relational.core.conversion.IdValueSource;
|
36 | 38 | import org.springframework.data.relational.core.mapping.Column;
|
37 | 39 | import org.springframework.data.relational.core.mapping.RelationalMappingContext;
|
|
42 | 44 | * Unit tests for {@link SqlParametersFactory}.
|
43 | 45 | *
|
44 | 46 | * @author Chirag Tailor
|
| 47 | + * @author Sergey Korotaev |
45 | 48 | */
|
46 | 49 | class SqlParametersFactoryTest {
|
47 | 50 |
|
@@ -162,6 +165,69 @@ void parameterNamesGetSanitized() {
|
162 | 165 | assertThat(sqlParameterSource.getValue("val&ue")).isNull();
|
163 | 166 | }
|
164 | 167 |
|
| 168 | + @Test |
| 169 | + // GH-1935 |
| 170 | + void enumParameterIsNotNullReturnCorrectSqlTypeFromConverter() { |
| 171 | + |
| 172 | + WithEnumEntity entity = new WithEnumEntity(23L, DummyEnum.ONE); |
| 173 | + |
| 174 | + SqlParametersFactory sqlParametersFactory = createSqlParametersFactoryWithConverters( |
| 175 | + singletonList(WritingEnumConverter.INSTANCE)); |
| 176 | + |
| 177 | + SqlIdentifierParameterSource sqlParameterSource = sqlParametersFactory.forInsert(entity, |
| 178 | + WithEnumEntity.class, Identifier.empty(), IdValueSource.PROVIDED); |
| 179 | + |
| 180 | + assertThat(sqlParameterSource.getValue("id")).isEqualTo(23L); |
| 181 | + assertThat(sqlParameterSource.getValue("dummy_enum")).isEqualTo(DummyEnum.ONE.name()); |
| 182 | + assertThat(sqlParameterSource.getSqlType("dummy_enum")).isEqualTo(1111); |
| 183 | + } |
| 184 | + |
| 185 | + @Test |
| 186 | + // GH-1935 |
| 187 | + void enumParameterIsNullReturnCorrectSqlTypeFromConverter() { |
| 188 | + WithEnumEntity entity = new WithEnumEntity(23L, null); |
| 189 | + |
| 190 | + SqlParametersFactory sqlParametersFactory = createSqlParametersFactoryWithConverters( |
| 191 | + singletonList(WritingEnumConverter.INSTANCE)); |
| 192 | + |
| 193 | + SqlIdentifierParameterSource sqlParameterSource = sqlParametersFactory.forInsert(entity, |
| 194 | + WithEnumEntity.class, Identifier.empty(), IdValueSource.PROVIDED); |
| 195 | + |
| 196 | + assertThat(sqlParameterSource.getValue("id")).isEqualTo(23L); |
| 197 | + assertThat(sqlParameterSource.getSqlType("dummy_enum")).isEqualTo(JDBCType.NULL.getVendorTypeNumber()); |
| 198 | + assertThat(sqlParameterSource.getValue("dummy_enum")).isNull(); |
| 199 | + } |
| 200 | + |
| 201 | + @Test |
| 202 | + // GH-1935 |
| 203 | + void enumParameterIsNotNullReturnCorrectSqlTypeWithoutConverter() { |
| 204 | + |
| 205 | + WithEnumEntity entity = new WithEnumEntity(23L, DummyEnum.ONE); |
| 206 | + |
| 207 | + SqlIdentifierParameterSource sqlParameterSource = sqlParametersFactory.forInsert(entity, |
| 208 | + WithEnumEntity.class, Identifier.empty(), IdValueSource.PROVIDED); |
| 209 | + |
| 210 | + assertThat(sqlParameterSource.getValue("id")).isEqualTo(23L); |
| 211 | + assertThat(sqlParameterSource.getValue("dummy_enum")).isEqualTo(DummyEnum.ONE.name()); |
| 212 | + assertThat(sqlParameterSource.getSqlType("dummy_enum")).isEqualTo(12); |
| 213 | + |
| 214 | + } |
| 215 | + |
| 216 | + @Test |
| 217 | + // GH-1935 |
| 218 | + void enumParameterIsNullReturnCorrectSqlTypeWithoutConverter() { |
| 219 | + |
| 220 | + WithEnumEntity entity = new WithEnumEntity(23L, null); |
| 221 | + |
| 222 | + SqlIdentifierParameterSource sqlParameterSource = sqlParametersFactory.forInsert(entity, |
| 223 | + WithEnumEntity.class, Identifier.empty(), IdValueSource.PROVIDED); |
| 224 | + |
| 225 | + assertThat(sqlParameterSource.getValue("id")).isEqualTo(23L); |
| 226 | + assertThat(sqlParameterSource.getSqlType("dummy_enum")).isEqualTo(JDBCType.NULL.getVendorTypeNumber()); |
| 227 | + assertThat(sqlParameterSource.getValue("dummy_enum")).isNull(); |
| 228 | + |
| 229 | + } |
| 230 | + |
165 | 231 | @WritingConverter
|
166 | 232 | enum IdValueToStringConverter implements Converter<IdValue, String> {
|
167 | 233 |
|
@@ -229,6 +295,17 @@ public String toString() {
|
229 | 295 | }
|
230 | 296 | }
|
231 | 297 |
|
| 298 | + @WritingConverter |
| 299 | + enum WritingEnumConverter implements Converter<DummyEnum, JdbcValue> { |
| 300 | + |
| 301 | + INSTANCE; |
| 302 | + |
| 303 | + @Override |
| 304 | + public JdbcValue convert(DummyEnum source) { |
| 305 | + return JdbcValue.of(source.name().toUpperCase(), JDBCType.OTHER); |
| 306 | + } |
| 307 | + } |
| 308 | + |
232 | 309 | @WritingConverter
|
233 | 310 | enum BooleanToStringConverter implements Converter<Boolean, String> {
|
234 | 311 |
|
@@ -295,6 +372,21 @@ public WithIllegalCharacters(Long id, String value) {
|
295 | 372 | }
|
296 | 373 | }
|
297 | 374 |
|
| 375 | + private static class WithEnumEntity { |
| 376 | + @Id Long id; |
| 377 | + |
| 378 | + DummyEnum dummyEnum; |
| 379 | + |
| 380 | + public WithEnumEntity(Long id, DummyEnum dummyEnum) { |
| 381 | + this.id = id; |
| 382 | + this.dummyEnum = dummyEnum; |
| 383 | + } |
| 384 | + } |
| 385 | + |
| 386 | + private enum DummyEnum { |
| 387 | + ONE, TWO |
| 388 | + } |
| 389 | + |
298 | 390 | private SqlParametersFactory createSqlParametersFactoryWithConverters(List<?> converters) {
|
299 | 391 |
|
300 | 392 | MappingJdbcConverter converter = new MappingJdbcConverter(context, relationResolver,
|
|
0 commit comments