Unit test for PollableCondition.

git-svn-id: https://svn.apache.org/repos/asf/qpid/trunk/qpid@767666 13f79535-47bb-0310-9956-ffa450edef68

2 files changed, 119 insertions, 1 deletions

diff --git a/cpp/src/tests/Makefile.am b/cpp/src/tests/Makefile.am
index 26a13fea48..161428fcad 100644
--- a/cpp/src/tests/Makefile.am
+++ b/cpp/src/tests/Makefile.am
@@ -97,7 +97,8 @@ unit_test_SOURCES= unit_test.cpp unit_test.h \
 	RateFlowcontrolTest.cpp \
 	FrameDecoder.cpp \
 	ReplicationTest.cpp \
-	ClientMessageTest.cpp
+	ClientMessageTest.cpp \
+	PollableCondition.cpp
 
 if HAVE_XML
 unit_test_SOURCES+= XmlClientSessionTest.cpp
diff --git a/cpp/src/tests/PollableCondition.cpp b/cpp/src/tests/PollableCondition.cpp
new file mode 100644
index 0000000000..843262087a
--- /dev/null
+++ b/cpp/src/tests/PollableCondition.cpp
@@ -0,0 +1,117 @@
+/*
+ *
+ * Licensed to the Apache Software Foundation (ASF) under one
+ * or more contributor license agreements.  See the NOTICE file
+ * distributed with this work for additional information
+ * regarding copyright ownership.  The ASF licenses this file
+ * to you 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 "test_tools.h"
+#include "unit_test.h"
+#include "qpid/sys/Poller.h"
+#include "qpid/sys/PollableCondition.h"
+#include "qpid/sys/Monitor.h"
+#include "qpid/sys/Time.h"
+#include "qpid/sys/Thread.h"
+#include <boost/bind.hpp>
+
+
+QPID_AUTO_TEST_SUITE(PollableConditionTest)
+
+using namespace qpid::sys;
+
+const Duration SHORT = TIME_SEC/100;
+const Duration LONG = TIME_SEC/10;
+
+class  Callback {
+  public:    
+    enum Action { NONE, DISARM, CLEAR, DISARM_CLEAR };
+
+    Callback() : count() {}
+
+    void call(PollableCondition& pc) {
+        Mutex::ScopedLock l(lock);
+        ++count;
+        switch(action) {
+          case NONE: break; 
+          case DISARM:  pc.disarm(); break;
+          case CLEAR: pc.clear(); break;
+          case DISARM_CLEAR: pc.disarm(); pc.clear(); break;
+        }
+        action = NONE;
+        lock.notify();
+    }
+
+    bool isCalling() { Mutex::ScopedLock l(lock); return wait(LONG); }
+
+    bool isNotCalling() { Mutex::ScopedLock l(lock); return !wait(SHORT); }
+
+    bool nextCall(Action a=NONE) {
+        Mutex::ScopedLock l(lock);
+        action = a;
+        return wait(LONG);
+    }
+    
+  private:
+    bool wait(Duration timeout) {        
+        int n = count;
+        AbsTime deadline(now(), timeout);
+        while (n == count && lock.wait(deadline))
+               ;
+        return n != count;
+    }
+
+    Monitor lock;
+    int count;
+    Action action;
+};
+
+QPID_AUTO_TEST_CASE(testPollableCondition) {
+    boost::shared_ptr<Poller> poller(new Poller());
+    Callback callback;
+    PollableCondition pc(boost::bind(&Callback::call, &callback, _1), poller);
+
+    Thread runner = Thread(*poller);
+    
+    BOOST_CHECK(callback.isNotCalling()); // condition is not set or armed.
+
+    pc.rearm();                          
+    BOOST_CHECK(callback.isNotCalling()); // Armed but not set
+
+    pc.set();
+    BOOST_CHECK(callback.isCalling()); // Armed and set.
+    BOOST_CHECK(callback.isCalling()); // Still armed and set.
+
+    callback.nextCall(Callback::DISARM);
+    BOOST_CHECK(callback.isNotCalling()); // set but not armed
+
+    pc.rearm();
+    BOOST_CHECK(callback.isCalling()); // Armed and set.
+    callback.nextCall(Callback::CLEAR);    
+    BOOST_CHECK(callback.isNotCalling()); // armed but not set
+
+    pc.set();
+    BOOST_CHECK(callback.isCalling()); // Armed and set.
+    callback.nextCall(Callback::DISARM_CLEAR);    
+    BOOST_CHECK(callback.isNotCalling()); // not armed or set.
+
+    poller->shutdown();
+    runner.join();
+}
+
+QPID_AUTO_TEST_SUITE_END()
+
+