// Licensed to the .NET Foundation under one or more agreements.
// The .NET Foundation licenses this file to you under the MIT license.
 
using System.IO.Pipelines;
 
namespace Microsoft.AspNetCore.SignalR.Microbenchmarks.Shared;
 
public class TestPipeWriter : PipeWriter
{
    // huge buffer that should be large enough for writing any content
    private readonly byte[] _buffer = new byte[10000];
 
    public bool ForceAsync { get; set; }
 
    public override void Advance(int bytes)
    {
    }
 
    public override Memory<byte> GetMemory(int sizeHint = 0)
    {
        return _buffer;
    }
 
    public override Span<byte> GetSpan(int sizeHint = 0)
    {
        return _buffer;
    }
 
    public override void CancelPendingFlush()
    {
        throw new NotImplementedException();
    }
 
    public override void Complete(Exception exception = null)
    {
        throw new NotImplementedException();
    }
 
    public override ValueTask<FlushResult> FlushAsync(CancellationToken cancellationToken = new CancellationToken())
    {
        if (!ForceAsync)
        {
            return default;
        }
 
        return new ValueTask<FlushResult>(ForceAsyncResult());
    }
 
    public async Task<FlushResult> ForceAsyncResult()
    {
        await Task.CompletedTask.ConfigureAwait(ConfigureAwaitOptions.ForceYielding);
        return default;
    }
}