// Licensed to the .NET Foundation under one or more agreements.
// The .NET Foundation licenses this file to you under the MIT license.
 
namespace Microsoft.AspNetCore.Http.Features;
 
public class StreamResponseBodyFeatureTests
{
    [Fact]
    public async Task CompleteAsyncCallsStartAsync()
    {
        // Arrange
        var stream = new MemoryStream();
        var streamResponseBodyFeature = new TestStreamResponseBodyFeature(stream);
 
        // Act
        await streamResponseBodyFeature.CompleteAsync();
 
        //Assert
        Assert.Equal(1, streamResponseBodyFeature.StartCalled);
    }
 
    [Fact]
    public async Task CompleteAsyncWontCallsStartAsyncIfAlreadyStarted()
    {
        // Arrange
        var stream = new MemoryStream();
        var streamResponseBodyFeature = new TestStreamResponseBodyFeature(stream);
        await streamResponseBodyFeature.StartAsync();
 
        // Act
        await streamResponseBodyFeature.CompleteAsync();
 
        //Assert
        Assert.Equal(1, streamResponseBodyFeature.StartCalled);
    }
 
    [Fact]
    public void DisableBufferingCallsInnerFeature()
    {
        // Arrange
        var stream = new MemoryStream();
 
        var innerFeature = new InnerDisableBufferingFeature(stream, null);
        var streamResponseBodyFeature = new StreamResponseBodyFeature(stream, innerFeature);
 
        // Act
        streamResponseBodyFeature.DisableBuffering();
 
        //Assert
        Assert.True(innerFeature.DisableBufferingCalled);
    }
}
 
public class TestStreamResponseBodyFeature : StreamResponseBodyFeature
{
    public TestStreamResponseBodyFeature(Stream stream)
        : base(stream)
    {
 
    }
 
    public override Task StartAsync(CancellationToken cancellationToken = default)
    {
        StartCalled++;
        return base.StartAsync(cancellationToken);
    }
 
    public int StartCalled { get; private set; }
}
 
public class InnerDisableBufferingFeature : StreamResponseBodyFeature
{
    public InnerDisableBufferingFeature(Stream stream, IHttpResponseBodyFeature priorFeature)
        : base(stream, priorFeature)
    {
    }
 
    public override void DisableBuffering()
    {
        DisableBufferingCalled = true;
    }
 
    public bool DisableBufferingCalled { get; set; }
}