1 files changed, 67 insertions, 5 deletions
diff --git a/AudioManagerUtilities/test/AmSocketHandlerTest/CAmSocketHandlerTest.h b/AudioManagerUtilities/test/AmSocketHandlerTest/CAmSocketHandlerTest.h
index ba2bf51..269e5da 100644
--- a/AudioManagerUtilities/test/AmSocketHandlerTest/CAmSocketHandlerTest.h
+++ b/AudioManagerUtilities/test/AmSocketHandlerTest/CAmSocketHandlerTest.h
@@ -101,20 +101,20 @@ namespace am
             UNIX, INET
         };
         CAmSamplePlugin(CAmSocketHandler *mySocketHandler, sockType_e socketType);
-        ~CAmSamplePlugin()
+        virtual ~CAmSamplePlugin()
         {
         }
         ;
         void connectSocket(const pollfd pollfd, const sh_pollHandle_t handle, void* userData);
-        void receiveData(const pollfd pollfd, const sh_pollHandle_t handle, void* userData);
-        bool dispatchData(const sh_pollHandle_t handle, void* userData);
-        bool check(const sh_pollHandle_t handle, void* userData);
+        virtual void receiveData(const pollfd pollfd, const sh_pollHandle_t handle, void* userData);
+        virtual bool dispatchData(const sh_pollHandle_t handle, void* userData);
+        virtual bool check(const sh_pollHandle_t handle, void* userData);
         TAmShPollFired<CAmSamplePlugin> connectFiredCB;
         TAmShPollFired<CAmSamplePlugin> receiveFiredCB;
         TAmShPollDispatch<CAmSamplePlugin> sampleDispatchCB;
         TAmShPollCheck<CAmSamplePlugin> sampleCheckCB;
 
-    private:
+    protected:
         CAmSocketHandler *mSocketHandler;
         sh_pollHandle_t mConnecthandle, mReceiveHandle;
         std::queue<std::string> msgList;
@@ -162,6 +162,54 @@ namespace am
         TAmShTimerCallBack<CAmTimer> pTimerCallback;
     };
 
+    class CAmTimerStressTest: public MockIAmTimerCb
+    {
+        CAmSocketHandler *mpSocketHandler;
+        timespec mUpdateTimeout;
+        int32_t mRepeats;
+        int32_t mId;
+        int32_t mHandle;
+    public:
+        explicit CAmTimerStressTest(CAmSocketHandler *SocketHandler, const timespec &timeout, const int32_t repeats = 0u);
+        virtual ~CAmTimerStressTest();
+        
+        int32_t getId() { return mId; }
+        void setId(const int32_t id) { mId=id; }
+
+        int32_t getHandle() { return mHandle; }
+        void setHandle(const int32_t id) { mHandle=id; }
+        
+        timespec getUpdateTimeout( ) { return mUpdateTimeout; }
+        
+        void timerCallback(sh_timerHandle_t handle, void * userData);
+
+        TAmShTimerCallBack<CAmTimerStressTest> pTimerCallback;
+    };
+    
+    class CAmTimerStressTest2: public MockIAmTimerCb
+    {
+        CAmSocketHandler *mpSocketHandler;
+        timespec mUpdateTimeout;
+        int32_t mRepeats;
+        int32_t mId;
+        int32_t mHandle;
+    public:
+        explicit CAmTimerStressTest2(CAmSocketHandler *SocketHandler, const timespec &timeout, const int32_t repeats = 0u);
+        virtual ~CAmTimerStressTest2();
+        
+        int32_t getId() { return mId; }
+        void setId(const int32_t id) { mId=id; }
+
+        int32_t getHandle() { return mHandle; }
+        void setHandle(const int32_t id) { mHandle=id; }
+        
+        timespec getUpdateTimeout( ) { return mUpdateTimeout; }
+
+        void timerCallback(sh_timerHandle_t handle, void * userData);
+
+        TAmShTimerCallBack<CAmTimerStressTest2> pTimerCallback;
+    };
+    
     class CAmTimerMeasurment: public MockIAmTimerCb
     {
         CAmSocketHandler *mSocketHandler;
@@ -188,6 +236,20 @@ namespace am
         void SetUp();
         void TearDown();
     };
+    
+    class CAmSamplePluginStressTest: public CAmSamplePlugin
+    {
+        std::vector<CAmTimerStressTest2*> mTimers;
+    public:
+        CAmSamplePluginStressTest(CAmSocketHandler *mySocketHandler, sockType_e socketType);
+        ~CAmSamplePluginStressTest();
+        
+        virtual void receiveData(const pollfd pollfd, const sh_pollHandle_t handle, void* userData);
+        virtual bool dispatchData(const sh_pollHandle_t handle, void* userData);
+        virtual bool check(const sh_pollHandle_t handle, void* userData);
+        
+        std::vector<CAmTimerStressTest2*> & getTimers() { return  mTimers; }
+    };
 
 } /* namespace am */
 #endif /* SOCKETHANDLERTEST_H_ */