// Licensed to the .NET Foundation under one or more agreements.
// The .NET Foundation licenses this file to you under the MIT license.
 
using System.Net;
using Microsoft.AspNetCore.Builder;
using Microsoft.AspNetCore.Hosting;
using Microsoft.AspNetCore.Http;
using Microsoft.AspNetCore.TestHost;
using Microsoft.Extensions.DependencyInjection;
using Microsoft.Extensions.Hosting;
 
namespace Microsoft.AspNetCore.Diagnostics.HealthChecks;
 
public class HealthCheckEndpointRouteBuilderExtensionsTest
{
    [Fact]
    public void ThrowFriendlyErrorWhenServicesNotRegistered()
    {
        using var host = new HostBuilder()
            .ConfigureWebHost(webHostBuilder =>
            {
                webHostBuilder
                .UseTestServer()
                .Configure(app =>
                {
                    app.UseRouting();
                    app.UseEndpoints(endpoints =>
                    {
                        endpoints.MapHealthChecks("/healthz");
                    });
                })
                .ConfigureServices(services =>
                {
                    services.AddRouting();
                });
            }).Build();
 
        var ex = Assert.Throws<InvalidOperationException>(() => host.Start());
 
        Assert.Equal(
            "Unable to find the required services. Please add all the required services by calling " +
            "'IServiceCollection.AddHealthChecks' inside the call to 'ConfigureServices(...)' " +
            "in the application startup code.",
            ex.Message);
    }
 
    [Fact]
    public async Task MapHealthChecks_ReturnsOk()
    {
        // Arrange
        using var host = new HostBuilder()
            .ConfigureWebHost(webHostBuilder =>
            {
                webHostBuilder
                .UseTestServer()
                .Configure(app =>
                {
                    app.UseRouting();
                    app.UseEndpoints(endpoints =>
                    {
                        endpoints.MapHealthChecks("/healthz");
                    });
                })
                .ConfigureServices(services =>
                {
                    services.AddRouting();
                    services.AddHealthChecks();
                });
            }).Build();
 
        await host.StartAsync();
 
        var server = host.GetTestServer();
        var client = server.CreateClient();
 
        // Act
        var response = await client.GetAsync("/healthz");
 
        // Assert
        Assert.Equal(HttpStatusCode.OK, response.StatusCode);
        Assert.Equal("text/plain", response.Content.Headers.ContentType.ToString());
        Assert.Equal("Healthy", await response.Content.ReadAsStringAsync());
    }
 
    [Fact]
    public async Task MapHealthChecks_WithOptions_ReturnsOk()
    {
        // Arrange
        using var host = new HostBuilder()
            .ConfigureWebHost(webHostBuilder =>
            {
                webHostBuilder
                .UseTestServer()
                .Configure(app =>
                {
                    app.UseRouting();
                    app.UseEndpoints(endpoints =>
                    {
                        endpoints.MapHealthChecks("/healthz", new HealthCheckOptions
                        {
                            ResponseWriter = async (context, report) =>
                            {
                                context.Response.ContentType = "text/plain";
                                await context.Response.WriteAsync("Custom!");
                            }
                        });
                    });
                })
                .ConfigureServices(services =>
                {
                    services.AddRouting();
                    services.AddHealthChecks();
                });
            }).Build();
 
        await host.StartAsync();
 
        var server = host.GetTestServer();
        var client = server.CreateClient();
 
        // Act
        var response = await client.GetAsync("/healthz");
 
        // Assert
        Assert.Equal(HttpStatusCode.OK, response.StatusCode);
        Assert.Equal("text/plain", response.Content.Headers.ContentType.ToString());
        Assert.Equal("Custom!", await response.Content.ReadAsStringAsync());
    }
}