cloud/blockstore/libs/client/client_ut.cpp - main - githubmirror/nbs

cloud/blockstore/libs/client/client_ut.cpp

72 строки2 KB

yegorskii

Refactoring. Extracted server&client TLS functionality into separate class (#5723)

28 май 2026, 10:48

28 май 2026, 10:48a059f24

#include "session.h"

#include "client.h"
#include "config.h"

#include <cloud/blockstore/libs/diagnostics/server_stats.h>
#include <cloud/blockstore/libs/service/context.h>
#include <cloud/blockstore/libs/service/service_test.h>
#include <cloud/storage/core/libs/common/error.h>
#include <cloud/storage/core/libs/common/scheduler_test.h>
#include <cloud/storage/core/libs/common/timer.h>
#include <cloud/storage/core/libs/diagnostics/logging.h>
#include <cloud/storage/core/libs/diagnostics/monitoring.h>
#include <cloud/storage/core/libs/grpc/tls_certificate_provider.h>

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

#include <util/folder/path.h>
#include <util/generic/guid.h>
#include <util/generic/scope.h>

namespace NCloud::NBlockStore::NClient {

using namespace NThreading;

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

Y_UNIT_TEST_SUITE(TClientTest)
{
    Y_UNIT_TEST(ShouldReturnNotImplementedErrorStatusCodeForUnsupportedRequestsFromDataEndpoint)
    {
        TPortManager portManager;
        ui16 dataPort = portManager.GetPort(9001);

        NProto::TClientAppConfig clientAppConfig;
        auto& clientConfig = *clientAppConfig.MutableClientConfig();
        clientConfig.SetPort(dataPort);
        clientConfig.SetClientId(CreateGuidAsString());

        auto result = CreateClient(
            std::make_shared<TClientAppConfig>(clientAppConfig),
            CreateWallClockTimer(),
            CreateSchedulerStub(),
            CreateLoggingService("console"),
            CreateMonitoringServiceStub(),
            CreateServerStatsStub(),
            CreateCertificateProviderStub());

        auto client = result.ExtractResult();
        UNIT_ASSERT_C(client, FormatError(result.GetError()));

        auto clientEndpoint = client->CreateDataEndpoint();

        client->Start();
        clientEndpoint->Start();
        Y_DEFER {
            clientEndpoint->Stop();
            client->Stop();
        };

        auto futurePing = clientEndpoint->Ping(
            MakeIntrusive<TCallContext>(),
            std::make_shared<NProto::TPingRequest>());
        UNIT_ASSERT(futurePing.HasValue());
        UNIT_ASSERT_VALUES_EQUAL(
            E_NOT_IMPLEMENTED,
            futurePing.GetValue().GetError().GetCode());
    }
}

}   // namespace NCloud::NBlockStore::NClient