Fix it so socket is built just on main tests.
[m6w6/libmemcached] / libtest / test.cc
index a1300a8ad2fd1302b1dfdeac228e3c682af44e31..d71057d36bdf14f8f93457a644d5a96b771585fe 100644 (file)
@@ -1,13 +1,40 @@
-/* uTest
- * Copyright (C) 2011 Data Differential, http://datadifferential.com/
- * Copyright (C) 2006-2009 Brian Aker
- * All rights reserved.
+/*  vim:expandtab:shiftwidth=2:tabstop=2:smarttab:
+ * 
+ *  uTest, libtest
+ *
+ *  Copyright (C) 2011 Data Differential, http://datadifferential.com/
+ *  Copyright (C) 2006-2009 Brian Aker
+ *
+ *  Redistribution and use in source and binary forms, with or without
+ *  modification, are permitted provided that the following conditions are
+ *  met:
+ *
+ *      * Redistributions of source code must retain the above copyright
+ *  notice, this list of conditions and the following disclaimer.
+ *
+ *      * Redistributions in binary form must reproduce the above
+ *  copyright notice, this list of conditions and the following disclaimer
+ *  in the documentation and/or other materials provided with the
+ *  distribution.
+ *
+ *      * The names of its contributors may not be used to endorse or
+ *  promote products derived from this software without specific prior
+ *  written permission.
+ *
+ *  THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
+ *  "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
+ *  LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
+ *  A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
+ *  OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
+ *  SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
+ *  LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
+ *  DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
+ *  THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
+ *  (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
+ *  OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
  *
- * Use and distribution licensed under the BSD license.  See
- * the COPYING file in the parent directory for full text.
  */
 
-
 #include <libtest/common.h>
 
 #include <cassert>
 #include <fnmatch.h>
 #include <iostream>
 
+#include <signal.h>
+
 #include <libtest/stats.h>
+#include <libtest/signal.h>
 
 #ifndef __INTEL_COMPILER
 #pragma GCC diagnostic ignored "-Wold-style-cast"
 #endif
 
+using namespace libtest;
+
 static in_port_t global_port= 0;
 static char global_socket[1024];
 
 in_port_t default_port()
 {
-  assert(global_port);
   return global_port;
 }
  
@@ -47,23 +78,31 @@ const char *default_socket()
   assert(global_socket[0]);
   return global_socket;
 }
+
+bool test_is_local()
+{
+  return (getenv("LIBTEST_LOCAL"));
+}
+
 void set_default_socket(const char *socket)
 {
-  strncpy(global_socket, socket, strlen(socket));
+  if (socket)
+  {
+    strncpy(global_socket, socket, strlen(socket));
+  }
 }
 
 static void stats_print(Stats *stats)
 {
-  std::cout << "\tTotal Collections\t\t\t\t" << stats->collection_total << std::endl;
-  std::cout << "\tFailed Collections\t\t\t\t" << stats->collection_failed << std::endl;
-  std::cout << "\tSkipped Collections\t\t\t\t" << stats->collection_skipped << std::endl;
-  std::cout << "\tSucceeded Collections\t\t\t\t" << stats->collection_success << std::endl;
-  std::cout << std::endl;
-  std::cout << "Total\t\t\t\t" << stats->total << std::endl;
-  std::cout << "\tFailed\t\t\t" << stats->failed << std::endl;
-  std::cout << "\tSkipped\t\t\t" << stats->skipped << std::endl;
-  std::cout << "\tSucceeded\t\t" << stats->success << std::endl;
+  Out << "\tTotal Collections\t\t\t\t" << stats->collection_total;
+  Out << "\tFailed Collections\t\t\t\t" << stats->collection_failed;
+  Out << "\tSkipped Collections\t\t\t\t" << stats->collection_skipped;
+  Out << "\tSucceeded Collections\t\t\t\t" << stats->collection_success;
+  Outn();
+  Out << "Total\t\t\t\t" << stats->total;
+  Out << "\tFailed\t\t\t" << stats->failed;
+  Out << "\tSkipped\t\t\t" << stats->skipped;
+  Out << "\tSucceeded\t\t" << stats->success;
 }
 
 static long int timedif(struct timeval a, struct timeval b)
@@ -116,60 +155,34 @@ void create_core(void)
   }
 }
 
