chore: Ensure that stderr is empty after executing scripts (#1116)

0xced · HofmeisterAn · web-flow · commit 9156593723eb · 2024-03-14T19:44:35.000+01:00
Co-authored-by: Andre Hofmeister &lt;9199345+HofmeisterAn@users.noreply.github.com&gt;
diff --git a/src/Testcontainers.MariaDb/MariaDbBuilder.cs b/src/Testcontainers.MariaDb/MariaDbBuilder.cs
@@ -89,7 +89,8 @@ protected override MariaDbBuilder Init()
             .WithPortBinding(MariaDbPort, true)
             .WithDatabase(DefaultDatabase)
             .WithUsername(DefaultUsername)
-            .WithPassword(DefaultPassword);
+            .WithPassword(DefaultPassword)
+            .WithStartupCallback((container, ct) => container.WriteConfigurationFileAsync(ct));
     }
 
     /// <inheritdoc />
@@ -135,7 +136,7 @@ private sealed class WaitUntil : IWaitUntil
         /// <param name="configuration">The container configuration.</param>
         public WaitUntil(MariaDbConfiguration configuration)
         {
-            _command = new List<string> { "mariadb", "--protocol=TCP", $"--port={MariaDbPort}", $"--user={configuration.Username}", $"--password={configuration.Password}", configuration.Database, "--wait", "--silent", "--execute=SELECT 1;" };
+            _command = new List<string> { "mariadb", configuration.Database, "--wait", "--silent", "--execute=SELECT 1;" };
         }
 
         /// <inheritdoc />
diff --git a/src/Testcontainers.MariaDb/MariaDbContainer.cs b/src/Testcontainers.MariaDb/MariaDbContainer.cs
@@ -44,7 +44,25 @@ public async Task<ExecResult> ExecScriptAsync(string scriptContent, Cancellation
         await CopyAsync(Encoding.Default.GetBytes(scriptContent), scriptFilePath, Unix.FileMode644, ct)
             .ConfigureAwait(false);
 
-        return await ExecAsync(new[] { "mariadb", "--protocol=TCP", $"--port={MariaDbBuilder.MariaDbPort}", $"--user={_configuration.Username}", $"--password={_configuration.Password}", _configuration.Database, $"--execute=source {scriptFilePath};" }, ct)
+        return await ExecAsync(new[] { "mariadb", _configuration.Database, $"--execute=source {scriptFilePath};" }, ct)
             .ConfigureAwait(false);
     }
+
+    /// <summary>
+    /// Write an unobfuscated MariaDb configuration file that configures the client
+    /// login path. This prevents warnings in the <see cref="ExecScriptAsync" />
+    /// result about using a password on the command line.
+    /// </summary>
+    /// <param name="ct">Cancellation token.</param>
+    /// <returns>Task that completes when the configuration file has been executed.</returns>
+    internal Task WriteConfigurationFileAsync(CancellationToken ct = default)
+    {
+        var config = new StringWriter();
+        config.NewLine = "\n";
+        config.WriteLine("[client]");
+        config.WriteLine("protocol=TCP");
+        config.WriteLine($"user={_configuration.Username}");
+        config.WriteLine($"password={_configuration.Password}");
+        return CopyAsync(Encoding.Default.GetBytes(config.ToString()), "/etc/mysql/my.cnf", UnixFileModes.UserRead | UnixFileModes.UserWrite, ct);
+    }
 }
diff --git a/src/Testcontainers.MongoDb/MongoDbContainer.cs b/src/Testcontainers.MongoDb/MongoDbContainer.cs
@@ -42,7 +42,20 @@ public async Task<ExecResult> ExecScriptAsync(string scriptContent, Cancellation
         await CopyAsync(Encoding.Default.GetBytes(scriptContent), scriptFilePath, Unix.FileMode644, ct)
             .ConfigureAwait(false);
 
-        return await ExecAsync(new MongoDbShellCommand($"load('{scriptFilePath}')", _configuration.Username, _configuration.Password), ct)
+        var whichMongoDbShell = await ExecAsync(new[] { "which", "mongosh" }, ct)
+            .ConfigureAwait(false);
+
+        var command = new[]
+        {
+            whichMongoDbShell.ExitCode == 0 ? "mongosh" : "mongo",
+            "--username", _configuration.Username,
+            "--password", _configuration.Password,
+            "--quiet",
+            "--eval",
+            $"load('{scriptFilePath}')",
+        };
+
+        return await ExecAsync(command, ct)
             .ConfigureAwait(false);
     }
 }
