Added tests for low an high level

Author: alex-kulakov

Orm/Xtensive.Orm.Tests.Sql/DriverFactoryTest.cs

@@ -1,4 +1,4 @@
-// Copyright (C) 2003-2010 Xtensive LLC.
+// Copyright (C) 2009-2021 Xtensive LLC.
 // This code is distributed under MIT license terms.
 // See the License.txt file in the project root for more information.
 
@@ -8,6 +8,44 @@
 using Xtensive.Orm;
 using Xtensive.Orm.Building.Builders;
 using Xtensive.Sql;
+using Xtensive.Orm.Tests.Sql.DriverFactoryTestTypes;
+
+namespace Xtensive.Orm.Tests.Sql.DriverFactoryTestTypes
+{
+  public class TestConnectionHandler : IConnectionHandler
+  {
+    public int OpeningCounter = 0;
+    public int OpenedCounter = 0;
+    public int OpeningInitCounter = 0;
+    public int OpeningFailedCounter = 0;
+
+    public void ConnectionOpening(ConnectionEventData eventData)
+    {
+      OpeningCounter++;
+    }
+
+    public void ConnectionOpened(ConnectionEventData eventData)
+    {
+      OpenedCounter++;
+    }
+
+    public void ConnectionInitialization(ConnectionInitEventData eventData)
+    {
+      OpeningInitCounter++;
+    }
+
+    public void ConnectionOpeningFailed(ConnectionErrorEventData eventData)
+    {
+      OpeningFailedCounter++;
+    }
+  }
+
+  public static class StaticCounter
+  {
+    public static int OpeningReached;
+    public static int OpenedReached;
+  }
+}
 
 namespace Xtensive.Orm.Tests.Sql
 {
@@ -95,6 +133,61 @@ public void SqlServerConnectionCheckTest()
       Assert.That(GetCheckConnectionIsAliveFlag(driver), Is.False);
     }
 
+    [Test]
+    public void ConnectionHandlerTest()
+    {
+      var handlerInstance = new TestConnectionHandler();
+      var handlersArray = new[] { handlerInstance };
+      var descriptor = ProviderDescriptor.Get(provider);
+      var factory = (SqlDriverFactory) Activator.CreateInstance(descriptor.DriverFactory);
+
+      Assert.That(handlerInstance.OpeningCounter, Is.EqualTo(0));
+      Assert.That(handlerInstance.OpeningInitCounter, Is.EqualTo(0));
+      Assert.That(handlerInstance.OpenedCounter, Is.EqualTo(0));
+      Assert.That(handlerInstance.OpeningFailedCounter, Is.EqualTo(0));
+
+      var configuration = new SqlDriverConfiguration(handlersArray);
+      _ = factory.GetDriver(new ConnectionInfo(Url), configuration);
+      Assert.That(handlerInstance.OpeningCounter, Is.EqualTo(1));
+      Assert.That(handlerInstance.OpeningInitCounter, Is.EqualTo(0));
+      Assert.That(handlerInstance.OpenedCounter, Is.EqualTo(1));
+      Assert.That(handlerInstance.OpeningFailedCounter, Is.EqualTo(0));
+
+      configuration = new SqlDriverConfiguration(handlersArray) { EnsureConnectionIsAlive = true };
+      _ = factory.GetDriver(new ConnectionInfo(Url), configuration);
+      Assert.That(handlerInstance.OpeningCounter, Is.EqualTo(2));
+      if (provider == WellKnown.Provider.SqlServer)
+        Assert.That(handlerInstance.OpeningInitCounter, Is.EqualTo(1));
+      else
+        Assert.That(handlerInstance.OpeningInitCounter, Is.EqualTo(0));
+      Assert.That(handlerInstance.OpenedCounter, Is.EqualTo(2));
+      Assert.That(handlerInstance.OpeningFailedCounter, Is.EqualTo(0));
+
+      configuration = new SqlDriverConfiguration(handlersArray) { ConnectionInitializationSql = InitQueryPerProvider(provider) };
+      _ = factory.GetDriver(new ConnectionInfo(Url), configuration);
+      Assert.That(handlerInstance.OpeningCounter, Is.EqualTo(3));
+      if (provider == WellKnown.Provider.SqlServer)
+        Assert.That(handlerInstance.OpeningInitCounter, Is.EqualTo(2));
+      else
+        Assert.That(handlerInstance.OpeningInitCounter, Is.EqualTo(1));
+      Assert.That(handlerInstance.OpenedCounter, Is.EqualTo(3));
+      Assert.That(handlerInstance.OpeningFailedCounter, Is.EqualTo(0));
+
+      configuration = new SqlDriverConfiguration(handlersArray) { ConnectionInitializationSql = "dummy string to trigger error" };
+      try {
+        _ = factory.GetDriver(new ConnectionInfo(Url), configuration);
+      }
+      catch {
+        //skip it
+      }
+      Assert.That(handlerInstance.OpeningCounter, Is.EqualTo(4));
+      if (provider == WellKnown.Provider.SqlServer)
+        Assert.That(handlerInstance.OpeningInitCounter, Is.EqualTo(3));
+      else
+        Assert.That(handlerInstance.OpeningInitCounter, Is.EqualTo(2));
+      Assert.That(handlerInstance.OpenedCounter, Is.EqualTo(3));
+      Assert.That(handlerInstance.OpeningFailedCounter, Is.EqualTo(1));
+    }
 
     private static void TestProvider(string providerName, string connectionString, string connectionUrl)
     {
@@ -109,5 +202,25 @@ private static bool GetCheckConnectionIsAliveFlag(SqlDriver driver)
       return (bool) type.GetField(fieldName, System.Reflection.BindingFlags.Instance | System.Reflection.BindingFlags.NonPublic)
         .GetValue(driver);
     }
+
+    private static string InitQueryPerProvider(string currentProvider)
+    {
+      switch (currentProvider) {
+        case WellKnown.Provider.Firebird:
+          return "select current_timestamp from RDB$DATABASE;";
+        case WellKnown.Provider.MySql:
+          return "SELECT 0";
+        case WellKnown.Provider.Oracle:
+          return "select current_timestamp from DUAL";
+        case WellKnown.Provider.PostgreSql:
+          return "SELECT 0";
+        case WellKnown.Provider.SqlServer:
+          return "SELECT 0";
+        case WellKnown.Provider.Sqlite:
+          return "SELECT 0";
+        default:
+          throw new ArgumentOutOfRangeException(currentProvider);
+      }
+    }
   }
 }