git.m6w6.name Git - m6w6/libmemcached/blob - testing/lib/common.hpp

   1 #pragma once
   2
   3 #include <chrono>
   4 #include <iostream>
   5 #include <map>
   6 #include <optional>
   7 #include <string>
   8 #include <sstream>
   9 #include <stdexcept>
  10 #include <thread>
  11 #include <variant>
  12 #include <vector>
  13
  14 #include "../lib/catch.hpp"
  15
  16 #include "libmemcached/memcached.h"
  17
  18 using namespace std;
  19 using socket_or_port_t = variant<string, int>;
  20
  21 /**
  22  * Useful macros for testing
  23  */
  24 #define S(s) (s),strlen(s)
  25 #define LOOPED_SECTION(tests) \
  26   for (auto &[name, test] : tests) DYNAMIC_SECTION("test " << name)
  27 #define REQUIRE_SUCCESS(rc) REQUIRE_THAT(rc, test.returns.success())
  28 #define REQUIRE_RC(rc, call) REQUIRE_THAT(call, test.returns(rc))
  29
  30
  31 const char *getenv_else(const char *var, const char *defval);
  32 unsigned random_num(unsigned min, unsigned max);
  33 unsigned random_port();
  34 string random_socket();
  35 string random_socket_or_port_string(const string &what);
  36 string random_socket_or_port_flag(const string &binary);
  37
  38 inline auto random_socket_or_port_arg() {
  39   return make_pair(&random_socket_or_port_flag, &random_socket_or_port_string);
  40 }
  41
  42 class MemcachedPtr {
  43 public:
  44   memcached_st memc;
  45
  46   explicit
  47   MemcachedPtr(memcached_st *memc_) {
  48     memset(&memc, 0, sizeof(memc));
  49     REQUIRE(memcached_clone(&memc, memc_));
  50   }
  51   MemcachedPtr()
  52   : MemcachedPtr(nullptr)
  53   {}
  54   ~MemcachedPtr() {
  55     memcached_free(&memc);
  56   }
  57   memcached_st *operator * () {
  58     return &memc;
  59   }
  60 };
  61
  62 class Malloced {
  63   void *ptr;
  64 public:
  65   Malloced(void *ptr_)
  66   : ptr{ptr_}
  67   {}
  68   ~Malloced() {
  69     free(ptr);
  70   }
  71   void *operator *() {
  72     return ptr;
  73   }
  74 };