diff --git a/src/Testcontainers.MongoDb/MongoDbShellCommand.cs b/src/Testcontainers.MongoDb/MongoDbShellCommand.cs
diff --git a/src/Testcontainers.MongoDb/Usings.cs b/src/Testcontainers.MongoDb/Usings.cs
@@ -1,5 +1,4 @@
 global using System;
-global using System.Collections.Generic;
 global using System.IO;
 global using System.Linq;
 global using System.Text;
diff --git a/src/Testcontainers.MySql/MySqlBuilder.cs b/src/Testcontainers.MySql/MySqlBuilder.cs
@@ -89,7 +89,8 @@ protected override MySqlBuilder Init()
             .WithPortBinding(MySqlPort, true)
             .WithDatabase(DefaultDatabase)
             .WithUsername(DefaultUsername)
-            .WithPassword(DefaultPassword);
+            .WithPassword(DefaultPassword)
+            .WithStartupCallback((container, ct) => container.WriteConfigurationFileAsync(ct));
     }
 
     /// <inheritdoc />
@@ -135,7 +136,7 @@ private sealed class WaitUntil : IWaitUntil
         /// <param name="configuration">The container configuration.</param>
         public WaitUntil(MySqlConfiguration configuration)
         {
-            _command = new List<string> { "mysql", "--protocol=TCP", $"--port={MySqlPort}", $"--user={configuration.Username}", $"--password={configuration.Password}", configuration.Database, "--wait", "--silent", "--execute=SELECT 1;" };
+            _command = new List<string> { "mysql", configuration.Database, "--wait", "--silent", "--execute=SELECT 1;" };
         }
 
         /// <inheritdoc />
diff --git a/src/Testcontainers.MySql/MySqlContainer.cs b/src/Testcontainers.MySql/MySqlContainer.cs
@@ -44,7 +44,25 @@ public async Task<ExecResult> ExecScriptAsync(string scriptContent, Cancellation
         await CopyAsync(Encoding.Default.GetBytes(scriptContent), scriptFilePath, Unix.FileMode644, ct)
             .ConfigureAwait(false);
 
-        return await ExecAsync(new[] { "mysql", "--protocol=TCP", $"--port={MySqlBuilder.MySqlPort}", $"--user={_configuration.Username}", $"--password={_configuration.Password}", _configuration.Database, $"--execute=source {scriptFilePath};" }, ct)
+        return await ExecAsync(new[] { "mysql", _configuration.Database, $"--execute=source {scriptFilePath};" }, ct)
             .ConfigureAwait(false);
     }
+
+    /// <summary>
+    /// Write an unobfuscated MySql configuration file that configures the client
+    /// login path. This prevents warnings in the <see cref="ExecScriptAsync" />
+    /// result about using a password on the command line.
+    /// </summary>
+    /// <param name="ct">Cancellation token.</param>
+    /// <returns>Task that completes when the configuration file has been executed.</returns>
+    internal Task WriteConfigurationFileAsync(CancellationToken ct = default)
+    {
+        var config = new StringWriter();
+        config.NewLine = "\n";
+        config.WriteLine("[client]");
+        config.WriteLine("protocol=TCP");
+        config.WriteLine($"user={_configuration.Username}");
+        config.WriteLine($"password={_configuration.Password}");
+        return CopyAsync(Encoding.Default.GetBytes(config.ToString()), "/etc/mysql/my.cnf", UnixFileModes.UserRead | UnixFileModes.UserWrite, ct);
+    }
 }
diff --git a/src/Testcontainers/Builders/ContainerBuilder`3.cs b/src/Testcontainers/Builders/ContainerBuilder`3.cs
@@ -366,9 +366,9 @@ public TBuilderEntity WithWaitStrategy(IWaitForContainerOS waitStrategy)
     }
 
     /// <inheritdoc />
