1 files changed, 30 insertions, 0 deletions
diff --git a/AudioManagerUtilities/test/AmSocketHandlerTest/CAmSocketHandlerTest.h b/AudioManagerUtilities/test/AmSocketHandlerTest/CAmSocketHandlerTest.h
index 63f79cf..d38f8c1 100644
--- a/AudioManagerUtilities/test/AmSocketHandlerTest/CAmSocketHandlerTest.h
+++ b/AudioManagerUtilities/test/AmSocketHandlerTest/CAmSocketHandlerTest.h
@@ -47,6 +47,15 @@ namespace am
         virtual void timerCallback(sh_timerHandle_t handle, void * userData)=0;
     };
 
+    class IAmSignalHandler
+    {
+    public:
+        virtual ~IAmSignalHandler()
+        {
+        }
+        virtual void signalHandlerAction(const sh_pollHandle_t handle, const unsigned sig, void* userData)=0;
+    };
+
     class IAmSocketHandlerCb
     {
     public:
@@ -65,6 +74,12 @@ namespace am
                 void(sh_timerHandle_t handle, void *userData));
     };
 
+    class MockIAmSignalHandler: public IAmSignalHandler
+    {
+    public:
+        MOCK_METHOD3(signalHandlerAction, void (const sh_pollHandle_t handle, const unsigned sig, void* userData));
+    };
+
     class MockSocketHandlerCb: public IAmSocketHandlerCb
     {
     public:
@@ -116,6 +131,21 @@ namespace am
         TAmShTimerCallBack<CAmTimerSockethandlerController> pTimerCallback;
     };
 
+    class CAmTimerSignalHandler: public MockIAmTimerCb
+    {
+        unsigned mIndex;
+        std::set<unsigned> mSignals;
+        CAmSocketHandler *mpSocketHandler;
+        timespec mUpdateTimeout;
+    public:
+        explicit CAmTimerSignalHandler(CAmSocketHandler *SocketHandler, const timespec &timeout, const std::set<unsigned> & signals);
+        virtual ~CAmTimerSignalHandler();
+
+        void timerCallback(sh_timerHandle_t handle, void * userData);
+
+        TAmShTimerCallBack<CAmTimerSignalHandler> pTimerCallback;
+    };
+
     class CAmTimer: public MockIAmTimerCb
     {
         CAmSocketHandler *mpSocketHandler;