fix linter + added tests

Author: KirillKurdyukov

src/Ydb.Sdk/src/IDriver.cs

@@ -48,10 +48,10 @@ public interface IBidirectionalStream<in TRequest, out TResponse> : IDisposable
 public abstract class BaseDriver : IDriver
 {
     private readonly ICredentialsProvider? _credentialsProvider;
-    
+
     protected readonly DriverConfig Config;
     protected readonly ILogger Logger;
-    
+
     internal readonly GrpcChannelFactory GrpcChannelFactory;
     internal readonly ChannelPool<GrpcChannel> ChannelPool;
 
@@ -66,10 +66,10 @@ ILogger logger
         Config = config;
         Logger = logger;
         LoggerFactory = loggerFactory;
-        
+
         GrpcChannelFactory = new GrpcChannelFactory(LoggerFactory, Config);
         ChannelPool = new ChannelPool<GrpcChannel>(LoggerFactory, GrpcChannelFactory);
-        
+
         _credentialsProvider = Config.User != null
             ? new CachedCredentialsProvider(
                 new StaticCredentialsAuthClient(Config, GrpcChannelFactory, LoggerFactory),

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

@@ -7,31 +7,20 @@ namespace Ydb.Sdk.Tests.Ado;
 
 public class YdbAdoUserPasswordTests : YdbAdoNetFixture
 {
+    private readonly string _user = "kurdyukovkirya" + Random.Shared.Next();
+
     public YdbAdoUserPasswordTests(YdbFactoryFixture fixture) : base(fixture)
     {
     }
 
     [Fact]
     public async Task Authentication_WhenUserAndPassword_ReturnValidConnection()
     {
-        await using var connection = await CreateOpenConnectionAsync();
-        var ydbCommand = connection.CreateCommand();
-        var kurdyukovkirya = "kurdyukovkirya" + Random.Shared.Next();
-        ydbCommand.CommandText = $"CREATE USER {kurdyukovkirya} PASSWORD 'password'";
-        await ydbCommand.ExecuteNonQueryAsync();
-        await connection.CloseAsync();
-
-        await using var userPasswordConnection =
-            new YdbConnection($"{ConnectionString};User={kurdyukovkirya};Password=password;");
+        await using var userPasswordConnection = new YdbConnection(
+            $"{ConnectionString};User={_user};Password=password;");
         await userPasswordConnection.OpenAsync();
-        ydbCommand = userPasswordConnection.CreateCommand();
-        ydbCommand.CommandText = "SELECT 1 + 2";
-        Assert.Equal(3, await ydbCommand.ExecuteScalarAsync());
-
-        await using var newConnection = await CreateOpenConnectionAsync();
-        ydbCommand = newConnection.CreateCommand();
-        ydbCommand.CommandText = $"DROP USER {kurdyukovkirya};";
-        await ydbCommand.ExecuteNonQueryAsync();
+        Assert.Equal(3, await new YdbCommand(userPasswordConnection)
+            { CommandText = "SELECT 1 + 2" }.ExecuteScalarAsync());
     }
 
     [Fact]
@@ -45,4 +34,31 @@ public async Task ExecuteNonQueryAsync_WhenCreateUser_ReturnEmptyResultSet()
         dbCommand.CommandText = $"DROP USER {user};";
         await dbCommand.ExecuteNonQueryAsync();
     }
+
+    [Fact]
+    public async Task DisableDiscovery_WhenUserIsCreatedAndPropertyIsTrue_SimpleWorking()
+    {
+        await using var userPasswordConnection = new YdbConnection(
+            $"{ConnectionString};User={_user};Password=password;DisableDiscovery=true");
+        await userPasswordConnection.OpenAsync();
+        var ydbCommand = userPasswordConnection.CreateCommand();
+        ydbCommand.CommandText = "SELECT 1 + 2";
+        Assert.Equal(3, await ydbCommand.ExecuteScalarAsync());
+    }
+
+    protected override async Task OnInitializeAsync()
+    {
+        await using var connection = await CreateOpenConnectionAsync();
+        var ydbCommand = connection.CreateCommand();
+        ydbCommand.CommandText = $"CREATE USER {_user} PASSWORD 'password'";
+        await ydbCommand.ExecuteNonQueryAsync();
+    }
+
+    protected override async Task OnDisposeAsync()
+    {
+        await using var ydbConnection = await CreateOpenConnectionAsync();
+        var ydbCommand = ydbConnection.CreateCommand();
+        ydbCommand.CommandText = $"DROP USER {_user};";
+        await ydbCommand.ExecuteNonQueryAsync();
+    }
 }