// Licensed to the .NET Foundation under one or more agreements.
// The .NET Foundation licenses this file to you under the MIT license.
 
using System;
using System.IO;
using System.Net.Http;
using System.Net.Sockets;
using System.Threading.Tasks;
using Microsoft.AspNetCore.Server.IIS.FunctionalTests;
using Microsoft.AspNetCore.Server.IIS.FunctionalTests.Utilities;
using Microsoft.AspNetCore.Server.IntegrationTesting;
using Microsoft.AspNetCore.Server.IntegrationTesting.IIS;
using Microsoft.AspNetCore.InternalTesting;
using Xunit;
 
namespace Microsoft.AspNetCore.Server.IIS.IISExpress.FunctionalTests;
 
[Collection(PublishedSitesCollection.Name)]
public class IISExpressShutdownTests : IISFunctionalTestBase
{
 
    public IISExpressShutdownTests(PublishedSitesFixture fixture) : base(fixture)
    {
    }
 
    [ConditionalFact]
    public async Task ServerShutsDownWhenMainExits()
    {
        var parameters = Fixture.GetBaseDeploymentParameters();
        var deploymentResult = await DeployAsync(parameters);
        try
        {
            await deploymentResult.HttpClient.GetAsync("/Shutdown");
        }
        catch (HttpRequestException ex) when (ex.InnerException is IOException)
        {
            // Server might close a connection before request completes
        }
        finally
        {
            deploymentResult.HttpClient.Dispose();
        }
 
        deploymentResult.AssertWorkerProcessStop();
    }
 
    [ConditionalFact]
    [SkipOnHelix("Unsupported queue", Queues = "Windows.Amd64.VS2022.Pre.Open;")]
    public async Task ServerShutsDownWhenMainExitsStress()
    {
        var parameters = Fixture.GetBaseDeploymentParameters();
        var deploymentResult = await StartAsync(parameters);
 
        var load = Helpers.StressLoad(deploymentResult.HttpClient, "/HelloWorld", response =>
        {
            var statusCode = (int)response.StatusCode;
            Assert.True(statusCode == 200 || statusCode == 503, "Status code was " + statusCode);
        });
 
        try
        {
            await deploymentResult.HttpClient.GetAsync("/Shutdown");
            await load;
        }
        catch (HttpRequestException ex) when (ex.InnerException is IOException | ex.InnerException is SocketException)
        {
            // Server might close a connection before request completes
        }
        finally
        {
            deploymentResult.HttpClient.Dispose();
        }
 
        deploymentResult.AssertWorkerProcessStop();
    }
 
    [ConditionalFact]
    [SkipNonHelix("https://github.com/dotnet/aspnetcore/issues/25107")]
    public async Task GracefulShutdown_DoesNotCrashProcess()
    {
        var parameters = Fixture.GetBaseDeploymentParameters();
        var result = await DeployAsync(parameters);
 
        var response = await result.HttpClient.GetAsync("/HelloWorld");
        response.EnsureSuccessStatusCode();
        StopServer(gracefulShutdown: true);
        Assert.True(result.HostProcess.ExitCode == 0);
    }
 
    [ConditionalFact]
    public async Task ForcefulShutdown_DoesCrashProcess()
    {
        var parameters = Fixture.GetBaseDeploymentParameters();
        var result = await DeployAsync(parameters);
 
        var response = await result.HttpClient.GetAsync("/HelloWorld");
        response.EnsureSuccessStatusCode();
        StopServer(gracefulShutdown: false);
        Assert.True(result.HostProcess.ExitCode == 1);
    }
}