aboutsummaryrefslogtreecommitdiffstats
path: root/library/cpp/testing/mock_server/server.cpp
diff options
context:
space:
mode:
authoralexv-smirnov <alex@ydb.tech>2022-08-18 16:52:30 +0300
committeralexv-smirnov <alex@ydb.tech>2022-08-18 16:52:30 +0300
commitc140abc954b61ab7d86af80bdeced01482d9971a (patch)
treec47d70fa3213240d5e0eb59787a5325782a360de /library/cpp/testing/mock_server/server.cpp
parent0ce07b9705ed20e3fce2759eae41496014ca4c33 (diff)
downloadydb-c140abc954b61ab7d86af80bdeced01482d9971a.tar.gz
temp fix ydb oss sync config to unlock sync on /vendor dependency
Diffstat (limited to 'library/cpp/testing/mock_server/server.cpp')
-rw-r--r--library/cpp/testing/mock_server/server.cpp70
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 0000000000..d310c76ad3
--- /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;
+ }
+}