feat: add owner registration and dispose logic for implicit sessions

Author: LiamHamsters

src/Ydb.Sdk/src/Ado/PoolManager.cs

@@ -37,6 +37,7 @@ CancellationToken cancellationToken
             ISessionSource newSessionPool;
             if (settings.EnableImplicitSession)
             {
+                driver.RegisterOwner();
                 newSessionPool = new ImplicitSessionSource(driver);
             }
             else

src/Ydb.Sdk/src/Ado/Session/ImplicitSessionSource.cs

@@ -41,7 +41,7 @@ private bool TryAcquireLease()
 
     internal void ReleaseLease() => Interlocked.Decrement(ref _activeLeaseCount);
 
-    public ValueTask DisposeAsync()
+    public async ValueTask DisposeAsync()
     {
         Interlocked.Exchange(ref _isDisposed, 1);
 
@@ -51,8 +51,7 @@ public ValueTask DisposeAsync()
             spinner.SpinOnce();
         }
 
+        await _driver.DisposeAsync();
         _onBecameEmpty?.Invoke();
-
-        return ValueTask.CompletedTask;
     }
 }

src/Ydb.Sdk/test/Ydb.Sdk.Ado.Tests/Session/YdbImplicitStressTests.cs

@@ -6,7 +6,12 @@ namespace Ydb.Sdk.Ado.Tests.Session;
 
 public class YdbImplicitStressTests : TestBase
 {
-    private static IDriver DummyDriver() => new Mock<IDriver>(MockBehavior.Strict).Object;
+    private static IDriver DummyDriver()
+    {
+        var m = new Mock<IDriver>(MockBehavior.Loose);
+        m.Setup(d => d.DisposeAsync()).Returns(ValueTask.CompletedTask);
+        return m.Object;
+    }
 
     private sealed class Counter
     {