3 files changed, 99 insertions, 1 deletions
diff --git a/cpp/src/tests/unit/Cpg.cpp b/cpp/src/tests/unit/Cpg.cpp
new file mode 100644
index 0000000000..74c6532338
--- /dev/null
+++ b/cpp/src/tests/unit/Cpg.cpp
@@ -0,0 +1,97 @@
+/*
+ *
+ * Copyright (c) 2006 The Apache Software Foundation
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *    http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ *
+ */
+
+#define BOOST_AUTO_TEST_MAIN    // Must come before #include<boost/test/*>
+#include "test_tools.h"
+#include "qpid/cluster/Cpg.h"
+#include <string>
+#include <iostream>
+#include <iterator>
+#include <vector>
+#include <algorithm>
+
+using namespace std;
+using namespace qpid::cluster;
+
+// For debugging: op << for CPG types.
+
+ostream& operator<<(ostream& o, const cpg_name* n) {
+    return o << qpid::cluster::Cpg::str(*n);
+}
+
+ostream& operator<<(ostream& o, const cpg_address& a) {
+    return o << "(" << a.nodeid <<","<<a.pid<<","<<a.reason<<")";
+}
+
+template <class T>
+ostream& operator<<(ostream& o, const pair<T*, int>& array) {
+    o << "{ ";
+    ostream_iterator<cpg_address> i(o, " ");
+    copy(array.first, array.first+array.second, i);
+    cout << "}";
+    return o;
+}
+
+const string testGroup("foo");
+vector<string> delivered;
+vector<int> configChanges;
+
+void testDeliver (
+    cpg_handle_t /*handle*/,
+    struct cpg_name *group,
+    uint32_t /*nodeid*/,
+    uint32_t /*pid*/,
+    void* msg,
+    int msg_len)
+{
+    BOOST_CHECK_EQUAL(testGroup, Cpg::str(*group));
+    delivered.push_back(string((char*)msg,msg_len));
+}
+
+void testConfigChange(
+    cpg_handle_t /*handle*/,
+    struct cpg_name *group,
+    struct cpg_address */*members*/, int nMembers,
+    struct cpg_address */*left*/, int /*nLeft*/,
+    struct cpg_address */*joined*/, int /*nJoined*/
+)
+{
+    BOOST_CHECK_EQUAL(testGroup, Cpg::str(*group));
+    configChanges.push_back(nMembers);
+}
+
+BOOST_AUTO_TEST_CASE(basic) {
+    // Verify basic functionality of cpg. This will catch any
+    // openais configuration or permission errors.
+    // 
+    Cpg cpg(&testDeliver, &testConfigChange);
+    Cpg::Name group("foo");
+
+    cpg.join(group);
+    iovec iov = { (void*)"Hello!", 6 };
+    cpg.mcast(group, &iov, 1);
+    cpg.leave(group);
+
+    cpg.dispatch(CPG_DISPATCH_ONE); // Wait for at least one.
+    cpg.dispatch(CPG_DISPATCH_ALL);
+    BOOST_REQUIRE_EQUAL(1u, delivered.size());
+    BOOST_CHECK_EQUAL("Hello!", delivered.front());
+    BOOST_REQUIRE_EQUAL(2u, configChanges.size());
+    BOOST_CHECK_EQUAL(1, configChanges[0]);
+    BOOST_CHECK_EQUAL(0, configChanges[1]);
+}
diff --git a/cpp/src/tests/unit/logging.cpp b/cpp/src/tests/unit/logging.cpp
index 3dbe68ad2b..c80bf7b337 100644
--- a/cpp/src/tests/unit/logging.cpp
+++ b/cpp/src/tests/unit/logging.cpp
@@ -16,6 +16,7 @@
  *
  */
 
+#define BOOST_AUTO_TEST_MAIN    // Must come before #include<boost/test/*>
 #include "test_tools.h"
 #include "qpid/log/Logger.h"
 #include "qpid/log/Options.h"
diff --git a/cpp/src/tests/unit/test_tools.h b/cpp/src/tests/unit/test_tools.h
index a941cf40eb..faa198af9a 100644
--- a/cpp/src/tests/unit/test_tools.h
+++ b/cpp/src/tests/unit/test_tools.h
@@ -19,7 +19,6 @@
  *
  */
 
-#define BOOST_AUTO_TEST_MAIN    // Must come before #include<boost/test/*>
 #include <boost/test/auto_unit_test.hpp>
 #include <boost/assign/list_of.hpp>
 #include <boost/regex.hpp>
@@ -51,3 +50,4 @@ inline bool regexPredicate(const std::string& re, const std::string& text) {
     BOOST_CHECK_PREDICATE(regexPredicate, (re)(text))
 
 #endif  /*!TEST_TOOLS_H*/
+