-
-static test_return_t _runner_default(test_callback_fn func, void *p)
-{
-  if (func)
-  {
-    return func(p);
-  }
-
-  return TEST_SUCCESS;
-}
-
-static Runner defualt_runners= {
-  _runner_default,
-  _runner_default,
-  _runner_default
-};
-
-static test_return_t _default_callback(void *p)
-{
-  (void)p;
-
-  return TEST_SUCCESS;
-}
-
-Framework::Framework() :
-  collections(NULL),
-  _create(NULL),
-  _destroy(NULL),
-  collection_startup(_default_callback),
-  collection_shutdown(_default_callback),
-  _on_error(NULL),
-  runner(&defualt_runners)
-{
-}
-
-
+static Framework *world= NULL;
 int main(int argc, char *argv[])
 {
-  Framework *world= new Framework();
+  srandom((unsigned int)time(NULL));
+
+  world= new Framework();
 
   if (not world)
   {
     return EXIT_FAILURE;
   }
 
+  libtest::SignalThread signal;
+  if (not signal.setup())
+  {
+    return EXIT_FAILURE;
+  }
+
   Stats stats;
 
   get_world(world);
 
   test_return_t error;
-  void *world_ptr= world->create(&error);
+  void *creators_ptr= world->create(error);
   if (test_failed(error))
   {
-    std::cerr << "create() failed" << std::endl;
+    Error << "create() failed";
+    delete world;
     return EXIT_FAILURE;
   }
 
@@ -185,7 +198,7 @@ int main(int argc, char *argv[])
 
   if (collection_to_run)
   {
-    std::cout << "Only testing " <<  collection_to_run << std::endl;
+    Out << "Only testing " <<  collection_to_run;
   }
 
   char *wildcard= NULL;
@@ -194,7 +207,7 @@ int main(int argc, char *argv[])
     wildcard= argv[2];
   }
 