-    public TBuilderEntity WithStartupCallback(Func<IContainer, CancellationToken, Task> startupCallback)
+    public TBuilderEntity WithStartupCallback(Func<TContainerEntity, CancellationToken, Task> startupCallback)
     {
-      return Clone(new ContainerConfiguration(startupCallback: startupCallback));
+      return Clone(new ContainerConfiguration(startupCallback: (container, ct) => startupCallback((TContainerEntity)container, ct)));
     }
 
     /// <inheritdoc />
diff --git a/src/Testcontainers/Builders/IContainerBuilder`2.cs b/src/Testcontainers/Builders/IContainerBuilder`2.cs
@@ -452,6 +452,6 @@ public interface IContainerBuilder<out TBuilderEntity, out TContainerEntity> : I
     /// <param name="startupCallback">The callback method to invoke.</param>
     /// <returns>A configured instance of <typeparamref name="TBuilderEntity" />.</returns>
     [PublicAPI]
-    TBuilderEntity WithStartupCallback(Func<IContainer, CancellationToken, Task> startupCallback);
+    TBuilderEntity WithStartupCallback(Func<TContainerEntity, CancellationToken, Task> startupCallback);
   }
 }
diff --git a/src/Testcontainers/Clients/DockerApiClient.cs b/src/Testcontainers/Clients/DockerApiClient.cs
@@ -113,7 +113,7 @@ await RuntimeInitialized.WaitAsync(ct)
       }
       catch(Exception e)
       {
-        Logger.LogError(e, "Failed to retrieve Docker container runtime information.");
+        Logger.LogError(e, "Failed to retrieve Docker container runtime information");
       }
       finally
       {
diff --git a/src/Testcontainers/Containers/ResourceReaperState.cs b/src/Testcontainers/Containers/ResourceReaperState.cs
@@ -4,6 +4,7 @@ namespace DotNet.Testcontainers.Containers
   using System.Threading;
   using DotNet.Testcontainers.Configurations;
   using DotNet.Testcontainers.Images;
+  using Microsoft.Extensions.Logging;
 
   /// <summary>
   /// Resource Reaper states.
@@ -24,7 +25,7 @@ public enum ResourceReaperState
     /// <see cref="ResourceReaper" /> maintains the TCP connection to Ryuk.
     /// </summary>
     /// <remarks>
-    /// <see cref="ResourceReaper.GetAndStartNewAsync(IDockerEndpointAuthenticationConfiguration, IImage, IMount, bool, TimeSpan, CancellationToken)" /> will complete now.
+    /// <see cref="ResourceReaper.GetAndStartNewAsync(IDockerEndpointAuthenticationConfiguration, IImage, IMount, ILogger, bool, TimeSpan, CancellationToken)" /> will complete now.
     /// </remarks>
     MaintainingConnection,
 
diff --git a/tests/Testcontainers.ClickHouse.Tests/ClickHouseContainerTest.cs b/tests/Testcontainers.ClickHouse.Tests/ClickHouseContainerTest.cs
@@ -39,7 +39,8 @@ public async Task ExecScriptReturnsSuccessful()
         var execResult = await _clickHouseContainer.ExecScriptAsync(scriptContent)
             .ConfigureAwait(true);
 
-        // When
+        // Then
         Assert.True(0L.Equals(execResult.ExitCode), execResult.Stderr);
+        Assert.Empty(execResult.Stderr);
     }
 }
diff --git a/tests/Testcontainers.CockroachDb.Tests/CockroachDbContainerTest.cs b/tests/Testcontainers.CockroachDb.Tests/CockroachDbContainerTest.cs
@@ -39,7 +39,8 @@ public async Task ExecScriptReturnsSuccessful()
         var execResult = await _cockroachDbContainer.ExecScriptAsync(scriptContent)
             .ConfigureAwait(true);
 
-        // When
+        // Then
         Assert.True(0L.Equals(execResult.ExitCode), execResult.Stderr);
+        Assert.Empty(execResult.Stderr);
     }
 }
diff --git a/tests/Testcontainers.FirebirdSql.Tests/FirebirdSqlContainerTest.cs b/tests/Testcontainers.FirebirdSql.Tests/FirebirdSqlContainerTest.cs
@@ -44,7 +44,7 @@ public async Task ExecScriptReturnsSuccessful()
         var execResult = await _firebirdSqlContainer.ExecScriptAsync(scriptContent)
             .ConfigureAwait(true);
 
