simple_server.hpp

Go to the documentation of this file.

#pragma once
 
/// @file userver/utest/simple_server.hpp
/// @brief @copybrief utest::SimpleServer
 
#include <cstdint>
#include <functional>
#include <initializer_list>
#include <memory>
#include <string>
#include <vector>
 
USERVER_NAMESPACE_BEGIN
 
namespace utest {
 
/// @ingroup userver_utest
///
/// Toy server for simple network testing.
///
/// In constructor opens specified ports in localhost address and listens on
/// them. On each accepted data packet calls user callback.
///
/// ## Example usage:
/// @snippet testing_test.cpp  Sample SimpleServer usage
class SimpleServer final {
 public:
  struct Response {
    enum Commands {
      kWriteAndClose,
      kTryReadMore,
      kWriteAndContinue,
    };
 
    std::string data_to_send{};
    Commands command{kWriteAndClose};
  };
 
  using Request = std::string;
  using OnRequest = std::function<Response(const Request&)>;
 
  using Port = unsigned short;
  enum Protocol { kTcpIpV4, kTcpIpV6 };
 
  SimpleServer(OnRequest callback, Protocol protocol = kTcpIpV4);
  ~SimpleServer();
 
  Port GetPort() const;
 
  enum class Schema {
    kHttp,
    kHttps,
  };
 
  std::string GetBaseUrl(Schema type = Schema::kHttp) const;
 
  std::uint64_t GetConnectionsOpenedCount() const;
 
 private:
  class Impl;
  const std::unique_ptr<Impl> pimpl_;
};
 
}  // namespace utest
 
USERVER_NAMESPACE_END