// Licensed to the .NET Foundation under one or more agreements.
// The .NET Foundation licenses this file to you under the MIT license.
 
using Microsoft.AspNetCore.Http;
using Microsoft.AspNetCore.Routing.Patterns;
using Microsoft.Extensions.DependencyInjection;
 
namespace Microsoft.AspNetCore.Routing.Internal;
 
public class DfaGraphWriterTest
{
    private DfaGraphWriter CreateGraphWriter()
    {
        ServiceCollection services = new ServiceCollection();
        services.AddLogging();
        services.AddRouting();
 
        return new DfaGraphWriter(services.BuildServiceProvider());
    }
 
    [Fact]
    public void Write_ExcludeNonRouteEndpoint()
    {
        // Arrange
        var graphWriter = CreateGraphWriter();
        var writer = new StringWriter();
        var endpointsDataSource = new DefaultEndpointDataSource(new Endpoint((context) => null, EndpointMetadataCollection.Empty, string.Empty));
 
        // Act
        graphWriter.Write(endpointsDataSource, writer);
 
        // Assert
        Assert.Equal(String.Join(Environment.NewLine, @"digraph DFA {",
@"0 [label=""/""]",
"}") + Environment.NewLine, writer.ToString());
    }
 
    [Fact]
    public void Write_ExcludeRouteEndpointWithSuppressMatchingMetadata()
    {
        // Arrange
        var graphWriter = CreateGraphWriter();
        var writer = new StringWriter();
        var endpointsDataSource = new DefaultEndpointDataSource(
            new RouteEndpoint(
                (context) => null,
                RoutePatternFactory.Parse("/"),
                0,
                new EndpointMetadataCollection(new SuppressMatchingMetadata()),
                string.Empty));
 
        // Act
        graphWriter.Write(endpointsDataSource, writer);
 
        // Assert
        Assert.Equal(String.Join(Environment.NewLine, @"digraph DFA {",
@"0 [label=""/""]",
@"}") + Environment.NewLine, writer.ToString());
    }
 
    [Fact]
    public void Write_IncludeRouteEndpointWithPolicy()
    {
        // Arrange
        var graphWriter = CreateGraphWriter();
        var writer = new StringWriter();
        var endpointsDataSource = new DefaultEndpointDataSource(
            new RouteEndpoint(
                (context) => null,
                RoutePatternFactory.Parse("/"),
                0,
                new EndpointMetadataCollection(new HttpMethodMetadata(new[] { "GET" })),
                string.Empty));
 
        // Act
        graphWriter.Write(endpointsDataSource, writer);
 
        // Assert
        var sdf = writer.ToString();
        Assert.Equal(String.Join(Environment.NewLine, @"digraph DFA {",
@"0 [label=""/ HTTP: GET""]",
@"1 [label=""/ HTTP: *""]",
@"2 -> 0 [label=""HTTP: GET""]",
@"2 -> 1 [label=""HTTP: *""]",
@"2 [label=""/""]",
@"}") + Environment.NewLine, sdf);
    }
}