fix tests

ydb-platform · Jan 30, 2024 · c46a764 · c46a764
1 parent 1a82150
commit c46a764
Show file tree

Hide file tree

Showing 4 changed files with 37 additions and 4 deletions.
diff --git a/cloud/blockstore/libs/endpoints/endpoint_manager_ut.cpp b/cloud/blockstore/libs/endpoints/endpoint_manager_ut.cpp
@@ -15,7 +15,7 @@
 #include <cloud/blockstore/libs/encryption/encryption_client.h>
 #include <cloud/blockstore/libs/encryption/encryption_key.h>
 #include <cloud/blockstore/libs/endpoints_grpc/socket_endpoint_listener.h>
-#include <cloud/blockstore/libs/server/client_acceptor.h>
+#include <cloud/blockstore/libs/server/client_storage_factory.h>
 #include <cloud/blockstore/libs/service/service_test.h>
 #include <cloud/blockstore/libs/service/storage_provider.h>
 
@@ -913,7 +913,7 @@ Y_UNIT_TEST_SUITE(TEndpointManagerTest)
         TBootstrap bootstrap(CreateTestService(mountedVolumes));
 
         auto grpcListener = CreateSocketEndpointListener(bootstrap.Logging, 16);
-        grpcListener->SetClientAcceptor(CreateClientAcceptorStub());
+        grpcListener->SetClientStorageFactory(CreateClientStorageFactoryStub());
 
         auto manager = CreateEndpointManager(
             bootstrap,

diff --git a/cloud/blockstore/libs/server/client_storage_factory.cpp b/cloud/blockstore/libs/server/client_storage_factory.cpp
@@ -1 +1,29 @@
 #include "client_storage_factory.h"
+
+namespace NCloud::NBlockStore::NServer {
+
+using namespace NCloud::NStorage::NServer;
+
+////////////////////////////////////////////////////////////////////////////////
+
+struct TClientStorageFactoryStub
+    : public IClientStorageFactory
+{
+    NStorage::NServer::IClientStoragePtr CreateClientStorage(
+        IBlockStorePtr service) override
+    {
+        Y_UNUSED(service);
+        return CreateClientStorageStub();
+    }
+};
+
+////////////////////////////////////////////////////////////////////////////////
+
+IClientStorageFactoryPtr CreateClientStorageFactoryStub()
+{
+    return std::make_shared<TClientStorageFactoryStub>();
+}
+
+////////////////////////////////////////////////////////////////////////////////
+
+}   // namespace NCloud::NBlockStore::NServer
diff --git a/cloud/blockstore/libs/server/client_storage_factory.h b/cloud/blockstore/libs/server/client_storage_factory.h
@@ -1,5 +1,7 @@
 #pragma once
 
+#include "public.h"
+
 #include <cloud/blockstore/libs/service/public.h>
 
 #include <cloud/storage/core/libs/uds/client_storage.h>
@@ -18,4 +20,8 @@ struct IClientStorageFactory
 
 ////////////////////////////////////////////////////////////////////////////////
 
+IClientStorageFactoryPtr CreateClientStorageFactoryStub();
+
+////////////////////////////////////////////////////////////////////////////////
+
 }   // namespace NCloud::NBlockStore::NServer
diff --git a/cloud/blockstore/libs/server/public.h b/cloud/blockstore/libs/server/public.h
@@ -19,8 +19,7 @@ struct IServer;
 using IServerPtr = std::shared_ptr<IServer>;
 
 struct IClientStorageFactory;
-using IClientStorageFactoryPtr =
-    std::shared_ptr<IClientStorageFactory>;
+using IClientStorageFactoryPtr = std::shared_ptr<IClientStorageFactory>;
 
 }   // namespace NServer
 }   // namespace NCloud::NBlockStore