CockroachDbContainer.cs

namespace Testcontainers.CockroachDb;

/// <inheritdoc cref="DockerContainer" />
[PublicAPI]
public sealed class CockroachDbContainer : DockerContainer, IDatabaseContainer
{
    private readonly CockroachDbConfiguration _configuration;

    /// <summary>
    /// Initializes a new instance of the <see cref="CockroachDbContainer" /> class.
    /// </summary>
    /// <param name="configuration">The container configuration.</param>
    public CockroachDbContainer(CockroachDbConfiguration configuration)
        : base(configuration)
    {
        _configuration = configuration;
    }

    /// <summary>
    /// Gets the CockroachDb connection string.
    /// </summary>
    /// <returns>The CockroachDb connection string.</returns>
    public string GetConnectionString()
    {
        var properties = new Dictionary<string, string>();
        properties.Add("Host", Hostname);
        properties.Add("Port", GetMappedPublicPort(CockroachDbBuilder.CockroachDbPort).ToString());
        properties.Add("Database", _configuration.Database);
        properties.Add("Username", _configuration.Username);
        properties.Add("Password", _configuration.Password);
        return string.Join(";", properties.Select(property => string.Join("=", property.Key, property.Value)));
    }

    /// <summary>
    /// Executes the SQL script in the CockroachDb container.
    /// </summary>
    /// <param name="scriptContent">The content of the SQL script to execute.</param>
    /// <param name="ct">Cancellation token.</param>
    /// <returns>Task that completes when the SQL script has been executed.</returns>
    public async Task<ExecResult> ExecScriptAsync(string scriptContent, CancellationToken ct = default)
    {
        var scriptFilePath = string.Join("/", string.Empty, "tmp", Guid.NewGuid().ToString("D"), Path.GetRandomFileName());

        await CopyAsync(Encoding.Default.GetBytes(scriptContent), scriptFilePath, fileMode: Unix.FileMode644, ct: ct)
            .ConfigureAwait(false);

        return await ExecAsync(new[] { "cockroach", "sql", "--insecure", "--file", scriptFilePath }, ct)
            .ConfigureAwait(false);
    }
}