1 files changed, 97 insertions, 0 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]);
+}