diff options
author | alexv-smirnov <alex@ydb.tech> | 2022-08-18 16:52:30 +0300 |
---|---|---|
committer | alexv-smirnov <alex@ydb.tech> | 2022-08-18 16:52:30 +0300 |
commit | c140abc954b61ab7d86af80bdeced01482d9971a (patch) | |
tree | c47d70fa3213240d5e0eb59787a5325782a360de /library/cpp/testing/mock_server/server.h | |
parent | 0ce07b9705ed20e3fce2759eae41496014ca4c33 (diff) | |
download | ydb-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.h | 52 |
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; + } + }; +} |