package com.atlassian.sal.spring.connection;
   
   import com.atlassian.sal.api.rdbms.ConnectionCallback;
   import com.atlassian.sal.spring.connection.SpringHostConnectionAccessor.ConnectionProvider;
   import org.junit.Before;
   import org.junit.Rule;
   import org.junit.Test;
   import org.junit.rules.ExpectedException;
   import org.junit.runner.RunWith;
  import org.mockito.Mock;
  import org.mockito.runners.MockitoJUnitRunner;
  import org.springframework.transaction.PlatformTransactionManager;
  
  import java.sql.Connection;
  import java.sql.SQLException;
  
  import static org.hamcrest.CoreMatchers.is;
  import static org.hamcrest.MatcherAssert.assertThat;
  import static org.mockito.Mockito.doThrow;
  import static org.mockito.Mockito.verify;
  import static org.mockito.Mockito.verifyNoMoreInteractions;
  import static org.mockito.Mockito.when;
  
  @RunWith (MockitoJUnitRunner.class)
  public class TestSpringHostConnectionAccessor
  {
      @Rule
      public ExpectedException expectedException = ExpectedException.none();
  
      private SpringHostConnectionAccessor springHostConnectionAccessor;
  
      @Mock
      private ConnectionProvider connectionProvider;
      @Mock
      private Connection connection;
      @Mock
      private PlatformTransactionManager transactionManager;
      @Mock
      private ConnectionCallback<Object> callback;
      @Mock
      private Object result;
  
      @Before
      public void before()
      {
          springHostConnectionAccessor = new SpringHostConnectionAccessor(connectionProvider, transactionManager);
      }
  
      @Test
      public void execute() throws SQLException
      {
          when(connectionProvider.getConnection()).thenReturn(connection);
          when(callback.execute(connection)).thenReturn(result);
  
          assertThat(springHostConnectionAccessor.execute(true, true, callback), is(result));
  
          verify(callback).execute(connection);
          verifyNoMoreInteractions(callback);
  
          verify(connection).close();
          verifyNoMoreInteractions(connection);
      }
  
      @Test
      public void executeCloseFail() throws SQLException
      {
          when(connectionProvider.getConnection()).thenReturn(connection);
          when(callback.execute(connection)).thenReturn(result);
          doThrow(new SQLException()).when(connection).close();
  
          assertThat(springHostConnectionAccessor.execute(true, true, callback), is(result));
  
          verify(callback).execute(connection);
          verifyNoMoreInteractions(callback);
  
          verify(connection).close();
          verifyNoMoreInteractions(connection);
      }
  }