-        // When
+        // Then
         Assert.True(0L.Equals(execResult.ExitCode), execResult.Stderr);
         Assert.Empty(execResult.Stderr);
     }
diff --git a/tests/Testcontainers.MariaDb.Tests/MariaDbContainerTest.cs b/tests/Testcontainers.MariaDb.Tests/MariaDbContainerTest.cs
@@ -44,8 +44,9 @@ public async Task ExecScriptReturnsSuccessful()
         var execResult = await _mariaDbContainer.ExecScriptAsync(scriptContent)
             .ConfigureAwait(true);
 
-        // When
+        // Then
         Assert.True(0L.Equals(execResult.ExitCode), execResult.Stderr);
+        Assert.Empty(execResult.Stderr);
     }
 
     [UsedImplicitly]
diff --git a/tests/Testcontainers.MongoDb.Tests/MongoDbContainerTest.cs b/tests/Testcontainers.MongoDb.Tests/MongoDbContainerTest.cs
@@ -44,8 +44,9 @@ public async Task ExecScriptReturnsSuccessful()
         var execResult = await _mongoDbContainer.ExecScriptAsync(scriptContent)
             .ConfigureAwait(true);
 
-        // When
+        // Then
         Assert.True(0L.Equals(execResult.ExitCode), execResult.Stderr);
+        Assert.Empty(execResult.Stderr);
     }
 
     [UsedImplicitly]
diff --git a/tests/Testcontainers.MsSql.Tests/MsSqlContainerTest.cs b/tests/Testcontainers.MsSql.Tests/MsSqlContainerTest.cs
@@ -39,7 +39,8 @@ public async Task ExecScriptReturnsSuccessful()
         var execResult = await _msSqlContainer.ExecScriptAsync(scriptContent)
             .ConfigureAwait(true);
 
-        // When
+        // Then
         Assert.True(0L.Equals(execResult.ExitCode), execResult.Stderr);
+        Assert.Empty(execResult.Stderr);
     }
 }
diff --git a/tests/Testcontainers.MySql.Tests/MySqlContainerTest.cs b/tests/Testcontainers.MySql.Tests/MySqlContainerTest.cs
@@ -44,8 +44,9 @@ public async Task ExecScriptReturnsSuccessful()
         var execResult = await _mySqlContainer.ExecScriptAsync(scriptContent)
             .ConfigureAwait(true);
 
-        // When
+        // Then
         Assert.True(0L.Equals(execResult.ExitCode), execResult.Stderr);
+        Assert.Empty(execResult.Stderr);
     }
 
     [UsedImplicitly]
diff --git a/tests/Testcontainers.Oracle.Tests/OracleContainerTest.cs b/tests/Testcontainers.Oracle.Tests/OracleContainerTest.cs
@@ -39,7 +39,8 @@ public async Task ExecScriptReturnsSuccessful()
         var execResult = await _oracleContainer.ExecScriptAsync(scriptContent)
             .ConfigureAwait(true);
 
-        // When
+        // Then
         Assert.True(0L.Equals(execResult.ExitCode), execResult.Stderr);
+        Assert.Empty(execResult.Stderr);
     }
 }
diff --git a/tests/Testcontainers.PostgreSql.Tests/PostgreSqlContainerTest.cs b/tests/Testcontainers.PostgreSql.Tests/PostgreSqlContainerTest.cs
@@ -39,8 +39,9 @@ public async Task ExecScriptReturnsSuccessful()
         var execResult = await _postgreSqlContainer.ExecScriptAsync(scriptContent)
             .ConfigureAwait(true);
 
-        // When
+        // Then
         Assert.True(0L.Equals(execResult.ExitCode), execResult.Stderr);
+        Assert.Empty(execResult.Stderr);
     }
 
     public sealed class ReuseContainerTest : IClassFixture<SharedPostgreSqlInstance>, IDisposable
diff --git a/tests/Testcontainers.Redis.Tests/RedisContainerTest.cs b/tests/Testcontainers.Redis.Tests/RedisContainerTest.cs
@@ -33,7 +33,9 @@ public async Task ExecScriptReturnsSuccessful()
         var execResult = await _redisContainer.ExecScriptAsync(scriptContent)
             .ConfigureAwait(true);
 
