/*
    *    Copyright 2009-2024 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
    *
    *       https://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.apache.ibatis.executor.resultset;
  
  import static org.junit.jupiter.api.Assertions.assertEquals;
  import static org.mockito.ArgumentMatchers.any;
  import static org.mockito.Mockito.mock;
  import static org.mockito.Mockito.when;
  
  import java.sql.Connection;
  import java.sql.DatabaseMetaData;
  import java.sql.ResultSet;
  import java.sql.ResultSetMetaData;
  import java.sql.SQLException;
  import java.sql.Statement;
  import java.sql.Types;
  import java.util.ArrayList;
  import java.util.HashMap;
  import java.util.List;
  
  import org.apache.ibatis.builder.StaticSqlSource;
  import org.apache.ibatis.executor.Executor;
  import org.apache.ibatis.executor.ExecutorException;
  import org.apache.ibatis.executor.parameter.ParameterHandler;
  import org.apache.ibatis.mapping.BoundSql;
  import org.apache.ibatis.mapping.MappedStatement;
  import org.apache.ibatis.mapping.ResultMap;
  import org.apache.ibatis.mapping.ResultMapping;
  import org.apache.ibatis.mapping.SqlCommandType;
  import org.apache.ibatis.session.Configuration;
  import org.apache.ibatis.session.ResultHandler;
  import org.apache.ibatis.session.RowBounds;
  import org.apache.ibatis.type.TypeHandler;
  import org.apache.ibatis.type.TypeHandlerRegistry;
  import org.junit.jupiter.api.Assertions;
  import org.junit.jupiter.api.Test;
  import org.junit.jupiter.api.extension.ExtendWith;
  import org.mockito.Mock;
  import org.mockito.junit.jupiter.MockitoExtension;
  
  @ExtendWith(MockitoExtension.class)
  class DefaultResultSetHandlerTest {
  
    @Mock
    private Statement stmt;
    @Mock
    private ResultSet rs;
    @Mock
    private ResultSetMetaData rsmd;
    @Mock
    private Connection conn;
    @Mock
    private DatabaseMetaData dbmd;
  
    /**
     * Contrary to the spec, some drivers require case-sensitive column names when getting result.
     *
     * @see <a href="https://github.com/mybatis/old-google-code-issues/issues/557">Issue 557</a>
     */
    @Test
    void shouldRetainColumnNameCase() throws Exception {
  
      final MappedStatement ms = getMappedStatement();
  
      final Executor executor = null;
      final ParameterHandler parameterHandler = null;
      final ResultHandler resultHandler = null;
      final BoundSql boundSql = null;
      final RowBounds rowBounds = new RowBounds(0, 100);
      final DefaultResultSetHandler fastResultSetHandler = new DefaultResultSetHandler(executor, ms, parameterHandler,
          resultHandler, boundSql, rowBounds);
  
      when(stmt.getResultSet()).thenReturn(rs);
      when(rs.getMetaData()).thenReturn(rsmd);
      when(rs.getType()).thenReturn(ResultSet.TYPE_FORWARD_ONLY);
      when(rs.next()).thenReturn(true).thenReturn(false);
      when(rs.getInt("CoLuMn1")).thenReturn(100);
      when(rsmd.getColumnCount()).thenReturn(1);
      when(rsmd.getColumnLabel(1)).thenReturn("CoLuMn1");
      when(rsmd.getColumnType(1)).thenReturn(Types.INTEGER);
      when(rsmd.getColumnClassName(1)).thenReturn(Integer.class.getCanonicalName());
      when(stmt.getConnection()).thenReturn(conn);
      when(conn.getMetaData()).thenReturn(dbmd);
      when(dbmd.supportsMultipleResultSets()).thenReturn(false); // for simplicity.
  
      final List<Object> results = fastResultSetHandler.handleResultSets(stmt);
     assertEquals(1, results.size());
     assertEquals(100, ((HashMap) results.get(0)).get("cOlUmN1"));
   }
 
   @Test
   void shouldThrowExceptionWithColumnName() throws Exception {
     final MappedStatement ms = getMappedStatement();
     final RowBounds rowBounds = new RowBounds(0, 100);
 
     final DefaultResultSetHandler defaultResultSetHandler = new DefaultResultSetHandler(null/* executor */, ms,
         null/* parameterHandler */, null/* resultHandler */, null/* boundSql */, rowBounds);
 
     final ResultSetWrapper rsw = mock(ResultSetWrapper.class);
     when(rsw.getResultSet()).thenReturn(mock(ResultSet.class));
 
     final ResultMapping resultMapping = mock(ResultMapping.class);
     final TypeHandler typeHandler = mock(TypeHandler.class);
     when(resultMapping.getColumn()).thenReturn("column");
     when(resultMapping.getTypeHandler()).thenReturn(typeHandler);
     when(typeHandler.getResult(any(ResultSet.class), any(String.class))).thenThrow(new SQLException("exception"));
     List<ResultMapping> constructorMappings = List.of(resultMapping);
 
     try {
       defaultResultSetHandler.createParameterizedResultObject(rsw, null/* resultType */, constructorMappings,
           null/* constructorArgTypes */, null/* constructorArgs */, null/* columnPrefix */);
       Assertions.fail("Should have thrown ExecutorException");
     } catch (Exception e) {
       Assertions.assertTrue(e instanceof ExecutorException, "Expected ExecutorException");
       Assertions.assertTrue(e.getMessage().contains("mapping: " + resultMapping.toString()));
     }
   }
 
   MappedStatement getMappedStatement() {
     final Configuration config = new Configuration();
     final TypeHandlerRegistry registry = config.getTypeHandlerRegistry();
     return new MappedStatement.Builder(config, "testSelect", new StaticSqlSource(config, "some select statement"),
         SqlCommandType.SELECT).resultMaps(new ArrayList<ResultMap>() {
           private static final long serialVersionUID = 1L;
           {
             add(new ResultMap.Builder(config, "testMap", HashMap.class, new ArrayList<ResultMapping>() {
               private static final long serialVersionUID = 1L;
               {
                 add(new ResultMapping.Builder(config, "cOlUmN1", "CoLuMn1", registry.getTypeHandler(Integer.class))
                     .build());
               }
             }).build());
           }
         }).build();
   }
 
 }