frostfs-sdk-csharp/src/FrostFS.SDK.Tests/Mocks/ClientStreamWriter.cs

using System.Collections.ObjectModel;

using FrostFS.SDK.ProtosV2.Interfaces;

using Grpc.Core;

namespace FrostFS.SDK.Tests;

public class ClientStreamWriter : IClientStreamWriter<IRequest>
{
    private WriteOptions? _options;

    public Collection<IRequest> Messages { get; } = [];
    public bool CompletedTask { get; private set; }

    public WriteOptions? WriteOptions
    {
        get => _options;
        set => _options = value;
    }

    public Task CompleteAsync()
    {
        CompletedTask = true;
        return Task.CompletedTask;
    }

    public Task WriteAsync(IRequest message)
    {
        Object.PutRequest pr = new((Object.PutRequest)message);
        Messages.Add(pr);
        return Task.CompletedTask;
    }
}