Add test of connection handlers for driver factory

Author: alex-kulakov

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

@@ -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,18 @@ 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);
+      }
+    }
   }
 }