From: Brian Aker <brian@gaz>
Date: Sat, 19 Dec 2009 02:25:46 +0000 (-0800)
Subject: Update test.c
X-Git-Tag: 0.37~49
X-Git-Url: https://git.m6w6.name/?a=commitdiff_plain;h=9ce1a0267694910094f492bd204d2166290a2427;p=awesomized%2Flibmemcached

Update test.c
---

diff --git a/tests/test.c b/tests/test.c
index 44f18eea..a10cc476 100644
--- a/tests/test.c
+++ b/tests/test.c
@@ -19,7 +19,6 @@
 #include <unistd.h>
 #include <time.h>
 #include <fnmatch.h>
-#include "server.h"
 
 #include "test.h"
 
@@ -172,7 +171,7 @@ int main(int argc, char *argv[])
       }
 
 
-      if (next->pre)
+      if (next->pre && world.runner->pre)
       {
         return_code= world.runner->pre(next->pre, world_ptr);
 
@@ -187,9 +186,9 @@ int main(int argc, char *argv[])
       gettimeofday(&end_time, NULL);
       load_time= timedif(end_time, start_time);
 
-      if (next->post)
+      if (next->post && world.runner->post)
       {
-        (void) world.runner->pre(next->pre, world_ptr);
+        (void) world.runner->post(next->post, world_ptr);
       }
 
       if (world.post_run)