-        // When
+        // Then
+        Assert.True(0L.Equals(execResult.ExitCode), execResult.Stderr);
         Assert.True("Hello, scripting!\n".Equals(execResult.Stdout), execResult.Stdout);
+        Assert.Empty(execResult.Stderr);
     }
 }

Original file line number	Diff line number	Diff line change
`@@ -89,7 +89,8 @@ protected override MariaDbBuilder Init()`
`89`	`89`	`.WithPortBinding(MariaDbPort, true)`
`90`	`90`	`.WithDatabase(DefaultDatabase)`
`91`	`91`	`.WithUsername(DefaultUsername)`
`92`		`- .WithPassword(DefaultPassword);`
	`92`	`+ .WithPassword(DefaultPassword)`
	`93`	`+ .WithStartupCallback((container, ct) => container.WriteConfigurationFileAsync(ct));`
`93`	`94`	`}`
`94`	`95`
`95`	`96`	`/// <inheritdoc />`
`@@ -135,7 +136,7 @@ private sealed class WaitUntil : IWaitUntil`
`135`	`136`	`/// <param name="configuration">The container configuration.</param>`
`136`	`137`	`public WaitUntil(MariaDbConfiguration configuration)`
`137`	`138`	`{`
`138`		`- _command = new List<string> { "mariadb", "--protocol=TCP", $"--port={MariaDbPort}", $"--user={configuration.Username}", $"--password={configuration.Password}", configuration.Database, "--wait", "--silent", "--execute=SELECT 1;" };`
	`139`	`+ _command = new List<string> { "mariadb", configuration.Database, "--wait", "--silent", "--execute=SELECT 1;" };`
`139`	`140`	`}`
`140`	`141`
`141`	`142`	`/// <inheritdoc />`
Original file line number	Diff line number	Diff line change
`@@ -1,5 +1,4 @@`
`1`	`1`	`global using System;`
`2`		`-global using System.Collections.Generic;`
`3`	`2`	`global using System.IO;`
`4`	`3`	`global using System.Linq;`
`5`	`4`	`global using System.Text;`
Original file line number	Diff line number	Diff line change
`@@ -89,7 +89,8 @@ protected override MySqlBuilder Init()`
`89`	`89`	`.WithPortBinding(MySqlPort, true)`
`90`	`90`	`.WithDatabase(DefaultDatabase)`
`91`	`91`	`.WithUsername(DefaultUsername)`
`92`		`- .WithPassword(DefaultPassword);`
	`92`	`+ .WithPassword(DefaultPassword)`
	`93`	`+ .WithStartupCallback((container, ct) => container.WriteConfigurationFileAsync(ct));`
`93`	`94`	`}`
`94`	`95`
`95`	`96`	`/// <inheritdoc />`
`@@ -135,7 +136,7 @@ private sealed class WaitUntil : IWaitUntil`
`135`	`136`	`/// <param name="configuration">The container configuration.</param>`
`136`	`137`	`public WaitUntil(MySqlConfiguration configuration)`
`137`	`138`	`{`
`138`		`- _command = new List<string> { "mysql", "--protocol=TCP", $"--port={MySqlPort}", $"--user={configuration.Username}", $"--password={configuration.Password}", configuration.Database, "--wait", "--silent", "--execute=SELECT 1;" };`
	`139`	`+ _command = new List<string> { "mysql", configuration.Database, "--wait", "--silent", "--execute=SELECT 1;" };`
`139`	`140`	`}`
`140`	`141`
`141`	`142`	`/// <inheritdoc />`
Original file line number	Diff line number	Diff line change
`@@ -366,9 +366,9 @@ public TBuilderEntity WithWaitStrategy(IWaitForContainerOS waitStrategy)`
`366`	`366`	`}`
`367`	`367`
`368`	`368`	`/// <inheritdoc />`
`369`		`- public TBuilderEntity WithStartupCallback(Func<IContainer, CancellationToken, Task> startupCallback)`
	`369`	`+ public TBuilderEntity WithStartupCallback(Func<TContainerEntity, CancellationToken, Task> startupCallback)`
