Supports ConversionService on SingleColumnRowMapper

Issue: SPR-16483

spring-jdbc/src/test/java/org/springframework/jdbc/core/SingleColumnRowMapperTest.java

@@ -0,0 +1,107 @@
+/*
+ * Copyright 2002-2018 the original author or authors.
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *      http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+package org.springframework.jdbc.core;
+
+import java.sql.ResultSet;
+import java.sql.ResultSetMetaData;
+import java.sql.SQLException;
+import java.sql.SQLFeatureNotSupportedException;
+import java.sql.Timestamp;
+import java.time.LocalDateTime;
+
+import org.junit.Test;
+import org.springframework.core.convert.support.DefaultConversionService;
+import org.springframework.dao.TypeMismatchDataAccessException;
+
+import static org.mockito.BDDMockito.*;
+import static org.junit.Assert.*;
+
+/**
+ * Tests for {@link SingleColumnRowMapper}.
+ *
+ * @author Kazuki Shimizu
+ * @since 5.0.4
+ */
+public class SingleColumnRowMapperTest {
+
+	@Test // SPR-16483
+	public void useDefaultConversionService() throws SQLException {
+		Timestamp timestamp = new Timestamp(0);
+
+		SingleColumnRowMapper<LocalDateTime> rowMapper = SingleColumnRowMapper.newInstance(LocalDateTime.class);
+
+		ResultSet resultSet = mock(ResultSet.class);
+		ResultSetMetaData metaData = mock(ResultSetMetaData.class);
+		given(metaData.getColumnCount()).willReturn(1);
+		given(resultSet.getMetaData()).willReturn(metaData);
+		given(resultSet.getObject(1, LocalDateTime.class))
+				.willThrow(new SQLFeatureNotSupportedException());
+		given(resultSet.getTimestamp(1)).willReturn(timestamp);
+
+		LocalDateTime actualLocalDateTime = rowMapper.mapRow(resultSet, 1);
+
+		assertEquals(timestamp.toLocalDateTime(), actualLocalDateTime);
+	}
+
+	@Test // SPR-16483
+	public void useCustomConversionService() throws SQLException {
+		Timestamp timestamp = new Timestamp(0);
+
+		DefaultConversionService myConversionService = new DefaultConversionService();
+		myConversionService.addConverter(Timestamp.class, MyLocalDateTime.class,
+				source -> new MyLocalDateTime(source.toLocalDateTime()));
+		SingleColumnRowMapper<MyLocalDateTime> rowMapper =
+				SingleColumnRowMapper.newInstance(MyLocalDateTime.class, myConversionService);
+
+		ResultSet resultSet = mock(ResultSet.class);
+		ResultSetMetaData metaData = mock(ResultSetMetaData.class);
+		given(metaData.getColumnCount()).willReturn(1);
+		given(resultSet.getMetaData()).willReturn(metaData);
+		given(resultSet.getObject(1, MyLocalDateTime.class))
+				.willThrow(new SQLFeatureNotSupportedException());
+		given(resultSet.getObject(1)).willReturn(timestamp);
+
+		MyLocalDateTime actualMyLocalDateTime = rowMapper.mapRow(resultSet, 1);
+
+		assertNotNull(actualMyLocalDateTime);
+		assertEquals(timestamp.toLocalDateTime(), actualMyLocalDateTime.value);
+	}
+
+	@Test(expected = TypeMismatchDataAccessException.class) // SPR-16483
+	public void doesNotUseConversionService() throws SQLException {
+		SingleColumnRowMapper<LocalDateTime> rowMapper =
+				SingleColumnRowMapper.newInstance(LocalDateTime.class, null);
+
+		ResultSet resultSet = mock(ResultSet.class);
+		ResultSetMetaData metaData = mock(ResultSetMetaData.class);
+		given(metaData.getColumnCount()).willReturn(1);
+		given(resultSet.getMetaData()).willReturn(metaData);
+		given(resultSet.getObject(1, LocalDateTime.class))
+				.willThrow(new SQLFeatureNotSupportedException());
+		given(resultSet.getTimestamp(1)).willReturn(new Timestamp(0));
+
+		rowMapper.mapRow(resultSet, 1);
+	}
+
+	private static class MyLocalDateTime {
+		private final LocalDateTime value;
+		private MyLocalDateTime(LocalDateTime value) {
+			this.value = value;
+		}
+	}
+
+}