ref: e88bb8886af9b10261b4febc21214b32c75b710b
dir: /test/common/WelsThreadPoolTest.cpp/
#include <gtest/gtest.h> #include <string.h> #include <string> #include <list> #include <map> #include "typedefs.h" #include "WelsThreadLib.h" #include "WelsThreadPool.h" #include "WelsTask.h" #include "WelsThreadPoolTest.h" #define TEST_TASK_NUM 30 class CSimpleTask : public IWelsTask { public: static uint32_t id; CSimpleTask (WelsCommon::IWelsTaskSink* pSink) : IWelsTask (pSink) { m_uiID = id ++; } virtual ~CSimpleTask() { } virtual int32_t Execute() { uint32_t uiSleepTime = (m_uiID > 99) ? 10 : m_uiID; WelsSleep (uiSleepTime); //printf ("Task %d executing\n", m_uiID); return cmResultSuccess; } private: uint32_t m_uiID; }; uint32_t CSimpleTask::id = 0; void* OneCallingFunc() { CThreadPoolTest cThreadPoolTest; CSimpleTask* aTasks[TEST_TASK_NUM]; CWelsThreadPool* pThreadPool = & (CWelsThreadPool::AddReference()); int32_t i; for (i = 0; i < TEST_TASK_NUM; i++) { aTasks[i] = new CSimpleTask (&cThreadPoolTest); } for (i = 0; i < TEST_TASK_NUM; i++) { pThreadPool->QueueTask (aTasks[i]); } while (cThreadPoolTest.GetTaskCount() < TEST_TASK_NUM) { WelsSleep (1); } for (i = 0; i < TEST_TASK_NUM; i++) { delete aTasks[i]; } pThreadPool->RemoveInstance(); return 0; } TEST (CThreadPoolTest, CThreadPoolTest) { OneCallingFunc(); int iRet = CWelsThreadPool::SetThreadNum (8); EXPECT_EQ (0, iRet); EXPECT_FALSE (CWelsThreadPool::IsReferenced()); CWelsThreadPool* pThreadPool = & (CWelsThreadPool::AddReference ()); EXPECT_TRUE(pThreadPool->IsReferenced()); EXPECT_EQ (8, pThreadPool->GetThreadNum()); iRet = CWelsThreadPool::SetThreadNum (4); EXPECT_TRUE (0 != iRet); EXPECT_EQ (8, pThreadPool->GetThreadNum()); pThreadPool->RemoveInstance(); iRet = CWelsThreadPool::SetThreadNum (4); EXPECT_EQ (0, iRet); pThreadPool = & (CWelsThreadPool::AddReference ()); EXPECT_TRUE (pThreadPool->IsReferenced()); EXPECT_EQ (4, pThreadPool->GetThreadNum()); pThreadPool->RemoveInstance(); EXPECT_FALSE (CWelsThreadPool::IsReferenced()); } TEST (CThreadPoolTest, CThreadPoolTestMulti) { int iCallingNum = 10; WELS_THREAD_HANDLE mThreadID[30]; int i = 0; for (i = 0; i < iCallingNum; i++) { WelsThreadCreate (& (mThreadID[i]), (LPWELS_THREAD_ROUTINE)OneCallingFunc, NULL, 0); WelsSleep (1); } for (i = iCallingNum; i < iCallingNum * 2; i++) { WelsThreadCreate (& (mThreadID[i]), (LPWELS_THREAD_ROUTINE)OneCallingFunc, NULL, 0); WelsSleep (1); WelsThreadJoin (mThreadID[i]); } for (i = 0; i < iCallingNum; i++) { WelsThreadJoin (mThreadID[i]); } for (i = iCallingNum * 2; i < iCallingNum * 3; i++) { WelsThreadCreate (& (mThreadID[i]), (LPWELS_THREAD_ROUTINE)OneCallingFunc, NULL, 0); WelsSleep (1); WelsThreadJoin (mThreadID[i]); } EXPECT_FALSE (CWelsThreadPool::IsReferenced()); }