X-Git-Url: https://git.m6w6.name/?a=blobdiff_plain;f=tests%2Frunner.h;h=6c7eaaff91c07017466362891592ff909cf146ea;hb=5fb9e63a9f6af5fca174939f7c5fffa360ad75d0;hp=0aacac928c98e11d53ebdbc470c7d2a817bfa19a;hpb=a713b6d9ee850ae5805c05e8336256cea1f71138;p=awesomized%2Flibmemcached

diff --git a/tests/runner.h b/tests/runner.h
index 0aacac92..6c7eaaff 100644
--- a/tests/runner.h
+++ b/tests/runner.h
@@ -47,6 +47,20 @@ public:
     return _runner_default(libmemcached_test_callback_fn(func), (libmemcached_test_container_st*)object);
   }
 
+  test_return_t flush(void* arg)
+  {
+    return flush((libmemcached_test_container_st*)arg);
+  }
+
+  test_return_t flush(libmemcached_test_container_st *container)
+  {
+    test_true(container->memc);
+    memcached_flush(container->memc, 0);
+    memcached_quit(container->memc);
+
+    return TEST_SUCCESS;
+  }
+
   test_return_t pre(test_callback_fn* func, void *object)
   {
     return _pre_runner_default(libmemcached_test_callback_fn(func), (libmemcached_test_container_st*)object);
@@ -84,6 +98,8 @@ private:
 
   test_return_t _pre_runner_default(libmemcached_test_callback_fn func, libmemcached_test_container_st *container)
   {
+    test_compare(true, check());
+
     if (func)
     {
       return func(container->parent);
@@ -94,7 +110,9 @@ private:
 
   test_return_t _post_runner_default(libmemcached_test_callback_fn func, libmemcached_test_container_st *container)
   {
+    test_compare(true, check());
     cleanup_pairs(NULL);
+
     if (func)
     {
       return func(container->parent);