1 files changed, 83 insertions, 0 deletions
diff --git a/cpp/src/tests/Cluster.h b/cpp/src/tests/Cluster.h
new file mode 100644
index 0000000000..7ca5445e10
--- /dev/null
+++ b/cpp/src/tests/Cluster.h
@@ -0,0 +1,83 @@
+#ifndef CLUSTER_H
+#define CLUSTER_H
+
+/*
+ *
+ * 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.
+ *
+ */
+
+#include "qpid/cluster/Cluster.h"
+#include "qpid/framing/AMQFrame.h"
+#include "qpid/framing/ChannelOkBody.h"
+#include "qpid/framing/BasicGetOkBody.h"
+#include "qpid/log/Logger.h"
+#include <iostream>
+#include <vector>
+
+/**
+ * Definitions for the Cluster.cpp and Cluster_child.cpp child program.
+ */
+
+// using namespace in header file is bad manners, but this is strictly for
+// the tests.
+using namespace std;
+using namespace qpid;
+using namespace qpid::cluster;
+using namespace qpid::framing;
+using namespace qpid::sys;
+
+struct TestCluster : public Cluster {
+    TestCluster(const std::string& name,
+                const std::string& url,
+                framing::FrameHandler& next,
+                framing::ProtocolVersion ver) : Cluster(name,url,next, ver) {}
+
+    /** Wait for the cluster to be of expected size (exactly) */
+    bool waitFor(size_t n) {
+        Mutex::ScopedLock l(lock);
+        AbsTime deadline(now(),2*TIME_SEC);
+        while(size() != n && lock.wait(deadline))
+            ;
+        return size() == n;
+    }
+};
+    
+struct VectorFrameHandler :
+    public std::vector<AMQFrame>, public FrameHandler, public Monitor
+    
+{
+    void handle(AMQFrame& f) {
+        ScopedLock l(*this);
+        push_back(f);
+        notifyAll();
+    }
+
+    /** Wait for vector to reach size n exactly */
+    bool waitFor(size_t n) {
+        ScopedLock l(*this);
+        AbsTime deadline(now(), 1*TIME_SEC);
+        while (size() != n && wait(deadline))
+            ;
+        return size() == n;
+    }
+};
+
+
+// namespace 
+
+
+
+#endif  /*!CLUSTER_H*/