cloud/blockstore/libs/vhost/server_ut.cpp - main - githubmirror/nbs

cloud/blockstore/libs/vhost/server_ut.cpp
1 163 строки38 KB
Grigoriy Yurgin
[Blockstore] Drop discard requests when needed (fixed version of #5626) (#5771)
20 апр 2026, 21:54
20 апр 2026, 21:54d7d8036
#include "server.h"

#include "vhost_test.h"

#include <cloud/blockstore/libs/diagnostics/critical_events.h>
#include <cloud/blockstore/libs/diagnostics/server_stats_test.h>
#include <cloud/blockstore/libs/diagnostics/volume_stats_test.h>
#include <cloud/blockstore/libs/service/device_handler.h>
#include <cloud/blockstore/libs/service/storage_test.h>
#include <cloud/storage/core/libs/common/error.h>
#include <cloud/storage/core/libs/common/sglist_test.h>
#include <cloud/storage/core/libs/diagnostics/logging.h>

#include <library/cpp/testing/unittest/registar.h>

#include <util/folder/path.h>
#include <util/generic/guid.h>
#include <util/generic/scope.h>
#include <util/system/tempfile.h>
#include <util/thread/factory.h>
#include <util/thread/lfqueue.h>

#include <atomic>

namespace NCloud::NBlockStore::NVhost {

using namespace NThreading;

namespace {

////////////////////////////////////////////////////////////////////////////////

struct TTestRequest
{
    EBlockStoreRequest Type = EBlockStoreRequest::ReadBlocks;
    ui64 StartIndex = 0;
    ui64 BlocksCount = 0;
    TSgList SgList;
};

////////////////////////////////////////////////////////////////////////////////

class TTestEnvironment
{
private:
    const size_t ThreadsCount = 2;

    const TFsPath SocketPath = TFsPath(CreateGuidAsString() + ".sock");
    const ui32 VhostQueuesCount = 1;
    const ui32 BlockSize;
    const ui64 BlocksCount = 256;
    const bool DropDiscardRequests;

    IServerPtr VhostServer;
    std::shared_ptr<TTestStorage> TestStorage;
    std::shared_ptr<ITestVhostDevice> VhostDevice;
    std::shared_ptr<TTestVhostQueueFactory> VhostQueueFactory;
    TLockFreeQueue<TTestRequest> RequestQueue;

    std::atomic_flag ServiceFrozen = false;
    TLockFreeQueue<TPromise<void>> FrozenPromises;

public:
    TTestEnvironment(ui32 blockSize, bool dropDiscardRequests = false)
        : BlockSize(blockSize)
        , DropDiscardRequests(dropDiscardRequests)
    {
        InitVhostDeviceEnvironment();
    }

    ~TTestEnvironment()
    {
        UninitVhostDeviceEnvironment();
    }

    void StopVhostServer()
    {
        VhostServer->Stop();
        VhostServer.reset();
    }

    std::shared_ptr<ITestVhostDevice> GetVhostDevice()
    {
        return VhostDevice;
    }

    TTestVhostQueueFactory& GetVhostQueueFactory()
    {
        return *VhostQueueFactory;
    }

    bool DequeueRequest(TTestRequest& request)
    {
        return RequestQueue.Dequeue(&request);
    }

    void FreezeService(bool freeze)
    {
        if (freeze) {
            ServiceFrozen.test_and_set();
        } else {
            ServiceFrozen.clear();
        }

        if (!freeze) {
            TPromise<void> promise;
            while (FrozenPromises.Dequeue(&promise)) {
                promise.SetValue();
            }
        }
    }

private:
    void InitVhostDeviceEnvironment()
    {
        TestStorage = std::make_shared<TTestStorage>();
        TestStorage->WriteBlocksLocalHandler =
            [&] (TCallContextPtr ctx, std::shared_ptr<NProto::TWriteBlocksLocalRequest> request) {
                Y_UNUSED(ctx);

                auto guard = request->Sglist.Acquire();
                UNIT_ASSERT(guard);
                auto sglist = guard.Get();
                UNIT_ASSERT(request->BlocksCount * BlockSize == SgListGetSize(sglist));

                RequestQueue.Enqueue({
                    EBlockStoreRequest::WriteBlocks,
                    request->GetStartIndex(),
                    request->BlocksCount,
                    std::move(sglist)});

                if (ServiceFrozen.test()) {
                    auto promise = NewPromise<void>();
                    auto future = promise.GetFuture();
                    FrozenPromises.Enqueue(std::move(promise));
                    return future.Apply([=] (const auto& future) {
                        Y_UNUSED(future);
                        return NProto::TWriteBlocksLocalResponse();
                    });
                }

                return MakeFuture(NProto::TWriteBlocksLocalResponse());
            };
        TestStorage->ReadBlocksLocalHandler =
            [&] (TCallContextPtr ctx, std::shared_ptr<NProto::TReadBlocksLocalRequest> request) {
                Y_UNUSED(ctx);

                auto guard = request->Sglist.Acquire();
                UNIT_ASSERT(guard);
                auto sglist = guard.Get();
                UNIT_ASSERT(request->GetBlocksCount() * BlockSize == SgListGetSize(sglist));

                RequestQueue.Enqueue({
                    EBlockStoreRequest::ReadBlocks,
                    request->GetStartIndex(),
                    request->GetBlocksCount(),
                    std::move(sglist)});

                if (ServiceFrozen.test()) {
                    auto promise = NewPromise<void>();
                    auto future = promise.GetFuture();
                    FrozenPromises.Enqueue(std::move(promise));
                    return future.Apply([=] (const auto& future) {
                        Y_UNUSED(future);
                        return NProto::TReadBlocksLocalResponse();
                    });
                }

                return MakeFuture(NProto::TReadBlocksLocalResponse());
            };

        TestStorage->ZeroBlocksHandler =
            [&](TCallContextPtr ctx,
                std::shared_ptr<NProto::TZeroBlocksRequest> request)
        {
            Y_UNUSED(ctx);

            RequestQueue.Enqueue(
                {EBlockStoreRequest::ZeroBlocks,
                 request->GetStartIndex(),
                 request->GetBlocksCount(),
                 {}});

            if (ServiceFrozen.test()) {
                auto promise = NewPromise<void>();
                auto future = promise.GetFuture();
                FrozenPromises.Enqueue(std::move(promise));
                return future.Apply(
                    [=](const auto& future)
                    {
                        Y_UNUSED(future);
                        return NProto::TZeroBlocksResponse();
                    });
            }

            return MakeFuture(NProto::TZeroBlocksResponse());
        };

        VhostQueueFactory = std::make_shared<TTestVhostQueueFactory>();

        TServerConfig serverConfig;
        serverConfig.ThreadsCount = ThreadsCount;

        VhostServer = CreateServer(
            CreateLoggingService("console"),
            CreateServerStatsStub(),
            VhostQueueFactory,
            CreateDefaultDeviceHandlerFactory(),
            serverConfig,
            TVhostCallbacks());

        VhostServer->Start();
        Sleep(TDuration::MilliSeconds(300));
        UNIT_ASSERT(VhostQueueFactory->Queues.size() == ThreadsCount);
        auto firstQueue = VhostQueueFactory->Queues.at(0);
        UNIT_ASSERT(firstQueue->IsRun());

        {
            TStorageOptions options;
            options.DiskId = "TestDiskId";
            options.BlockSize = BlockSize;
            options.BlocksCount = BlocksCount;
            options.VhostQueuesCount = VhostQueuesCount;
            options.UnalignedRequestsDisabled = false;
            options.OptimalIoSize = 4_MB;
            options.DropDiscardRequests = DropDiscardRequests;

            auto future = VhostServer->StartEndpoint(
                SocketPath.GetPath(),
                TestStorage,
                options);
            const auto& error = future.GetValue(TDuration::Seconds(5));
            UNIT_ASSERT_C(!HasError(error), error);
        }
        UNIT_ASSERT(firstQueue->GetDevices().size() == 1);
        VhostDevice = firstQueue->GetDevices().at(0);
        UNIT_ASSERT_VALUES_EQUAL(4_MB, VhostDevice->GetOptimalIoSize());
    }

    void UninitVhostDeviceEnvironment()
    {
        if (VhostServer) {
            auto future = VhostServer->StopEndpoint(SocketPath.GetPath());
            const auto& error = future.GetValue(TDuration::Seconds(5));
            UNIT_ASSERT_C(!HasError(error), error);
        }

        Sleep(TDuration::MilliSeconds(300));
        UNIT_ASSERT(VhostDevice->IsStopped());

        if (VhostServer) {
            VhostServer->Stop();
        }
    }
};

}   // namespace

////////////////////////////////////////////////////////////////////////////////

Y_UNIT_TEST_SUITE(TServerTest)
{
    Y_UNIT_TEST(ShouldStartStopVhostEndpoint)
    {
        auto logging = CreateLoggingService("console");
        InitVhostLog(logging);

        auto vhostQueueFactory = CreateVhostQueueFactory();

        auto vhostServer = CreateServer(
            logging,
            CreateServerStatsStub(),
            vhostQueueFactory,
            CreateDefaultDeviceHandlerFactory(),
            TServerConfig(),
            TVhostCallbacks());

        vhostServer->Start();

        const TFsPath socket(CreateGuidAsString() + ".sock");

        {
            TStorageOptions options;
            options.DiskId = "TestDiskId";
            options.BlockSize = DefaultBlockSize;
            options.BlocksCount = 42;
            options.VhostQueuesCount = 1;
            options.UnalignedRequestsDisabled = false;

            auto future = vhostServer->StartEndpoint(
                socket.GetPath(),
                std::make_shared<TTestStorage>(),
                options);
            const auto& error = future.GetValue(TDuration::Seconds(5));
            UNIT_ASSERT_C(!HasError(error), error);
        }

        UNIT_ASSERT(socket.Exists());

        {
            auto future = vhostServer->StopEndpoint(socket.GetPath());
            const auto& error = future.GetValue(TDuration::Seconds(5));
            UNIT_ASSERT_C(!HasError(error), error);
        }

        vhostServer->Stop();
    }

    Y_UNIT_TEST(ShouldStopVhostServerWithStartedEndpoints)
    {
        auto logging = CreateLoggingService("console");
        InitVhostLog(logging);

        auto vhostQueueFactory = CreateVhostQueueFactory();

        auto vhostServer = CreateServer(
            logging,
            CreateServerStatsStub(),
            vhostQueueFactory,
            CreateDefaultDeviceHandlerFactory(),
            TServerConfig(),
            TVhostCallbacks());

        vhostServer->Start();

        TStorageOptions options;
        options.DiskId = "TestDiskId";
        options.BlockSize = DefaultBlockSize;
        options.BlocksCount = 42;
        options.VhostQueuesCount = 1;
        options.UnalignedRequestsDisabled = false;

        const size_t endpointCount = 8;
        TString sockets[endpointCount];

        for (size_t i = 0; i < endpointCount; ++i) {
            char ch = '0' + i;
            sockets[i] = CreateGuidAsString() + ch + ".sock";

            auto future = vhostServer->StartEndpoint(
                sockets[i],
                std::make_shared<TTestStorage>(),
                options);
            const auto& error = future.GetValue(TDuration::Seconds(5));
            UNIT_ASSERT_C(!HasError(error), error);
            UNIT_ASSERT(TFsPath(sockets[i]).Exists());
        }

        vhostServer->Stop();
    }

    Y_UNIT_TEST(ShouldHandleVhostReadWriteRequests)
    {
        const ui32 blockSize = 4096;
        const ui64 firstSector = 8;
        const ui64 totalSectors = 32;
        const ui64 sectorSize = 512;

        UNIT_ASSERT(totalSectors * sectorSize % blockSize == 0);

        auto environment = TTestEnvironment(blockSize);
        auto device = environment.GetVhostDevice();

        TVector<TString> blocks;
        auto sgList = ResizeBlocks(
            blocks,
            totalSectors * sectorSize / blockSize,
            TString(blockSize, 'f'));

        {
            auto future = device->SendTestRequest(
                EBlockStoreRequest::WriteBlocks,
                firstSector * sectorSize,
                totalSectors * sectorSize,
                sgList);
            const auto& response = future.GetValue(TDuration::Seconds(5));
            UNIT_ASSERT(response == TVhostRequest::SUCCESS);

            TTestRequest request;
            bool res = environment.DequeueRequest(request);
            UNIT_ASSERT(res);
            UNIT_ASSERT(request.Type == EBlockStoreRequest::WriteBlocks);
            UNIT_ASSERT(request.StartIndex * blockSize == firstSector * sectorSize);
            UNIT_ASSERT(request.BlocksCount * blockSize == totalSectors * sectorSize);
            UNIT_ASSERT_VALUES_EQUAL(request.SgList, sgList);
            UNIT_ASSERT(!environment.DequeueRequest(request));
        }

        {
            auto future = device->SendTestRequest(
                EBlockStoreRequest::ReadBlocks,
                firstSector * sectorSize,
                totalSectors * sectorSize,
                sgList);
            const auto& response = future.GetValue(TDuration::Seconds(5));
            UNIT_ASSERT(response == TVhostRequest::SUCCESS);

            TTestRequest request;
            bool res = environment.DequeueRequest(request);
            UNIT_ASSERT(res);
            UNIT_ASSERT(request.Type == EBlockStoreRequest::ReadBlocks);
            UNIT_ASSERT(request.StartIndex * blockSize == firstSector * sectorSize);
            UNIT_ASSERT(request.BlocksCount * blockSize == totalSectors * sectorSize);
            UNIT_ASSERT_VALUES_EQUAL(request.SgList, sgList);
            UNIT_ASSERT(!environment.DequeueRequest(request));
        }
    }

    Y_UNIT_TEST(ShouldThrowCriticalEventIfFailedRequestQueueRunning)
    {
        NMonitoring::TDynamicCountersPtr counters = new NMonitoring::TDynamicCounters();
        InitCriticalEventsCounter(counters);
        auto configCounter =
            counters->GetCounter("AppCriticalEvents/VhostQueueRunningError", true);

        auto environment = TTestEnvironment(DefaultBlockSize);

        UNIT_ASSERT_VALUES_EQUAL(0, static_cast<int>(*configCounter));

        auto& factory = environment.GetVhostQueueFactory();
        factory.Queues.at(0)->Break();

        factory.FailedEvent.Reset();
        factory.FailedEvent.WaitT(TDuration::Seconds(1));
        factory.FailedEvent.Reset();
        factory.FailedEvent.WaitT(TDuration::Seconds(1));

        UNIT_ASSERT_VALUES_EQUAL(1, static_cast<int>(*configCounter));
    }

    Y_UNIT_TEST(ShouldGetFatalErrorIfEndpointHasInvalidSocketPath)
    {
        auto logging = CreateLoggingService("console");
        InitVhostLog(logging);

        auto vhostServer = CreateServer(
            logging,
            CreateServerStatsStub(),
            CreateVhostQueueFactory(),
            CreateDefaultDeviceHandlerFactory(),
            TServerConfig(),
            TVhostCallbacks());

        vhostServer->Start();

        TString socketPath("./invalid/path/to/socket");

        TStorageOptions options;
        options.DiskId = "TestDiskId";
        options.BlockSize = DefaultBlockSize;
        options.BlocksCount = 42;
        options.VhostQueuesCount = 1;
        options.UnalignedRequestsDisabled = false;

        auto future = vhostServer->StartEndpoint(
            socketPath,
            std::make_shared<TTestStorage>(),
            options);

        const auto& error = future.GetValue(TDuration::Seconds(5));
        UNIT_ASSERT_VALUES_EQUAL_C(
            EErrorKind::ErrorFatal,
            GetErrorKind(error),
            error);

        vhostServer->Stop();
    }

    Y_UNIT_TEST(ShouldStartEndpointIfSocketAlreadyExists)
    {
        auto logging = CreateLoggingService("console");
        InitVhostLog(logging);

        auto vhostQueueFactory = CreateVhostQueueFactory();

        auto vhostServer = CreateServer(
            logging,
            CreateServerStatsStub(),
            vhostQueueFactory,
            CreateDefaultDeviceHandlerFactory(),
            TServerConfig(),
            TVhostCallbacks());

        vhostServer->Start();

        const TFsPath socket(CreateGuidAsString() + ".sock");
        socket.Touch();
        Y_DEFER {
            socket.DeleteIfExists();
        };

        {
            TStorageOptions options;
            options.DiskId = "TestDiskId";
            options.BlockSize = DefaultBlockSize;
            options.BlocksCount = 42;
            options.VhostQueuesCount = 1;
            options.UnalignedRequestsDisabled = false;

            auto future = vhostServer->StartEndpoint(
                socket.GetPath(),
                std::make_shared<TTestStorage>(),
                options);
            const auto& error = future.GetValue(TDuration::Seconds(5));
            UNIT_ASSERT_C(!HasError(error), error);
        }

        vhostServer->Stop();
    }

    Y_UNIT_TEST(ShouldRemoveUnixSocketAfterStopEndpoint)
    {
        auto logging = CreateLoggingService("console");
        InitVhostLog(logging);

        auto vhostQueueFactory = CreateVhostQueueFactory();

        auto vhostServer = CreateServer(
            logging,
            CreateServerStatsStub(),
            vhostQueueFactory,
            CreateDefaultDeviceHandlerFactory(),
            TServerConfig(),
            TVhostCallbacks());

        vhostServer->Start();

        const TFsPath socket(CreateGuidAsString() + ".sock");

        {
            TStorageOptions options;
            options.DiskId = "TestDiskId";
            options.BlockSize = DefaultBlockSize;
            options.BlocksCount = 42;
            options.VhostQueuesCount = 1;
            options.UnalignedRequestsDisabled = false;

            auto future = vhostServer->StartEndpoint(
                socket.GetPath(),
                std::make_shared<TTestStorage>(),
                options);
            const auto& error = future.GetValue(TDuration::Seconds(5));
            UNIT_ASSERT_C(!HasError(error), error);
        }

        auto future = vhostServer->StopEndpoint(socket.GetPath());
        const auto& error = future.GetValue(TDuration::Seconds(5));
        UNIT_ASSERT_C(!HasError(error), error);
        UNIT_ASSERT(!socket.Exists());

        vhostServer->Stop();
    }

    Y_UNIT_TEST(ShouldNotRemoveUnixSocketAfterStopServer)
    {
        auto logging = CreateLoggingService("console");
        InitVhostLog(logging);

        auto vhostQueueFactory = CreateVhostQueueFactory();

        auto vhostServer = CreateServer(
            logging,
            CreateServerStatsStub(),
            vhostQueueFactory,
            CreateDefaultDeviceHandlerFactory(),
            TServerConfig(),
            TVhostCallbacks());

        vhostServer->Start();

        const TFsPath socket(CreateGuidAsString() + ".sock");

        {
            TStorageOptions options;
            options.DiskId = "TestDiskId";
            options.BlockSize = DefaultBlockSize;
            options.BlocksCount = 42;
            options.VhostQueuesCount = 1;
            options.UnalignedRequestsDisabled = false;

            auto future = vhostServer->StartEndpoint(
                socket.GetPath(),
                std::make_shared<TTestStorage>(),
                options);
            const auto& error = future.GetValue(TDuration::Seconds(5));
            UNIT_ASSERT_C(!HasError(error), error);
        }

        vhostServer->Stop();
        UNIT_ASSERT(socket.Exists());
    }

    Y_UNIT_TEST(ShouldCancelRequestsInFlightWhenStopEndpointOrStopServer)
    {
        TString unixSocketPath =
            MakeTempName(nullptr, CreateGuidAsString().c_str(), "sock");
        const ui32 blockSize = 4096;
        const ui64 startIndex = 3;
        const ui64 blocksCount = 41;

        auto promise = NewPromise<void>();

        auto testStorage = std::make_shared<TTestStorage>();
        testStorage->WriteBlocksLocalHandler =
            [&] (TCallContextPtr ctx, std::shared_ptr<NProto::TWriteBlocksLocalRequest> request) {
                Y_UNUSED(ctx);
                Y_UNUSED(request);
                return promise.GetFuture().Apply([] (const auto& f) {
                    Y_UNUSED(f);
                    return NProto::TWriteBlocksLocalResponse();
                });
            };
        testStorage->ReadBlocksLocalHandler =
            [&] (TCallContextPtr ctx, std::shared_ptr<NProto::TReadBlocksLocalRequest> request) {
                Y_UNUSED(ctx);
                Y_UNUSED(request);
                return promise.GetFuture().Apply([] (const auto& f) {
                    Y_UNUSED(f);
                    return NProto::TReadBlocksLocalResponse();
                });
            };

        auto queueFactory = std::make_shared<TTestVhostQueueFactory>();

        TServerConfig serverConfig;
        serverConfig.ThreadsCount = 2;

        size_t fatalErrorCount = 0;
        auto serverStats = std::make_shared<TTestServerStats>();
        serverStats->RequestCompletedHandler = [&] (
            TLog& log,
            TMetricRequest& metricRequest,
            TCallContext& callContext,
            const NProto::TError& error)
        {
            Y_UNUSED(log);
            Y_UNUSED(metricRequest);
            Y_UNUSED(callContext);
            if (GetDiagnosticsErrorKind(error)
                    == EDiagnosticsErrorKind::ErrorFatal)
            {
                ++fatalErrorCount;
            }
        };

        auto server = CreateServer(
            CreateLoggingService("console"),
            serverStats,
            queueFactory,
            CreateDefaultDeviceHandlerFactory(),
            serverConfig,
            TVhostCallbacks());

        server->Start();
        Sleep(TDuration::MilliSeconds(300));
        UNIT_ASSERT(queueFactory->Queues.size() == serverConfig.ThreadsCount);
        auto firstQueue = queueFactory->Queues.at(0);
        UNIT_ASSERT(firstQueue->IsRun());

        TStorageOptions options;
        options.DiskId = "testDiskId";
        options.BlockSize = blockSize;
        options.BlocksCount = 256;
        options.VhostQueuesCount = 1;
        options.UnalignedRequestsDisabled = false;

        {
            auto future = server->StartEndpoint(
                unixSocketPath,
                testStorage,
                options);
            const auto& error = future.GetValue(TDuration::Seconds(5));
            UNIT_ASSERT_C(!HasError(error), error);
        }
        UNIT_ASSERT(firstQueue->GetDevices().size() == 1);
        auto device = firstQueue->GetDevices().at(0);

        TVector<TString> blocks;
        auto sgList = ResizeBlocks(
            blocks,
            blocksCount,
            TString(blockSize, 'f'));

        auto writeFuture = device->SendTestRequest(
            EBlockStoreRequest::WriteBlocks,
            startIndex * blockSize,
            blocksCount * blockSize,
            sgList);

        auto readFuture = device->SendTestRequest(
            EBlockStoreRequest::ReadBlocks,
            startIndex * blockSize,
            blocksCount * blockSize,
            sgList);

        Sleep(TDuration::MilliSeconds(300));
        UNIT_ASSERT(!writeFuture.HasValue());
        UNIT_ASSERT(!readFuture.HasValue());
        UNIT_ASSERT_VALUES_EQUAL(0, fatalErrorCount);

        {
            device->DisableAutostop(true);
            auto future = server->StopEndpoint(unixSocketPath);

            for (size_t i = 0; i < 5; ++i) {
                auto type = (i % 2 == 0)
                    ? EBlockStoreRequest::WriteBlocks
                    : EBlockStoreRequest::ReadBlocks;
                auto reqFuture = device->SendTestRequest(
                    type,
                    startIndex * blockSize,
                    blocksCount * blockSize,
                    sgList);
                auto response = reqFuture.GetValue(TDuration::Seconds(5));
                UNIT_ASSERT(response == TVhostRequest::CANCELLED);
            }

            UNIT_ASSERT(!future.HasValue());
            device->DisableAutostop(false);
            auto response = future.GetValue(TDuration::Seconds(5));
            UNIT_ASSERT(!HasError(response));
        }

        auto writeResponse = writeFuture.GetValue(TDuration::Seconds(5));
        UNIT_ASSERT(writeResponse == TVhostRequest::CANCELLED);
        auto readResponse = readFuture.GetValue(TDuration::Seconds(5));
        UNIT_ASSERT(readResponse == TVhostRequest::CANCELLED);
        UNIT_ASSERT_VALUES_EQUAL(0, fatalErrorCount);

        {
            auto future = server->StartEndpoint(
                unixSocketPath,
                testStorage,
                options);
            const auto& error = future.GetValue(TDuration::Seconds(5));
            UNIT_ASSERT_C(!HasError(error), error);
        }
        device.reset();
        UNIT_ASSERT(firstQueue->GetDevices().size() == 1);
        device = firstQueue->GetDevices().at(0);

        writeFuture = device->SendTestRequest(
            EBlockStoreRequest::WriteBlocks,
            startIndex * blockSize,
            blocksCount * blockSize,
            sgList);

        readFuture = device->SendTestRequest(
            EBlockStoreRequest::ReadBlocks,
            startIndex * blockSize,
            blocksCount * blockSize,
            sgList);

        Sleep(TDuration::MilliSeconds(300));
        UNIT_ASSERT(!writeFuture.HasValue());
        UNIT_ASSERT(!readFuture.HasValue());
        UNIT_ASSERT_VALUES_EQUAL(0, fatalErrorCount);

        device->DisableAutostop(true);

        TManualEvent startEvent;
        TManualEvent stopEvent;
        SystemThreadFactory()->Run([&]() {
            startEvent.Signal();
            server->Stop();
            stopEvent.Signal();
        });
        startEvent.Wait();

        for (size_t i = 0; i < 5; ++i) {
            auto type = (i % 2 == 0)
                ? EBlockStoreRequest::WriteBlocks
                : EBlockStoreRequest::ReadBlocks;
            auto reqFuture = device->SendTestRequest(
                type,
                startIndex * blockSize,
                blocksCount * blockSize,
                sgList);
            auto response = reqFuture.GetValue(TDuration::Seconds(5));
            UNIT_ASSERT(response == TVhostRequest::CANCELLED);
        }
        device->DisableAutostop(false);

        writeResponse = writeFuture.GetValue(TDuration::Seconds(5));
        UNIT_ASSERT(writeResponse == TVhostRequest::CANCELLED);
        readResponse = readFuture.GetValue(TDuration::Seconds(5));
        UNIT_ASSERT(readResponse == TVhostRequest::CANCELLED);
        UNIT_ASSERT_VALUES_EQUAL(0, fatalErrorCount);

        stopEvent.Wait();
    }

    Y_UNIT_TEST(ShouldPassCorrectMetrics)
    {
        TString testDiskId = "testDiskId";
        const ui32 blockSize = 4096;
        const ui64 sectorSize = 512;
        ui64 firstSector = 0;
        ui64 totalSectors = 0;

        bool expectedUnaligned = false;
        ui64 expectedStartIndex = 0;
        ui64 expectedBlockCount = 0;

        UNIT_ASSERT(totalSectors * sectorSize % blockSize == 0);

        auto serverStats = std::make_shared<TTestServerStats>();

        ui32 requestCounter = 0;
        ui32 expectedRequestCounter = 0;

        serverStats->PrepareMetricRequestHandler = [&] (
            TMetricRequest& metricRequest,
            TString clientId,
            TString diskId,
            ui64 startIndex,
            ui32 requestBytes,
            bool unaligned)
        {
            Y_UNUSED(clientId);

            UNIT_ASSERT(diskId == testDiskId);
            metricRequest.DiskId = std::move(diskId);

            UNIT_ASSERT_VALUES_EQUAL(expectedUnaligned, unaligned);

            switch (metricRequest.RequestType)
            {
                case EBlockStoreRequest::ReadBlocks:
                case EBlockStoreRequest::WriteBlocks:
                case EBlockStoreRequest::ZeroBlocks:
                    UNIT_ASSERT_VALUES_EQUAL(expectedStartIndex, startIndex);
                    UNIT_ASSERT_VALUES_EQUAL(expectedBlockCount * blockSize, requestBytes);
                    break;
                case EBlockStoreRequest::MountVolume:
                case EBlockStoreRequest::UnmountVolume:
                    break;
                default:
                    UNIT_FAIL("Unexpected request");
                    break;
            }

            ++requestCounter;
        };

        auto testStorage = std::make_shared<TTestStorage>();
        testStorage->WriteBlocksLocalHandler =
            [&] (TCallContextPtr ctx, std::shared_ptr<NProto::TWriteBlocksLocalRequest> request) {
                Y_UNUSED(ctx);
                Y_UNUSED(request);
                return MakeFuture(NProto::TWriteBlocksLocalResponse());
            };
        testStorage->ReadBlocksLocalHandler =
            [&] (TCallContextPtr ctx, std::shared_ptr<NProto::TReadBlocksLocalRequest> request) {
                Y_UNUSED(ctx);
                Y_UNUSED(request);
                return MakeFuture(NProto::TReadBlocksLocalResponse());
            };

        auto queueFactory = std::make_shared<TTestVhostQueueFactory>();

        TServerConfig serverConfig;
        serverConfig.ThreadsCount = 2;

        auto server = CreateServer(
            CreateLoggingService("console"),
            serverStats,
            queueFactory,
            CreateDefaultDeviceHandlerFactory(),
            serverConfig,
            TVhostCallbacks());

        server->Start();
        Sleep(TDuration::MilliSeconds(300));
        UNIT_ASSERT(queueFactory->Queues.size() == serverConfig.ThreadsCount);
        auto firstQueue = queueFactory->Queues.at(0);
        UNIT_ASSERT(firstQueue->IsRun());

        {
            TStorageOptions options;
            options.DiskId = testDiskId;
            options.BlockSize = blockSize;
            options.BlocksCount = 256;
            options.VhostQueuesCount = 1;
            options.UnalignedRequestsDisabled = false;

            auto future = server->StartEndpoint(
                CreateGuidAsString() + ".sock",
                testStorage,
                options);
            const auto& error = future.GetValue(TDuration::Seconds(5));
            UNIT_ASSERT_C(!HasError(error), error);
        }
        UNIT_ASSERT(firstQueue->GetDevices().size() == 1);
        auto device = firstQueue->GetDevices().at(0);

        auto testIoRequets = [&] () {
            TVector<TString> blocks;
            auto sgList = ResizeBlocks(
                blocks,
                totalSectors,
                TString(sectorSize, 'f'));

            {
                auto future = device->SendTestRequest(
                    EBlockStoreRequest::WriteBlocks,
                    firstSector * sectorSize,
                    totalSectors * sectorSize,
                    sgList);
                const auto& response = future.GetValue(TDuration::Seconds(5));
                UNIT_ASSERT(response == TVhostRequest::SUCCESS);
                UNIT_ASSERT_VALUES_EQUAL(++expectedRequestCounter, requestCounter);
            }

            {
                auto future = device->SendTestRequest(
                    EBlockStoreRequest::ReadBlocks,
                    firstSector * sectorSize,
                    totalSectors * sectorSize,
                    sgList);
                const auto& response = future.GetValue(TDuration::Seconds(5));
                UNIT_ASSERT(response == TVhostRequest::SUCCESS);
                UNIT_ASSERT_VALUES_EQUAL(++expectedRequestCounter, requestCounter);
            }
        };

        firstSector = 8;
        totalSectors = 32;
        expectedUnaligned = false;
        expectedStartIndex = 1;
        expectedBlockCount = 4;
        testIoRequets();

        firstSector = 5;
        totalSectors = 16;
        expectedUnaligned = true;
        expectedStartIndex = 0;
        expectedBlockCount = 3;
        testIoRequets();

        firstSector = 16;
        totalSectors = 29;
        expectedUnaligned = true;
        expectedStartIndex = 2;
        expectedBlockCount = 4;
        testIoRequets();

        firstSector = 13;
        totalSectors = 11;
        expectedUnaligned = true;
        expectedStartIndex = 1;
        expectedBlockCount = 2;
        testIoRequets();
    }

    Y_UNIT_TEST(ShouldNotBeRaceOnStopEndpoint)
    {
        TString unixSocketPath = CreateGuidAsString() + ".sock";
        const ui32 blockSize = 4096;
        const ui64 startIndex = 3;
        const ui64 blocksCount = 2;

        TManualEvent handleRequestEvent;
        TManualEvent stopEndpointEvent;

        auto promise = NewPromise<NProto::TWriteBlocksLocalResponse>();

        auto testStorage = std::make_shared<TTestStorage>();
        testStorage->WriteBlocksLocalHandler =
            [&] (TCallContextPtr ctx, std::shared_ptr<NProto::TWriteBlocksLocalRequest> request) {
                Y_UNUSED(ctx);
                Y_UNUSED(request);
                handleRequestEvent.Signal();
                stopEndpointEvent.Wait();
                return promise.GetFuture();
            };

        auto queueFactory = std::make_shared<TTestVhostQueueFactory>();

        TServerConfig serverConfig;
        serverConfig.ThreadsCount = 2;

        auto server = CreateServer(
            CreateLoggingService("console"),
            CreateServerStatsStub(),
            queueFactory,
            CreateDefaultDeviceHandlerFactory(),
            serverConfig,
            TVhostCallbacks());

        server->Start();
        Sleep(TDuration::MilliSeconds(300));
        UNIT_ASSERT(queueFactory->Queues.size() == serverConfig.ThreadsCount);
        auto firstQueue = queueFactory->Queues.at(0);
        UNIT_ASSERT(firstQueue->IsRun());

        TStorageOptions options;
        options.DiskId = "testDiskId";
        options.BlockSize = blockSize;
        options.BlocksCount = 256;
        options.VhostQueuesCount = 1;
        options.UnalignedRequestsDisabled = false;

        {
            auto future = server->StartEndpoint(
                unixSocketPath,
                testStorage,
                options);
            const auto& error = future.GetValue(TDuration::Seconds(5));
            UNIT_ASSERT_C(!HasError(error), error);
        }
        UNIT_ASSERT(firstQueue->GetDevices().size() == 1);
        auto device = firstQueue->GetDevices().at(0);

        auto collector = CreateIncompleteRequestsCollectorStub();
        UNIT_ASSERT_VALUES_EQUAL(0, server->CollectRequests(collector));

        TVector<TString> blocks;
        auto sgList = ResizeBlocks(
            blocks,
            blocksCount,
            TString(blockSize, 'f'));

        auto future1 = device->SendTestRequest(
            EBlockStoreRequest::WriteBlocks,
            startIndex * blockSize,
            blocksCount * blockSize,
            sgList);

        handleRequestEvent.Wait();
        UNIT_ASSERT_VALUES_EQUAL(1, server->CollectRequests(collector));

        auto future2 = device->SendTestRequest(
            EBlockStoreRequest::WriteBlocks,
            startIndex * blockSize,
            blocksCount * blockSize,
            sgList);

        {
            auto future = server->StopEndpoint(unixSocketPath);
            UNIT_ASSERT_VALUES_EQUAL(0, server->CollectRequests(collector));

            stopEndpointEvent.Signal();

            auto response = future.GetValue(TDuration::Seconds(5));
            UNIT_ASSERT(!HasError(response));
        }

        future1.GetValue(TDuration::Seconds(5));
        future2.GetValue(TDuration::Seconds(5));

        server->Stop();
    }

    Y_UNIT_TEST(ShouldHandleVhostZeroBlocksRequests)
    {
        const ui32 blockSize = 4096;
        const ui64 firstSector = 8;
        const ui64 totalSectors = 32;
        const ui64 sectorSize = 512;

        UNIT_ASSERT(totalSectors * sectorSize % blockSize == 0);

        auto environment = TTestEnvironment(blockSize);
        auto device = environment.GetVhostDevice();

        {
            auto future = device->SendTestRequest(
                EBlockStoreRequest::ZeroBlocks,
                firstSector * sectorSize,
                totalSectors * sectorSize,
                {},
                true /* isDiscardRequest */);
            const auto& response = future.GetValue(TDuration::Seconds(5));
            UNIT_ASSERT(response == TVhostRequest::SUCCESS);

            TTestRequest request;
            bool res = environment.DequeueRequest(request);
            UNIT_ASSERT(res);
            UNIT_ASSERT(request.Type == EBlockStoreRequest::ZeroBlocks);
            UNIT_ASSERT(
                request.StartIndex * blockSize == firstSector * sectorSize);
            UNIT_ASSERT(
                request.BlocksCount * blockSize == totalSectors * sectorSize);
            UNIT_ASSERT(!environment.DequeueRequest(request));
        }

        {
            auto future = device->SendTestRequest(
                EBlockStoreRequest::ZeroBlocks,
                firstSector * sectorSize,
                totalSectors * sectorSize,
                {},
                false /* isDiscardRequest */);
            const auto& response = future.GetValue(TDuration::Seconds(5));
            UNIT_ASSERT(response == TVhostRequest::SUCCESS);

            TTestRequest request;
            bool res = environment.DequeueRequest(request);
            UNIT_ASSERT(res);
            UNIT_ASSERT(request.Type == EBlockStoreRequest::ZeroBlocks);
            UNIT_ASSERT(
                request.StartIndex * blockSize == firstSector * sectorSize);
            UNIT_ASSERT(
                request.BlocksCount * blockSize == totalSectors * sectorSize);
            UNIT_ASSERT(!environment.DequeueRequest(request));
        }
    }

    Y_UNIT_TEST(ShouldDropDiscardRequestsIfNeeded)
    {
        const ui32 blockSize = 4096;
        const ui64 firstSector = 8;
        const ui64 totalSectors = 32;
        const ui64 sectorSize = 512;

        UNIT_ASSERT(totalSectors * sectorSize % blockSize == 0);

        auto environment =
            TTestEnvironment(blockSize, true /* dropDiscardRequests */);
        auto device = environment.GetVhostDevice();

        {
            auto future = device->SendTestRequest(
                EBlockStoreRequest::ZeroBlocks,
                firstSector * sectorSize,
                totalSectors * sectorSize,
                {},
                true /* isDiscardRequest */);
            const auto& response = future.GetValue(TDuration::Seconds(5));
            UNIT_ASSERT(response == TVhostRequest::SUCCESS);

            // Should drop the discard request.
            TTestRequest request;
            UNIT_ASSERT(!environment.DequeueRequest(request));
        }

        {
            auto future = device->SendTestRequest(
                EBlockStoreRequest::ZeroBlocks,
                firstSector * sectorSize,
                totalSectors * sectorSize,
                {},
                false /* isDiscardRequest */);
            const auto& response = future.GetValue(TDuration::Seconds(5));
            UNIT_ASSERT(response == TVhostRequest::SUCCESS);

            // This is not a discard request but a write zeroes request.
            // Should not drop it.
            TTestRequest request;
            bool res = environment.DequeueRequest(request);
            UNIT_ASSERT(res);
            UNIT_ASSERT(request.Type == EBlockStoreRequest::ZeroBlocks);
            UNIT_ASSERT(
                request.StartIndex * blockSize == firstSector * sectorSize);
            UNIT_ASSERT(
                request.BlocksCount * blockSize == totalSectors * sectorSize);
            UNIT_ASSERT(!environment.DequeueRequest(request));
        }
    }
}

}   // namespace NCloud::NBlockStore::NVhost