/*
    * Copyright 2010-2025 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.mybatis.spring.batch;
  
  import static org.junit.jupiter.api.Assertions.fail;
  
  import java.util.Arrays;
  import java.util.Collections;
  import java.util.List;
  
  import org.apache.ibatis.cursor.Cursor;
  import org.apache.ibatis.session.ExecutorType;
  import org.apache.ibatis.session.SqlSession;
  import org.apache.ibatis.session.SqlSessionFactory;
  import org.assertj.core.api.Assertions;
  import org.junit.jupiter.api.Test;
  import org.junit.jupiter.api.extension.ExtendWith;
  import org.mockito.Mock;
  import org.mockito.Mockito;
  import org.mockito.junit.jupiter.MockitoExtension;
  import org.springframework.batch.infrastructure.item.ExecutionContext;
  import org.springframework.batch.infrastructure.item.ItemStreamException;
  
  /**
   * Tests for {@link MyBatisCursorItemReader}.
   */
  @ExtendWith(MockitoExtension.class)
  class MyBatisCursorItemReaderTest {
  
    @Mock
    private SqlSessionFactory sqlSessionFactory;
  
    @Mock
    private SqlSession sqlSession;
  
    @Mock
    private Cursor<Object> cursor;
  
    @Test
    void testCloseOnFailing() throws Exception {
  
      Mockito.when(this.sqlSessionFactory.openSession(ExecutorType.SIMPLE)).thenReturn(this.sqlSession);
      Mockito.when(this.sqlSession.selectCursor("selectFoo", Collections.singletonMap("id", 1)))
          .thenThrow(new RuntimeException("error."));
  
      var itemReader = new MyBatisCursorItemReader<Foo>();
      itemReader.setSqlSessionFactory(this.sqlSessionFactory);
      itemReader.setQueryId("selectFoo");
      itemReader.setParameterValues(Collections.singletonMap("id", 1));
      itemReader.afterPropertiesSet();
  
      var executionContext = new ExecutionContext();
      try {
        itemReader.open(executionContext);
        fail();
      } catch (ItemStreamException e) {
        Assertions.assertThat(e).hasMessage("Failed to initialize the reader").hasCause(new RuntimeException("error."));
      } finally {
        itemReader.close();
        Mockito.verify(this.sqlSession).close();
      }
  
    }
  
    @Test
    void testCloseBeforeOpen() {
      var itemReader = new MyBatisCursorItemReader<Foo>();
      itemReader.close();
    }
  
    private List<Object> getFoos() {
      return Arrays.asList(new Foo("foo1"), new Foo("foo2"), new Foo("foo3"));
    }
  
    // Note: Do not cleanup this 'foo' class
    private static class Foo {
      private final String name;
  
      Foo(String name) {
        this.name = name;
      }
  
      public String getName() {
        return this.name;
      }
    }
 
 }