last fixes

Author: KirillKurdyukov

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

@@ -157,9 +157,6 @@ private async ValueTask<ISession> RentAsync(CancellationToken cancellationToken)
 
             try
             {
-                if (IsDisposed)
-                    throw new YdbException("Session Source is disposed.");
-
                 var session = _sessionFactory.NewSession(this);
                 await session.Open(cancellationToken);
 

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

@@ -263,7 +263,7 @@ public async Task StressTest_HighContention_OpenClose()
         {
             IsBroken = sessionNum =>
             {
-                var isBroken = Random.Shared.NextDouble() < 0.05;
+                var isBroken = Random.Shared.NextDouble() < 0.2;
                 sessionIdIsBroken[sessionNum] = isBroken;
                 return isBroken;
             },
@@ -418,7 +418,7 @@ public MockPoolingSession NewSession(PoolingSessionSource<MockPoolingSession> so
 
                 return Task.CompletedTask;
             },
-            () => IsBroken(_numSession),
+            sessionNum => IsBroken(sessionNum),
             Interlocked.Increment(ref _sessionOpened)
         );
 
@@ -429,13 +429,13 @@ internal class MockPoolingSession(
     PoolingSessionSource<MockPoolingSession> source,
     Func<int, Task> mockOpen,
     Func<Task> mockDeleteSession,
-    Func<bool> mockIsBroken,
+    Func<int, bool> mockIsBroken,
     int sessionNum
 ) : PoolingSessionBase<MockPoolingSession>(source)
 {
     public int SessionId => sessionNum;
     public override IDriver Driver => null!;
-    public override bool IsBroken => mockIsBroken();
+    public override bool IsBroken => mockIsBroken(sessionNum);
 
     internal override Task Open(CancellationToken cancellationToken) => mockOpen(sessionNum);
     internal override Task DeleteSession() => mockDeleteSession();

src/Ydb.Sdk/test/Ydb.Sdk.Ado.Tests/YdbDataSourceTests.cs

@@ -28,18 +28,21 @@ public async Task OpenConnectionAsync_WhenMaxSessionPool10_ReturnOpenConnection(
     [Fact]
     public void CreateCommand_FromDataSource_ReturnDbCommand()
     {
+        var dataSource = new YdbDataSource(ConnectionString + ";MaxSessionPool=5");
         for (var i = 0; i < SelectedCount; i++)
         {
-            using var command = _dataSource.CreateCommand("SELECT 1;");
+            using var command = dataSource.CreateCommand("SELECT 1;");
             Assert.Equal(1, command.ExecuteScalar());
         }
 
-        _dataSource.Dispose();
+        dataSource.Dispose();
         for (var i = 0; i < SelectedCount; i++)
         {
-            using var command = _dataSource.CreateCommand("SELECT 1;");
+            using var command = dataSource.CreateCommand("SELECT 1;");
             Assert.Equal(1, command.ExecuteScalar());
         }
+
+        dataSource.Dispose();
     }
 
     [Fact]