Fixes to flaky tests which fail due to mocked connections coming back…

… to the connection pool

Snowflake.Data.Tests/IntegrationTests/SFDbCommandIT.cs

@@ -229,9 +229,6 @@ public void TestSimpleLargeResultSet()
         }
 
 
-        /*
-         * Disabled to make sure that configuration changes does not cause problems with appveyor
-         */
         [Test, NonParallelizable]
         public void TestUseV1ResultParser()
         {

Snowflake.Data.Tests/Mock/MockRetryUntilRestTimeout.cs

@@ -31,7 +31,7 @@ protected override async Task<HttpResponseMessage> SendAsync(HttpRequestMessage
                                                               string sid = "")
         {
             // Override the http timeout and set to 1ms to force all http request to timeout and retry
-            message.Properties[BaseRestRequest.HTTP_REQUEST_TIMEOUT_KEY] = TimeSpan.FromMilliseconds(1);
+            message.Properties[BaseRestRequest.HTTP_REQUEST_TIMEOUT_KEY] = TimeSpan.FromTicks(0);;
             return await (base.SendAsync(message, restTimeout, externalCancellationToken).ConfigureAwait(false));
         }
     }

Snowflake.Data.Tests/Mock/MockSnowflakeDbConnection.cs

@@ -92,5 +92,10 @@ private void OnSessionEstablished()
         {
             _connectionState = ConnectionState.Open;
         }
+
+        protected override bool CanReuseSession(TransactionRollbackStatus transactionRollbackStatus)
+        {
+            return false;
+        }
     }
 }

Snowflake.Data/Client/SnowflakeDbConnection.cs

@@ -37,7 +37,7 @@ public class SnowflakeDbConnection : DbConnection
         // Will fix that in a separated PR though as it's a different issue
         private static Boolean _isArrayBindStageCreated;
 
-        private enum TransactionRollbackStatus
+        protected enum TransactionRollbackStatus
         {
             Undefined, // used to indicate ignored transaction status when pool disabled
             Success,
@@ -232,7 +232,7 @@ public Task CloseAsync(CancellationToken cancellationToken)
             return taskCompletionSource.Task;
         }
 
-        private bool CanReuseSession(TransactionRollbackStatus transactionRollbackStatus)
+        protected virtual bool CanReuseSession(TransactionRollbackStatus transactionRollbackStatus)
         {
             return SnowflakeDbConnectionPool.GetPooling() && 
                    transactionRollbackStatus == TransactionRollbackStatus.Success;

Snowflake.Data/Core/HttpUtil.cs

@@ -364,7 +364,10 @@ protected override async Task<HttpResponseMessage> SendAsync(HttpRequestMessage
                         if (!httpTimeout.Equals(Timeout.InfiniteTimeSpan))
                         {
                             childCts = CancellationTokenSource.CreateLinkedTokenSource(cancellationToken);
-                            childCts.CancelAfter(httpTimeout);
+                            if (httpTimeout.Ticks == 0)
+                                childCts.Cancel();
+                            else
+                                childCts.CancelAfter(httpTimeout);                        
                         }
                         response = await base.SendAsync(requestMessage, childCts == null ?
                             cancellationToken : childCts.Token).ConfigureAwait(false);