// 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.Hosting;
using Microsoft.AspNetCore.TestHost;
using Microsoft.Extensions.Hosting;
 
namespace Microsoft.AspNetCore.Diagnostics.HealthChecks;
 
public class HealthCheckMiddlewareSampleTest
{
    [Fact]
    public async Task BasicStartup()
    {
        using var host = new HostBuilder()
            .ConfigureWebHost(webHostBuilder =>
            {
                webHostBuilder
                .UseTestServer()
                .UseStartup<HealthChecksSample.BasicStartup>();
            }).Build();
 
        await host.StartAsync();
 
        var server = host.GetTestServer();
        var client = server.CreateClient();
 
        var response = await client.GetAsync("/health");
        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 CustomWriterStartup()
    {
        using var host = new HostBuilder()
            .ConfigureWebHost(webHostBuilder =>
            {
                webHostBuilder
                .UseTestServer()
                .UseStartup<HealthChecksSample.CustomWriterStartup>();
            }).Build();
 
        await host.StartAsync();
 
        var server = host.GetTestServer();
        var client = server.CreateClient();
 
        var response = await client.GetAsync("/health");
        Assert.Equal(HttpStatusCode.OK, response.StatusCode);
        Assert.Equal("application/json", response.Content.Headers.ContentType.ToString());
 
        // Ignoring the body since it contains a bunch of statistics
    }
 
    [Fact]
    public async Task LivenessProbeStartup_Liveness()
    {
        using var host = new HostBuilder()
            .ConfigureWebHost(webHostBuilder =>
            {
                webHostBuilder
                .UseTestServer()
                .UseStartup<HealthChecksSample.LivenessProbeStartup>();
            }).Build();
 
        await host.StartAsync();
 
        var server = host.GetTestServer();
        var client = server.CreateClient();
 
        var response = await client.GetAsync("/health/live");
        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 LivenessProbeStartup_Readiness()
    {
        using var host = new HostBuilder()
            .ConfigureWebHost(webHostBuilder =>
            {
                webHostBuilder
                .UseTestServer()
                .UseStartup<HealthChecksSample.LivenessProbeStartup>();
            }).Build();
 
        await host.StartAsync();
 
        var server = host.GetTestServer();
        var client = server.CreateClient();
 
        var response = await client.GetAsync("/health/ready");
        Assert.Equal(HttpStatusCode.ServiceUnavailable, response.StatusCode);
        Assert.Equal("text/plain", response.Content.Headers.ContentType.ToString());
        Assert.Equal("Unhealthy", await response.Content.ReadAsStringAsync());
    }
}