-  for (collection_st *next= world->collections; next->name; next++)
+  for (collection_st *next= world->collections; next->name and (not signal.is_shutdown()); next++)
   {
     test_return_t collection_rc= TEST_SUCCESS;
     bool failed= false;
@@ -205,34 +218,36 @@ int main(int argc, char *argv[])
 
     stats.collection_total++;
 
-    collection_rc= world->startup(world_ptr);
+    collection_rc= world->startup(creators_ptr);
 
     if (collection_rc == TEST_SUCCESS and next->pre)
     {
-      collection_rc= world->runner->pre(next->pre, world_ptr);
+      collection_rc= world->runner()->pre(next->pre, creators_ptr);
     }
 
     switch (collection_rc)
     {
     case TEST_SUCCESS:
-      std::cerr << std::endl << next->name << std::endl << std::endl;
       break;
 
     case TEST_FATAL:
     case TEST_FAILURE:
-      std::cerr << std::endl << next->name << " [ failed ]" << std::endl << std::endl;
-      stats.collection_failed++;
+      Out << next->name << " [ failed ]";
+      failed= true;
+      signal.set_shutdown(SHUTDOWN_GRACEFUL);
       goto cleanup;
 
     case TEST_SKIPPED:
-      std::cerr << std::endl << next->name << " [ skipping ]" << std::endl << std::endl;
-      stats.collection_skipped++;
+      Out << next->name << " [ skipping ]";
+      skipped= true;
       goto cleanup;
 
     case TEST_MEMORY_ALLOCATION_FAILURE:
       test_assert(0, "Allocation failure, or unknown return");
     }
 
+    Out << "Collection: " << next->name;
+
     for (test_st *run= next->tests; run->name; run++)
     {
       struct timeval start_time, end_time;
@@ -240,108 +255,133 @@ int main(int argc, char *argv[])
 
       if (wildcard && fnmatch(wildcard, run->name, 0))
       {
-       continue;
+        continue;
       }
 
-      std::cerr << "\tTesting " << run->name;
-
       test_return_t return_code;
-      if (test_success(return_code= world->item.startup(world_ptr)))
+      if (test_success(return_code= world->item.startup(creators_ptr)))
       {
-        if (test_success(return_code= world->item.flush(world_ptr, run)))
+        if (test_success(return_code= world->item.flush(creators_ptr, run)))
         {
           // @note pre will fail is SKIPPED is returned
-          if (test_success(return_code= world->item.pre(world_ptr)))
+          if (test_success(return_code= world->item.pre(creators_ptr)))
           {
             { // Runner Code
               gettimeofday(&start_time, NULL);
-              return_code= world->runner->run(run->test_fn, world_ptr);
+              assert(world->runner());
+              assert(run->test_fn);
+              return_code= world->runner()->run(run->test_fn, creators_ptr);
               gettimeofday(&end_time, NULL);
               load_time= timedif(end_time, start_time);
             }
           }
 
           // @todo do something if post fails
-          (void)world->item.post(world_ptr);
+          (void)world->item.post(creators_ptr);
         }
-        else
+        else if (return_code == TEST_SKIPPED)
+        { }
+        else if (return_code == TEST_FAILURE)
         {
-          std::cerr << __FILE__ << ":" << __LINE__ << " item.flush(failure)" << std::endl;
+          Error << " item.flush(failure)";
+          signal.set_shutdown(SHUTDOWN_GRACEFUL);
         }
       }
-      else
+      else if (return_code == TEST_SKIPPED)
+      { }
+      else if (return_code == TEST_FAILURE)
       {
-        std::cerr << __FILE__ << ":" << __LINE__ << " item.startup(failure)" << std::endl;
+        Error << " item.startup(failure)";
+        signal.set_shutdown(SHUTDOWN_GRACEFUL);
       }
 
       stats.total++;
 
-      std::cerr << "\t\t\t\t\t";
-
       switch (return_code)
       {
       case TEST_SUCCESS:
-       std::cerr << load_time / 1000 << "." << load_time % 1000;
-       stats.success++;
-       break;
+        Out << "\tTesting " << run->name <<  "\t\t\t\t\t" << load_time / 1000 << "." << load_time % 1000 << "[ " << test_strerror(return_code) << " ]";
+        stats.success++;
+        break;
 
       case TEST_FATAL:
       case TEST_FAILURE:
-       stats.failed++;
-       failed= true;
-       break;
+        stats.failed++;
+        failed= true;
+        Out << "\tTesting " << run->name <<  "\t\t\t\t\t" << "[ " << test_strerror(return_code) << " ]";
+        break;
 
       case TEST_SKIPPED:
-       stats.skipped++;
-       skipped= true;
-       break;
+        stats.skipped++;
+        skipped= true;
+        Out << "\tTesting " << run->name <<  "\t\t\t\t\t" << "[ " << test_strerror(return_code) << " ]";
+        break;
 
       case TEST_MEMORY_ALLOCATION_FAILURE:
-       test_assert(0, "Memory Allocation Error");
+        test_assert(0, "Memory Allocation Error");
       }
 
-      std::cerr << "[ " << test_strerror(return_code) << " ]" << std::endl;
-
-      if (test_failed(world->on_error(return_code, world_ptr)))
+      if (test_failed(world->on_error(return_code, creators_ptr)))
       {
+        Error << "Failed while running on_error()";
+        signal.set_shutdown(SHUTDOWN_GRACEFUL);
         break;
       }
     }
 
-    if (next->post and world->runner->post)
+    (void) world->runner()->post(next->post, creators_ptr);
+
+cleanup:
+    if (failed == false and skipped == false)
     {
-      (void) world->runner->post(next->post, world_ptr);
+      stats.collection_success++;
     }
 
-    if (failed == 0 and skipped == 0)
+    if (failed)
     {
-      stats.collection_success++;
+      stats.collection_failed++;
+    }
+
+    if (skipped)
+    {
+      stats.collection_skipped++;
     }
-cleanup:
 
-    world->shutdown(world_ptr);
+    world->shutdown(creators_ptr);
+    Outn();
   }
 
-  if (stats.collection_failed || stats.collection_skipped)
+  if (not signal.is_shutdown())
   {
-    std::cerr << std::endl << std::endl <<  "Some test failures and/or skipped test occurred." << std::endl << std::endl;
-#if 0
-    print_failed_test();
-#endif
+    signal.set_shutdown(SHUTDOWN_GRACEFUL);
   }
-  else
+
+  int exit_code= EXIT_SUCCESS;
+  shutdown_t status= signal.get_shutdown();
+  if (status == SHUTDOWN_FORCED)
   {
-    std::cout << std::endl << std::endl <<  "All tests completed successfully." << std::endl << std::endl;
+    Out << "Tests were aborted.";
+    exit_code= EXIT_FAILURE;
   }
-
-  if (test_failed(world->destroy(world_ptr)))
+  else if (stats.collection_failed)
+  {
+    Out << "Some test failed.";
+    exit_code= EXIT_FAILURE;
+  }
+  else if (stats.collection_skipped)
   {
-    stats.failed++; // We do this to make our exit code return EXIT_FAILURE
+    Out << "Some tests were skipped.";
+  }
+  else
+  {
+    Out << "All tests completed successfully.";
   }
 
   stats_print(&stats);
 
   delete world;
 
-  return stats.failed == 0 ? EXIT_SUCCESS : EXIT_FAILURE;
+  Outn(); // Generate a blank to break up the messages if make check/test has been run
+
+  return exit_code;
 }