X-Git-Url: https://git.m6w6.name/?a=blobdiff_plain;f=test%2Flib%2FMemcachedCluster.hpp;h=0a4dcb6bc0a2b08ac3c57b71029174200599c210;hb=59d7373e0c1e0be8d8f3cc3dcf3c1df48554741d;hp=39859b7d90aee4e9bd162ba305a84210a10abf42;hpb=9f262c9ea92d4869715ca6f534c80075a8310ac1;p=m6w6%2Flibmemcached

diff --git a/test/lib/MemcachedCluster.hpp b/test/lib/MemcachedCluster.hpp
index 39859b7d..0a4dcb6b 100644
--- a/test/lib/MemcachedCluster.hpp
+++ b/test/lib/MemcachedCluster.hpp
@@ -7,35 +7,42 @@
 
 class MemcachedCluster {
 public:
+  using behavior_t = pair<memcached_behavior_t, uint64_t>;
+  using behaviors_t = vector<behavior_t>;
+
   Cluster cluster;
   memcached_st memc{empty_memc};
   ReturnMatcher returns{&memc};
 
   MemcachedCluster();
-  explicit
-  MemcachedCluster(Cluster &&cluster);
+  explicit MemcachedCluster(Cluster &&cluster, behaviors_t to_set = {});
   ~MemcachedCluster();
 
   MemcachedCluster(const MemcachedCluster &) = delete;
   MemcachedCluster &operator=(const MemcachedCluster &) = delete;
 
-  MemcachedCluster(MemcachedCluster &&mc);;
+  MemcachedCluster(MemcachedCluster &&mc);
   MemcachedCluster &operator=(MemcachedCluster &&mc);
 
   void enableBinaryProto(bool enable = true);
   void enableBuffering(bool enable = true);
   void enableReplication();
+  void enableUdp(bool enable = true);
   void flush();
 
   static MemcachedCluster mixed();
   static MemcachedCluster network();
   static MemcachedCluster socket();
+  static MemcachedCluster udp();
 
 #if LIBMEMCACHED_WITH_SASL_SUPPORT
   static MemcachedCluster sasl();
 #endif
 
+  void killOneServer();
+
 private:
+  behaviors_t to_set;
   static const memcached_st empty_memc;
 
   void init();