// 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 System.Net.Http;
using System.Net.Http.Headers;

namespace Microsoft.AspNetCore.Mvc.FunctionalTests;

public class OutputFormatterTest : IClassFixture<MvcTestFixture<BasicWebSite.StartupWithoutEndpointRouting>>
{
    public OutputFormatterTest(MvcTestFixture<BasicWebSite.StartupWithoutEndpointRouting> fixture)
    {
        Client = fixture.CreateDefaultClient();
    }

    public HttpClient Client { get; }

    [Theory]
    [InlineData("ReturnTaskOfString")]
    [InlineData("ReturnTaskOfObject_StringValue")]
    [InlineData("ReturnString")]
    [InlineData("ReturnObject_StringValue")]
    public async Task StringOutputFormatter_ForStringValues_GetsSelectedReturnsTextPlainContentType(string actionName)
    {
        // Arrange
        var expectedContentType = MediaTypeHeaderValue.Parse("text/plain;charset=utf-8");
        var expectedBody = actionName;

        // Act
        var response = await Client.GetAsync("http://localhost/TextPlain/" + actionName);

        // Assert
        Assert.Equal(expectedContentType, response.Content.Headers.ContentType);
        var body = await response.Content.ReadAsStringAsync();
        Assert.Equal(expectedBody, body);
    }

    [Theory]
    [InlineData("ReturnTaskOfObject_ObjectValue")]
    [InlineData("ReturnObject_ObjectValue")]
    public async Task JsonOutputFormatter_ForNonStringValue_GetsSelected(string actionName)
    {
        // Arrange
        var expectedContentType = MediaTypeHeaderValue.Parse("application/json;charset=utf-8");

        // Act
        var response = await Client.GetAsync("http://localhost/TextPlain/" + actionName);

        // Assert
        Assert.Equal(expectedContentType, response.Content.Headers.ContentType);
    }

    [Theory]
    [InlineData("ReturnTask")]
    [InlineData("ReturnVoid")]
    public async Task NoContentFormatter_ForVoidAndTaskReturnType_DoesNotRun(string actionName)
    {
        // Arrange & Act
        var response = await Client.GetAsync("http://localhost/NoContent/" + actionName);

        // Assert
        Assert.Null(response.Content.Headers.ContentType);
        var body = await response.Content.ReadAsStringAsync();
        // Response body is empty instead of null.
        Assert.Empty(body);
        Assert.Equal(HttpStatusCode.OK, response.StatusCode);
        Assert.Equal(0, response.Content.Headers.ContentLength);
    }

    [Theory]
    [InlineData("ReturnTaskOfString_NullValue")]
    [InlineData("ReturnTaskOfObject_NullValue")]
    [InlineData("ReturnObject_NullValue")]
    public async Task NoContentFormatter_ForNullValue_ByDefault_GetsSelectedAndWritesResponse(string actionName)
    {
        // Arrange & Act
        var response = await Client.GetAsync("http://localhost/NoContent/" + actionName);

        // Assert
        Assert.Null(response.Content.Headers.ContentType);
        var body = await response.Content.ReadAsStringAsync();
        // Response body is empty instead of null.
        Assert.Empty(body);
        Assert.Equal(HttpStatusCode.NoContent, response.StatusCode);
        Assert.Equal(0, response.Content.Headers.ContentLength);
    }

    [Theory]
    [InlineData("ReturnTaskOfString_NullValue")]
    [InlineData("ReturnTaskOfObject_NullValue")]
    [InlineData("ReturnObject_NullValue")]
    public async Task
        NoContentFormatter_ForNullValue_AndTreatNullAsNoContentFlagSetToFalse_DoesNotGetSelected(string actionName)
    {
        // Arrange & Act
        var response = await Client.GetAsync("http://localhost/NoContentDoNotTreatNullValueAsNoContent/" +
                                             actionName);

        // Assert
        Assert.Null(response.Content.Headers.ContentType);
        var body = await response.Content.ReadAsStringAsync();
        // Response body is empty instead of null.
        Assert.Empty(body);
        Assert.Equal(HttpStatusCode.NotAcceptable, response.StatusCode);
    }
}
