Ydb stable 22-4-4322.4.43

x-stable-origin-commit: 8d49d46cc834835bf3e50870516acd7376a63bcf

1 files changed, 70 insertions, 0 deletions

diff --git a/library/cpp/testing/mock_server/server.cpp b/library/cpp/testing/mock_server/server.cpp
new file mode 100644
index 00000000000..d310c76ad32
--- /dev/null
+++ b/library/cpp/testing/mock_server/server.cpp
@@ -0,0 +1,70 @@
+#include "server.h"
+
+#include <util/system/thread.h>
+
+namespace NMock {
+    class TMockServer::TCallBack: public THttpServer::ICallBack {
+        TGenerator Generator_;
+
+        TClientRequest* CreateClient() override {
+            return Generator_();
+        }
+
+    public:
+        void SetGenerator(TGenerator generator) {
+            Generator_ = generator;
+        }
+
+        TCallBack(TGenerator& generator)
+            : Generator_(generator)
+        {
+        }
+    };
+
+    static THttpServerOptions createDefaultOptions(ui16 port) {
+        THttpServerOptions o;
+        o.AddBindAddress("localhost", port);
+        o.SetThreads(1);
+        o.SetMaxConnections(300);
+        o.SetMaxQueueSize(30);
+        return o;
+    }
+
+    TMockServer::TMockServer(ui16 port, TGenerator generator)
+        : TMockServer(createDefaultOptions(port), generator)
+    {
+    }
+
+    TMockServer::TMockServer(const THttpServerOptions& options, TGenerator generator)
+        : Cb_(MakeHolder<TCallBack>(generator))
+        , Server_(MakeHolder<THttpServer>(Cb_.Get(), options))
+        , Thread_(MakeHolder<TThread>(Worker, this))
+        , Ev_(MakeHolder<TAutoEvent>())
+    {
+        Thread_->Start();
+        Ev_->Wait();
+    }
+
+    TMockServer::~TMockServer() {
+        Server_->Stop();
+        Thread_->Join();
+    }
+
+    size_t TMockServer::GetClientCount() const {
+        return Server_->GetClientCount();
+    }
+
+    void TMockServer::SetGenerator(TMockServer::TGenerator generator) {
+        Cb_->SetGenerator(generator);
+    }
+
+    void* TMockServer::Worker(void* arg) {
+        TMockServer& this_ = *static_cast<TMockServer*>(arg);
+
+        this_.Server_->Start();
+        this_.Ev_->Signal();
+        this_.Server_->Wait();
+
+        return nullptr;
+    }
+}