diff --git a/src/Ydb.Sdk/test/Ydb.Sdk.Ado.Tests/Sys/AdoNetQuerySessionTests.cs b/src/Ydb.Sdk/test/Ydb.Sdk.Ado.Tests/Sys/AdoNetQuerySessionTests.cs index 3463ed05..b36837a6 100644 --- a/src/Ydb.Sdk/test/Ydb.Sdk.Ado.Tests/Sys/AdoNetQuerySessionTests.cs +++ b/src/Ydb.Sdk/test/Ydb.Sdk.Ado.Tests/Sys/AdoNetQuerySessionTests.cs @@ -7,17 +7,16 @@ public class AdoNetQuerySessionTests : TestBase [Fact] public async Task QuerySessionPidTest() { + var expectedPid = Environment.ProcessId.ToString(); + await using var connection = await CreateOpenConnectionAsync(); var dbCommand = connection.CreateCommand(); - dbCommand.CommandText = "SELECT ClientPID FROM `.sys/query_sessions` LIMIT 1;"; - - var expectedPid = Environment.ProcessId.ToString(); + dbCommand.CommandText = "SELECT ClientPID FROM `.sys/query_sessions` WHERE ClientPID = @pid;"; + dbCommand.Parameters.Add(new YdbParameter("pid", expectedPid)); await dbCommand.ExecuteNonQueryAsync(); - await using var reader = await dbCommand.ExecuteReaderAsync(); + var actualPid = await dbCommand.ExecuteScalarAsync(); - Assert.True(reader.HasRows); - Assert.True(await reader.ReadAsync()); - Assert.Equal(reader.GetString(0), expectedPid); + Assert.Equal(expectedPid, actualPid); } }