// 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.Orchestrator;
using Microsoft.Extensions.DependencyInjection;
using Xunit;
 
namespace Aspire.Hosting.Tests.Orchestrator;
 
public class RelationshipEvaluatorTests
{
    [Fact]
    public void HandlesNestedChildren()
    {
        var builder = DistributedApplication.CreateBuilder();
 
        var parentResource = builder.AddContainer("parent", "image");
        var childResource = builder.AddResource(new CustomChildResource("child", parentResource.Resource));
        var grandChildResource = builder.AddResource(new CustomChildResource("grandchild", childResource.Resource));
        var greatGrandChildResource = builder.AddResource(new CustomChildResource("greatgrandchild", grandChildResource.Resource));
 
        var childWithAnnotationsResource = builder.AddContainer("child-with-annotations", "image")
            .WithParentRelationship(parentResource);
 
        var grandChildWithAnnotationsResource = builder.AddContainer("grandchild-with-annotations", "image")
            .WithParentRelationship(childWithAnnotationsResource);
 
        using var app = builder.Build();
        var appModel = app.Services.GetRequiredService<DistributedApplicationModel>();
 
        var parentChildLookup = RelationshipEvaluator.GetParentChildLookup(appModel);
        Assert.Equal(4, parentChildLookup.Count);
 
        Assert.Collection(parentChildLookup[parentResource.Resource],
            x => Assert.Equal(childResource.Resource, x),
            x => Assert.Equal(childWithAnnotationsResource.Resource, x));
 
        Assert.Single(parentChildLookup[childResource.Resource], grandChildResource.Resource);
        Assert.Single(parentChildLookup[grandChildResource.Resource], greatGrandChildResource.Resource);
 
        Assert.Empty(parentChildLookup[greatGrandChildResource.Resource]);
 
        Assert.Single(parentChildLookup[childWithAnnotationsResource.Resource], grandChildWithAnnotationsResource.Resource);
 
        Assert.Empty(parentChildLookup[grandChildWithAnnotationsResource.Resource]);
    }
 
    private sealed class CustomChildResource(string name, IResource parent) : Resource(name), IResourceWithParent
    {
        public IResource Parent => parent;
    }
}