1 files changed, 189 insertions, 0 deletions
diff --git a/cpp/test/Ice/retry/InstrumentationI.cpp b/cpp/test/Ice/retry/InstrumentationI.cpp
new file mode 100644
index 00000000000..5a61c6cb3ef
--- /dev/null
+++ b/cpp/test/Ice/retry/InstrumentationI.cpp
@@ -0,0 +1,189 @@
+// **********************************************************************
+//
+// Copyright (c) 2003-2014 ZeroC, Inc. All rights reserved.
+//
+// This copy of Ice is licensed to you under the terms described in the
+// ICE_LICENSE file included in this distribution.
+//
+// **********************************************************************
+
+#include <Ice/Ice.h>
+#include <TestCommon.h>
+
+namespace
+{
+
+int nRetry = 0; 
+int nFailure = 0;
+int nInvocation = 0;
+IceUtil::Mutex* staticMutex = 0;
+
+class Init
+{
+public:
+
+    Init()
+    {
+        staticMutex = new IceUtil::Mutex;
+    }
+
+    ~Init()
+    {
+        delete staticMutex;
+        staticMutex = 0;
+    }
+};
+Init init;
+
+class InvocationObserverI : public Ice::Instrumentation::InvocationObserver
+{
+public:
+
+    virtual void 
+    attach()
+    {
+    }
+
+    virtual void 
+    detach()
+    {
+        IceUtilInternal::MutexPtrLock<IceUtil::Mutex> lock(staticMutex);
+        ++nInvocation;
+    }
+
+    virtual void 
+    failed(const ::std::string&)
+    {
+        IceUtilInternal::MutexPtrLock<IceUtil::Mutex> lock(staticMutex);
+        ++nFailure;
+    }
+
+    virtual void 
+    retried()
+    {
+        IceUtilInternal::MutexPtrLock<IceUtil::Mutex> lock(staticMutex);
+        ++nRetry;
+    }
+
+    virtual void 
+    userException()
+    {
+    }
+
+    virtual ::Ice::Instrumentation::RemoteObserverPtr 
+    getRemoteObserver(const ::Ice::ConnectionInfoPtr&, const ::Ice::EndpointPtr&, ::Ice::Int, ::Ice::Int)
+    {
+        return 0;
+    }
+
+    virtual ::Ice::Instrumentation::RemoteObserverPtr 
+    getCollocatedObserver(::Ice::Int, ::Ice::Int)
+    {
+        return 0;
+    }
+
+};
+
+Ice::Instrumentation::InvocationObserverPtr invocationObserver = new InvocationObserverI();
+
+class CommunicatorObserverI : public Ice::Instrumentation::CommunicatorObserver
+{
+public:
+
+    virtual Ice::Instrumentation::ObserverPtr 
+    getConnectionEstablishmentObserver(const Ice::EndpointPtr&, const ::std::string&)
+    {
+        return 0;
+    }
+
+    virtual Ice::Instrumentation::ObserverPtr 
+    getEndpointLookupObserver(const Ice::EndpointPtr&)
+    {
+        return 0;
+    }
+
+    virtual Ice::Instrumentation::ConnectionObserverPtr 
+    getConnectionObserver(const Ice::ConnectionInfoPtr&, 
+                          const Ice::EndpointPtr&, 
+                          Ice::Instrumentation::ConnectionState,
+                          const Ice::Instrumentation::ConnectionObserverPtr&)
+    {
+        return 0;
+    }
+
+    virtual Ice::Instrumentation::ThreadObserverPtr 
+    getThreadObserver(const ::std::string&, 
+                      const ::std::string&, 
+                      Ice::Instrumentation::ThreadState, 
+                      const Ice::Instrumentation::ThreadObserverPtr&)
+    {
+        return 0;
+    }
+
+    virtual Ice::Instrumentation::InvocationObserverPtr 
+    getInvocationObserver(const Ice::ObjectPrx&, const ::std::string&, const Ice::Context&)
+    {
+        return invocationObserver;
+    }
+
+    virtual Ice::Instrumentation::DispatchObserverPtr 
+    getDispatchObserver(const Ice::Current&, Ice::Int)
+    {
+        return 0;
+    }
+
+    virtual void 
+    setObserverUpdater(const Ice::Instrumentation::ObserverUpdaterPtr&)
+    {
+    }
+};
+
+Ice::Instrumentation::CommunicatorObserverPtr communicatorObserver = new CommunicatorObserverI();
+
+void
+testEqual(int& value, int expected)
+{
+    int retry = 0;
+    while(++retry < 100)
+    {
+        {
+            IceUtilInternal::MutexPtrLock<IceUtil::Mutex> lock(staticMutex);
+            if(value == expected)
+            {
+                break;
+            }
+        }
+        IceUtil::ThreadControl::sleep(IceUtil::Time::milliSeconds(10));
+    }
+    if(value != expected)
+    {
+        std::cerr << "value = " << value << ", expected = " << expected << std::endl;
+        test(false);
+    }
+}
+
+}
+
+Ice::Instrumentation::CommunicatorObserverPtr
+getObserver()
+{
+    return communicatorObserver;
+}
+
+void
+testRetryCount(int expected)
+{
+    testEqual(nRetry, expected);
+}
+
+void
+testFailureCount(int expected)
+{
+    testEqual(nFailure, expected);
+}
+
+void
+testInvocationCount(int expected)
+{
+    testEqual(nInvocation, expected);
+}