X-Git-Url: http://git.indexdata.com/?a=blobdiff_plain;f=src%2Ftest_boost_threads.cpp;h=6e4880148e8a5a1686f2cb21f5f70f79612615ec;hb=da353f48b13fd79e446570d120369423f9c496dd;hp=8a26cab98212e70766b02f5d734f6d09aef013dd;hpb=c90fa1d9d80ba80198953b34c2e5a3134d1b5402;p=metaproxy-moved-to-github.git

diff --git a/src/test_boost_threads.cpp b/src/test_boost_threads.cpp
index 8a26cab..6e48801 100644
--- a/src/test_boost_threads.cpp
+++ b/src/test_boost_threads.cpp
@@ -1,60 +1,103 @@
+/* $Id: test_boost_threads.cpp,v 1.9 2007-01-25 14:05:54 adam Exp $
+   Copyright (c) 2005-2007, Index Data.
 
+   See the LICENSE file for details
+ */
+
+#include "config.hpp"
 #include <boost/thread/mutex.hpp>
 #include <boost/thread/thread.hpp>
-#include <iostream>
 
-boost::mutex io_mutex; // The iostreams are not guaranteed to be thread-safe!
+#define BOOST_AUTO_TEST_MAIN
+#include <boost/test/auto_unit_test.hpp>
+
+#include <list>
+#include <iostream>
 
 class counter
 {
    public:
       counter() : count(0) { }
       
-      int increment() {
-         boost::mutex::scoped_lock scoped_lock(mutex);
-         return ++count;
-      }
-      
-   private:
-      boost::mutex mutex;
-      int count;
+    int increment() {
+        boost::mutex::scoped_lock scoped_lock(mutex);
+        return ++count;
+    }
+    
+private:
+    boost::mutex mutex;
+    int count;
 };
 
+
 counter c;
 
-void change_count()
+class worker {
+public:
+    void operator() (void) {
+        c.increment();
+    }
+};
+
+#define USE_GROUP 1
+
+
+BOOST_AUTO_UNIT_TEST( thread_group )
 {
-   int i = c.increment();
-   boost::mutex::scoped_lock scoped_lock(io_mutex);
-   std::cout << "count == " << i << std::endl;
+    try 
+    {
+        const int num_threads = 4;
+        boost::thread_group thrds;
+        
+        for (int i=0; i < num_threads; ++i)
+        {
+            worker w;
+            thrds.add_thread(new boost::thread(w));
+        }
+        thrds.join_all();
+    }
+    catch (...) 
+    {
+        BOOST_CHECK(false);
+    }
+    BOOST_CHECK(c.increment() == 5);
 }
 
 
-
-int main(int, char*[])
+BOOST_AUTO_UNIT_TEST( thread_list )
 {
-   try 
-   {
-      const int num_threads = 4;
-      boost::thread_group thrds;
-      for (int i=0; i < num_threads; ++i)
-         thrds.create_thread(&change_count);
-      
-      thrds.join_all();
-      
-   }
-   catch (std::exception &e) 
-   {
-      std::cout << e.what() << "\n";
-      exit(1);
-   }
-   exit(0);
+    try 
+    {
+        const int num_threads = 4;
+        std::list<boost::thread *> thread_list;
+        
+        for (int i=0; i < num_threads; ++i)
+        {
+            worker w;
+            thread_list.push_back(new boost::thread(w));
+        }
+        std::list<boost::thread *>::iterator it;
+        for (it = thread_list.begin(); it != thread_list.end(); it++)
+        {
+            (*it)->join();
+            delete *it;
+        }
+
+    }
+    catch (...) 
+    {
+        BOOST_CHECK(false);
+    }
+    BOOST_CHECK(c.increment() == 10);
 }
 
+
+
 /*
  * Local variables:
  * c-basic-offset: 4
  * indent-tabs-mode: nil
+ * c-file-style: "stroustrup"
  * End:
  * vim: shiftwidth=4 tabstop=8 expandtab
  */