fix tests

Author: KirillKurdyukov

src/Ydb.Sdk/src/Services/Query/QueryClient.cs

@@ -9,7 +9,7 @@ public class QueryClientConfig
     public int MaxSessionPool
     {
         get => _masSessionPool;
-        set
+        init
         {
             if (value <= 0)
             {
@@ -20,18 +20,40 @@ public int MaxSessionPool
         }
     }
 
-    private int _masSessionPool = SessionPoolDefaultSettings.MaxSessionPool;
+    private readonly int _masSessionPool = SessionPoolDefaultSettings.MaxSessionPool;
+
+    public int CreateSessionTimeout
+    {
+        get => _createSessionTimeout;
+        init
+        {
+            if (value < 0)
+            {
+                throw new ArgumentOutOfRangeException(nameof(value), value,
+                    "Invalid create session timeout: " + value);
+            }
+
+            _createSessionTimeout = value;
+        }
+    }
+
+    private readonly int _createSessionTimeout = SessionPoolDefaultSettings.CreateSessionTimeoutSeconds;
 }
 
 public class QueryClient : IAsyncDisposable
 {
     private readonly SessionPool _sessionPool;
 
-    public QueryClient(Driver driver, QueryClientConfig? config = null)
+    public QueryClient(IDriver driver, QueryClientConfig? config = null)
     {
         config ??= new QueryClientConfig();
 
-        _sessionPool = new SessionPool(driver, new SessionPoolConfig(MaxSessionPool: config.MaxSessionPool));
+        _sessionPool = new SessionPool(driver,
+            new SessionPoolConfig(
+                CreateSessionTimeout: config.CreateSessionTimeout,
+                MaxSessionPool: config.MaxSessionPool
+            )
+        );
     }
 
     public Task<T> Stream<T>(string query, Func<ExecuteQueryStream, Task<T>> onStream,

src/Ydb.Sdk/tests/Ado/YdbConnectionStringBuilderTests.cs

@@ -43,7 +43,7 @@ public void InitConnectionStringBuilder_WhenExpectedKeys_ReturnUpdatedConnection
         var connectionString = new YdbConnectionStringBuilder(
             "Host=server;Port=2135;Database=/my/path;User=Kirill;UseTls=true;" +
             "ConnectTimeout=30;KeepAlivePingDelay=30;KeepAlivePingTimeout=60;" +
-            "EnableMultipleHttp2Connections=true;CreateTimeoutSession=30;" +
+            "EnableMultipleHttp2Connections=true;CreateSessionTimeout=30;" +
             "MaxSendMessageSize=1000000;MaxReceiveMessageSize=1000000;" +
             "DisableDiscovery=true"
         );
@@ -62,7 +62,7 @@ public void InitConnectionStringBuilder_WhenExpectedKeys_ReturnUpdatedConnection
         Assert.Equal(1000000, connectionString.MaxReceiveMessageSize);
         Assert.Equal("Host=server;Port=2135;Database=/my/path;User=Kirill;UseTls=True;" +
                      "ConnectTimeout=30;KeepAlivePingDelay=30;KeepAlivePingTimeout=60;" +
-                     "EnableMultipleHttp2Connections=True;CreateTimeoutSession=30;" +
+                     "EnableMultipleHttp2Connections=True;CreateSessionTimeout=30;" +
                      "MaxSendMessageSize=1000000;MaxReceiveMessageSize=1000000;" +
                      "DisableDiscovery=True", connectionString.ConnectionString);
         Assert.True(connectionString.DisableDiscovery);

src/Ydb.Sdk/tests/Pool/SessionPoolTests.cs

@@ -10,7 +10,7 @@ namespace Ydb.Sdk.Tests.Pool;
 [Trait("Category", "Unit")]
 public class SessionPoolTests
 {
-    private const int TestSessionPoolSize = 50;
+    internal const int TestSessionPoolSize = 50;
 
     private readonly TestSessionPool _testSessionPool = new();
 
@@ -82,7 +82,8 @@ internal class TestSessionPool : SessionPool<TestSession>
 
     public Status CreatedStatus { get; set; } = Status.Success;
 
-    public TestSessionPool() : base(NullLogger<TestSessionPool>.Instance, new SessionPoolConfig())
+    public TestSessionPool() : base(NullLogger<TestSessionPool>.Instance,
+        new SessionPoolConfig(MaxSessionPool: SessionPoolTests.TestSessionPoolSize, CreateSessionTimeout: 0))
     {
     }
 

src/Ydb.Sdk/tests/Query/QueryIntegrationTests.cs

@@ -7,15 +7,17 @@ namespace Ydb.Sdk.Tests.Query;
 
 [Collection("Integration QueryService test")]
 [Trait("Category", "Integration")]
-public class QueryIntegrationTests : IClassFixture<QueryClientFixture>, IAsyncLifetime
+public class QueryIntegrationTests : IClassFixture<QueryClientFixture>, IClassFixture<DriverFixture>, IAsyncLifetime
 {
     private static readonly TemporaryTables<QueryIntegrationTests> Tables = new();
 
     private readonly QueryClient _queryClient;
+    private readonly IDriver _driver;
 
-    public QueryIntegrationTests(QueryClientFixture queryClientFixture)
+    public QueryIntegrationTests(QueryClientFixture queryClientFixture, DriverFixture driverFixture)
     {
         _queryClient = queryClientFixture.QueryClient;
+        _driver = driverFixture.Driver;
     }
 
     [Fact]
@@ -216,12 +218,13 @@ await tx.Exec($"DELETE FROM {Tables.Episodes} WHERE series_id = 2 AND season_id
     [Fact]
     public async Task Stream_ReadingMore1000RowsFromChannel_ReturnChannelExecuteParts()
     {
+        await using var queryService = new QueryClient(_driver, new QueryClientConfig { CreateSessionTimeout = 0 });
         const int sizeSeasons = 10_000;
         var tasks = new Task[sizeSeasons];
 
         for (uint i = 0; i < sizeSeasons; i++)
         {
-            tasks[i] = _queryClient.Exec(
+            tasks[i] = queryService.Exec(
                 $"DECLARE $season_id AS Uint64; DECLARE $title AS Text;" +
                 $"INSERT INTO {Tables.Seasons} (series_id, season_id, title) VALUES (3, $season_id, $title)",
                 new Dictionary<string, YdbValue>
@@ -235,7 +238,7 @@ public async Task Stream_ReadingMore1000RowsFromChannel_ReturnChannelExecutePart
 
         var currentSeason = 0;
 
-        await _queryClient.Stream(
+        await queryService.Stream(
             $"SELECT title FROM {Tables.Seasons} ORDER BY series_id, season_id LIMIT {sizeSeasons} OFFSET 9",
             async stream =>
             {