|
1 | 1 | using Xunit; |
2 | 2 | using Ydb.Sdk.Services.Query; |
| 3 | +using Ydb.Sdk.Tests.Ado.Specification; |
3 | 4 | using Ydb.Sdk.Tests.Fixture; |
4 | 5 |
|
5 | 6 | namespace Ydb.Sdk.Tests.Sys; |
6 | 7 |
|
7 | | -public class QuerySessionTests : IClassFixture<QueryClientFixture> |
| 8 | +public class QuerySessionTests : YdbAdoNetFixture |
8 | 9 | { |
9 | | - private readonly QueryClient _queryClient; |
10 | | - |
11 | | - public QuerySessionTests(QueryClientFixture queryClientFixture) |
| 10 | + public QuerySessionTests(YdbFactoryFixture fixture) : base(fixture) |
12 | 11 | { |
13 | | - _queryClient = queryClientFixture.QueryClient; |
14 | 12 | } |
15 | 13 |
|
16 | 14 | [Fact] |
17 | 15 | public async Task QuerySessionPidTest() |
18 | 16 | { |
19 | | - const string sql = @"SELECT * FROM `.sys/query_sessions` LIMIT 1"; |
| 17 | + await using var connection = await CreateOpenConnectionAsync(); |
| 18 | + var dbCommand = connection.CreateCommand(); |
| 19 | + dbCommand.CommandText = "SELECT ClientPID FROM `.sys/query_sessions` LIMIT 1;"; |
| 20 | + |
20 | 21 | var expectedPid = Environment.ProcessId.ToString(); |
| 22 | + |
| 23 | + await dbCommand.ExecuteNonQueryAsync(); |
| 24 | + await using var reader = await dbCommand.ExecuteReaderAsync(); |
21 | 25 |
|
22 | | - await _queryClient.Exec(sql); |
23 | | - var sessionRow = await _queryClient.ReadRow(sql); |
24 | | - |
25 | | - Assert.NotNull(sessionRow); |
26 | | - Assert.Equal(sessionRow["ClientPID"].GetOptionalUtf8(), expectedPid); |
| 26 | + Assert.True(reader.HasRows); |
| 27 | + Assert.True(await reader.ReadAsync()); |
| 28 | + Assert.Equal(reader.GetString(0), expectedPid); |
27 | 29 | } |
28 | 30 | } |
0 commit comments