`370`	`370`	`{`
`371`		`- return Clone(new ContainerConfiguration(startupCallback: startupCallback));`
	`371`	`+ return Clone(new ContainerConfiguration(startupCallback: (container, ct) => startupCallback((TContainerEntity)container, ct)));`
`372`	`372`	`}`
`373`	`373`
`374`	`374`	`/// <inheritdoc />`
Original file line number	Diff line number	Diff line change
`@@ -452,6 +452,6 @@ public interface IContainerBuilder<out TBuilderEntity, out TContainerEntity> : I`
`452`	`452`	`/// <param name="startupCallback">The callback method to invoke.</param>`
`453`	`453`	`/// <returns>A configured instance of <typeparamref name="TBuilderEntity" />.</returns>`
`454`	`454`	`[PublicAPI]`
`455`		`- TBuilderEntity WithStartupCallback(Func<IContainer, CancellationToken, Task> startupCallback);`
	`455`	`+ TBuilderEntity WithStartupCallback(Func<TContainerEntity, CancellationToken, Task> startupCallback);`
`456`	`456`	`}`
`457`	`457`	`}`
Original file line number	Diff line number	Diff line change
`@@ -113,7 +113,7 @@ await RuntimeInitialized.WaitAsync(ct)`
`113`	`113`	`}`
`114`	`114`	`catch(Exception e)`
`115`	`115`	`{`
`116`		`- Logger.LogError(e, "Failed to retrieve Docker container runtime information.");`
	`116`	`+ Logger.LogError(e, "Failed to retrieve Docker container runtime information");`
`117`	`117`	`}`
`118`	`118`	`finally`
`119`	`119`	`{`
Original file line number	Diff line number	Diff line change
`@@ -39,7 +39,8 @@ public async Task ExecScriptReturnsSuccessful()`
`39`	`39`	`var execResult = await _clickHouseContainer.ExecScriptAsync(scriptContent)`
`40`	`40`	`.ConfigureAwait(true);`
`41`	`41`
`42`		`- // When`
	`42`	`+ // Then`
`43`	`43`	`Assert.True(0L.Equals(execResult.ExitCode), execResult.Stderr);`
	`44`	`+ Assert.Empty(execResult.Stderr);`
`44`	`45`	`}`
`45`	`46`	`}`
Original file line number	Diff line number	Diff line change
`@@ -44,7 +44,7 @@ public async Task ExecScriptReturnsSuccessful()`
`44`	`44`	`var execResult = await _firebirdSqlContainer.ExecScriptAsync(scriptContent)`
`45`	`45`	`.ConfigureAwait(true);`
`46`	`46`
`47`		`- // When`
	`47`	`+ // Then`
`48`	`48`	`Assert.True(0L.Equals(execResult.ExitCode), execResult.Stderr);`
`49`	`49`	`Assert.Empty(execResult.Stderr);`
`50`	`50`	`}`
Original file line number	Diff line number	Diff line change
`@@ -44,8 +44,9 @@ public async Task ExecScriptReturnsSuccessful()`
`44`	`44`	`var execResult = await _mariaDbContainer.ExecScriptAsync(scriptContent)`
`45`	`45`	`.ConfigureAwait(true);`
`46`	`46`
`47`		`- // When`
	`47`	`+ // Then`
`48`	`48`	`Assert.True(0L.Equals(execResult.ExitCode), execResult.Stderr);`
	`49`	`+ Assert.Empty(execResult.Stderr);`
`49`	`50`	`}`
`50`	`51`
`51`	`52`	`[UsedImplicitly]`
Original file line number	Diff line number	Diff line change
`@@ -39,8 +39,9 @@ public async Task ExecScriptReturnsSuccessful()`
`39`	`39`	`var execResult = await _postgreSqlContainer.ExecScriptAsync(scriptContent)`
`40`	`40`	`.ConfigureAwait(true);`
`41`	`41`
`42`		`- // When`
	`42`	`+ // Then`
`43`	`43`	`Assert.True(0L.Equals(execResult.ExitCode), execResult.Stderr);`
	`44`	`+ Assert.Empty(execResult.Stderr);`
`44`	`45`	`}`
`45`	`46`
`46`	`47`	`public sealed class ReuseContainerTest : IClassFixture<SharedPostgreSqlInstance>, IDisposable`