From 9ce1a0267694910094f492bd204d2166290a2427 Mon Sep 17 00:00:00 2001
From: Brian Aker <brian@gaz>
Date: Fri, 18 Dec 2009 18:25:46 -0800
Subject: [PATCH] Update test.c

---
 tests/test.c | 7 +++----
 1 file changed, 3 insertions(+), 4 deletions(-)

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)
-- 
2.30.2