// Licensed to the .NET Foundation under one or more agreements.
// The .NET Foundation licenses this file to you under the MIT license.
 
using Aspire.Hosting.ApplicationModel;
using Aspire.Hosting.Utils;
 
namespace Aspire.Hosting.Azure.Tests;
 
public class AzureCosmosDBDatabaseConnectionPropertiesTests
{
    [Fact]
    public void AzureCosmosDBDatabaseResourceGetConnectionPropertiesReturnsExpectedValues()
    {
        using var builder = TestDistributedApplicationBuilder.Create();
        var cosmosdb = builder.AddAzureCosmosDB("cosmosdb");
        var database = cosmosdb.AddCosmosDatabase("database", "mydb");
 
        var resource = Assert.Single(builder.Resources.OfType<AzureCosmosDBDatabaseResource>());
        var properties = ((IResourceWithConnectionString)resource).GetConnectionProperties().ToDictionary(x => x.Key, x => x.Value);
 
        Assert.Collection(
            properties,
            property =>
            {
                Assert.Equal("Uri", property.Key);
                Assert.Equal("{cosmosdb.outputs.connectionString}", property.Value.ValueExpression);
            },
            property =>
            {
                Assert.Equal("DatabaseName", property.Key);
                Assert.Equal("mydb", property.Value.ValueExpression);
            });
    }
 
    [Fact]
    public void AzureCosmosDBDatabaseResourceWithAccessKeyAuthenticationGetConnectionPropertiesReturnsExpectedValues()
    {
        using var builder = TestDistributedApplicationBuilder.Create();
        var cosmosdb = builder.AddAzureCosmosDB("cosmosdb").WithAccessKeyAuthentication();
        var database = cosmosdb.AddCosmosDatabase("database", "mydb");
 
        var resource = Assert.Single(builder.Resources.OfType<AzureCosmosDBDatabaseResource>());
        var properties = ((IResourceWithConnectionString)resource).GetConnectionProperties().ToDictionary(x => x.Key, x => x.Value);
 
        Assert.Collection(
            properties,
            property =>
            {
                Assert.Equal("Uri", property.Key);
                Assert.Equal("{cosmosdb.outputs.connectionString}", property.Value.ValueExpression);
            },
            property =>
            {
                Assert.Equal("AccountKey", property.Key);
                Assert.Equal("{cosmosdb-kv.secrets.primaryaccesskey--cosmosdb}", property.Value.ValueExpression);
            },
            property =>
            {
                Assert.Equal("ConnectionString", property.Key);
                Assert.Equal("{cosmosdb-kv.secrets.connectionstrings--cosmosdb}", property.Value.ValueExpression);
            },
            property =>
            {
                Assert.Equal("DatabaseName", property.Key);
                Assert.Equal("mydb", property.Value.ValueExpression);
            });
    }
 
    [Theory]
    [InlineData(false)]
    [InlineData(true)]
    public void AzureCosmosDBDatabaseResourceEmulatorGetConnectionPropertiesReturnsExpectedValues(bool useClassic)
    {
        using var builder = TestDistributedApplicationBuilder.Create();
        var cosmosdb = builder.AddAzureCosmosDB("cosmosdb");
        cosmosdb = useClassic ? cosmosdb.RunAsClassicEmulator() : cosmosdb.RunAsEmulator();
        var database = cosmosdb.AddCosmosDatabase("database", "mydb");
 
        var resource = Assert.Single(builder.Resources.OfType<AzureCosmosDBDatabaseResource>());
        var properties = ((IResourceWithConnectionString)resource).GetConnectionProperties().ToArray();
        var expectedConnectionString = useClassic
            ? "AccountKey=C2y6yDjf5/R+ob0N8A7Cgv30VRDJIWEHLM+4QDU5DE2nQ9nDuVTqobD4b8mGGyPMbIZnqyMsEcaGQy67XIw/Jw==;AccountEndpoint=https://{cosmosdb.bindings.emulator.host}:{cosmosdb.bindings.emulator.port};DisableServerCertificateValidation=True;"
            : "AccountKey=C2y6yDjf5/R+ob0N8A7Cgv30VRDJIWEHLM+4QDU5DE2nQ9nDuVTqobD4b8mGGyPMbIZnqyMsEcaGQy67XIw/Jw==;AccountEndpoint={cosmosdb.bindings.emulator.url}";
 
        Assert.Collection(
            properties,
            property =>
            {
                Assert.Equal("Uri", property.Key);
                Assert.Equal("{cosmosdb.bindings.emulator.url}", property.Value.ValueExpression);
            },
            property =>
            {
                Assert.Equal("AccountKey", property.Key);
                Assert.Equal("C2y6yDjf5/R+ob0N8A7Cgv30VRDJIWEHLM+4QDU5DE2nQ9nDuVTqobD4b8mGGyPMbIZnqyMsEcaGQy67XIw/Jw==", property.Value.ValueExpression);
            },
            property =>
            {
                Assert.Equal("ConnectionString", property.Key);
                Assert.Equal(expectedConnectionString, property.Value.ValueExpression);
            },
            property =>
            {
                Assert.Equal("DatabaseName", property.Key);
                Assert.Equal("mydb", property.Value.ValueExpression);
            });
    }
}