winamp/Src/external_dependencies/cpr/test/abstractServer.hpp
2024-09-24 14:54:57 +02:00

71 lines
1.9 KiB
C++

#ifndef CPR_TEST_ABSTRACT_SERVER_SERVER_H
#define CPR_TEST_ABSTRACT_SERVER_SERVER_H
#include <atomic>
#include <condition_variable>
#include <gtest/gtest.h>
#include <memory>
#include <mutex>
#include <string>
#include "cpr/cpr.h"
#include "mongoose.h"
namespace cpr {
// Helper struct for functions using timers to simulate slow connections
struct TimerArg {
mg_mgr* mgr;
mg_connection* connection;
unsigned long connection_id;
mg_timer timer;
unsigned counter;
explicit TimerArg(mg_mgr* m, mg_connection* c, mg_timer&& t) : mgr{m}, connection{c}, connection_id{0}, timer{t}, counter{0} {}
~TimerArg() {
mg_timer_free(&mgr->timers, &timer);
}
};
class AbstractServer : public testing::Environment {
public:
~AbstractServer() override = default;
void SetUp() override;
void TearDown() override;
void Start();
void Stop();
virtual std::string GetBaseUrl() = 0;
virtual uint16_t GetPort() = 0;
virtual void acceptConnection(mg_connection* conn) = 0;
virtual void OnRequest(mg_connection* conn, mg_http_message* msg) = 0;
private:
std::shared_ptr<std::thread> serverThread{nullptr};
std::mutex server_mutex;
std::condition_variable server_start_cv;
std::condition_variable server_stop_cv;
std::atomic<bool> should_run{false};
void Run();
protected:
mg_mgr mgr{};
std::vector<std::unique_ptr<TimerArg>> timer_args{};
virtual mg_connection* initServer(mg_mgr* mgr, mg_event_handler_t event_handler) = 0;
static std::string Base64Decode(const std::string& in);
static void SendError(mg_connection* conn, int code, std::string& reason);
static bool IsConnectionActive(mg_mgr* mgr, mg_connection* conn);
static uint16_t GetRemotePort(const mg_connection* conn);
static uint16_t GetLocalPort(const mg_connection* conn);
};
} // namespace cpr
#endif