/**
 * 
 */
package com.zaxxer.hikari;

import static org.mockito.Matchers.anyInt;
import static org.mockito.Mockito.doAnswer;
import static org.mockito.Mockito.when;

import java.sql.Connection;
import java.sql.SQLException;
import java.util.concurrent.TimeUnit;

import org.junit.Assert;
import org.junit.Test;
import org.mockito.invocation.InvocationOnMock;
import org.mockito.stubbing.Answer;

import com.zaxxer.hikari.mocks.MockDataSource;
import com.zaxxer.hikari.util.UtilityElf;

/**
 * Test for cases when db network connectivity goes down and close is called on existing connections. By default Hikari
 * blocks longer than getMaximumTimeout (it can hang for a lot of time depending on driver timeout settings). Closing
 * async the connections fixes this issue.
 *
 */
public class TestConnectionCloseBlocking {
   private volatile boolean shouldSleep = true;

   @Test
   public void testConnectionCloseBlocking() throws SQLException {
      HikariConfig config = new HikariConfig();
      config.setMinimumIdle(0);
      config.setMaximumPoolSize(1);
      config.setConnectionTimeout(1500);
      config.setDataSource(new CustomMockDataSource());

      HikariDataSource ds = new HikariDataSource(config);

      long startNanos = System.nanoTime();
      try {
         Connection connection = ds.getConnection();
         connection.close();
         // Hikari only checks for validity for connections with lastAccess > 1000 ms so we sleep for 1001 ms to force
         // Hikari to do a connection validation which will fail and will trigger the connection to be closed
         UtilityElf.quietlySleep(1001);
         startNanos = System.nanoTime();
         connection = ds.getConnection(); // on physical connection close we sleep 2 seconds
         Assert.assertTrue("Waited longer than timeout",
               (UtilityElf.elapsedNanos(startNanos) < config.getConnectionTimeout()));
      } catch (SQLException e) {
         Assert.assertTrue("getConnection failed because close connection took longer than timeout",
               (UtilityElf.elapsedNanos(startNanos) < config.getConnectionTimeout()));
      } finally {
         shouldSleep = false;
         ds.close();
      }
   }

   private class CustomMockDataSource extends MockDataSource {
      @Override
      public Connection getConnection() throws SQLException {
         Connection mockConnection = super.getConnection();
         when(mockConnection.isValid(anyInt())).thenReturn(false);
         doAnswer(new Answer<Void>() {
            @Override
            public Void answer(InvocationOnMock invocation) throws Throwable {
               if (shouldSleep) {
                  TimeUnit.SECONDS.sleep(2);
               }
               return null;
            }
         }).when(mockConnection).close();
         return mockConnection;
      }
   }

}

@@ -1226,29 +1226,24 @@
   long start
-Nanos
  = System.na
@@ -1236,32 +1236,41 @@
 rt = System.
-nanoTime
+currentTimeMillis
 ();%0A      tr
@@ -1638,21 +1638,16 @@
    start
-Nanos
  = Syste
@@ -1652,16 +1652,25 @@
 tem.
-nanoTime
+currentTimeMillis
 ();%0A
@@ -1853,32 +1853,28 @@
 .elapsed
-Nano
+TimeM
 s(start
-Nanos
 ) %3C conf
@@ -2028,32 +2028,32 @@
  than timeout%22,%0A
+
                (
@@ -2081,21 +2081,16 @@
 os(start
-Nanos
 ) %3C conf

Remove staggling nanoTime() calls.