aboutsummaryrefslogtreecommitdiffstats
path: root/library/cpp/testing/mock_server/server.h
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.h
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.h')
-rw-r--r--library/cpp/testing/mock_server/server.h52
1 files changed, 52 insertions, 0 deletions
diff --git a/library/cpp/testing/mock_server/server.h b/library/cpp/testing/mock_server/server.h
new file mode 100644
index 0000000000..6d57cebf40
--- /dev/null
+++ b/library/cpp/testing/mock_server/server.h
@@ -0,0 +1,52 @@
+#pragma once
+
+#include <library/cpp/http/misc/parsed_request.h>
+#include <library/cpp/http/server/http.h>
+#include <library/cpp/http/server/response.h>
+#include <library/cpp/testing/unittest/tests_data.h>
+
+#include <util/system/event.h>
+
+#include <functional>
+
+class TAutoEvent;
+class TThread;
+
+namespace NMock {
+ class TMockServer {
+ public:
+ using TGenerator = std::function<TRequestReplier*()>;
+
+ TMockServer(ui16 port, TGenerator generator);
+ TMockServer(const THttpServerOptions& options, TGenerator generator);
+
+ ~TMockServer();
+
+ size_t GetClientCount() const;
+ void SetGenerator(TGenerator generator);
+
+ private:
+ static void* Worker(void* arg);
+
+ class TCallBack;
+
+ THolder<TCallBack> Cb_;
+ THolder<THttpServer> Server_;
+ THolder<TThread> Thread_;
+ THolder<TAutoEvent> Ev_;
+ };
+
+ class TPong: public TRequestReplier {
+ public:
+ bool DoReply(const TReplyParams& params) override {
+ const TParsedHttpFull parsed(params.Input.FirstLine());
+
+ const HttpCodes code = parsed.Path == "/ping" ? HTTP_OK : HTTP_NOT_FOUND;
+
+ THttpResponse resp(code);
+ resp.OutTo(params.Output);
+
+ return true;
+ }
+ };
+}