[Common] Add sink to IWelsTask
Showing the changes to test/common/WelsThreadPoolTest.cpp (1 of 10 files). Show all
Unified · Split
| @@ -16,7 +16,7 @@ class CSimpleTask : public IWelsTask { · 15 unmodified lines | ||
| 16 | 16 | public: |
| 17 | 17 | static uint32_t id; |
| 18 | 18 | |
| 19 | − CSimpleTask() { | |
| 19 | + CSimpleTask (WelsCommon::IWelsTaskSink* pSink) : IWelsTask (pSink) { | |
| 20 | 20 | m_uiID = id ++; |
| 21 | 21 | } |
| 22 | 22 | |
| @@ -38,18 +38,25 @@ uint32_t CSimpleTask::id = 0; · 15 unmodified lines | ||
| 38 | 38 | |
| 39 | 39 | |
| 40 | 40 | TEST (CThreadPoolTest, CThreadPoolTest) { |
| 41 | − CSimpleTask tasks[TEST_TASK_NUM]; | |
| 42 | 41 | CThreadPoolTest cThreadPoolTest; |
| 42 | + CSimpleTask* aTasks[TEST_TASK_NUM]; | |
| 43 | 43 | CWelsThreadPool cThreadPool (&cThreadPoolTest); |
| 44 | 44 | |
| 45 | 45 | int32_t i; |
| 46 | + for (i = 0; i < TEST_TASK_NUM; i++) { | |
| 47 | + aTasks[i] = new CSimpleTask (&cThreadPoolTest); | |
| 48 | + } | |
| 46 | 49 | |
| 47 | 50 | for (i = 0; i < TEST_TASK_NUM; i++) { |
| 48 | − cThreadPool.QueueTask (&tasks[i]); | |
| 51 | + cThreadPool.QueueTask (aTasks[i]); | |
| 49 | 52 | } |
| 50 | 53 | |
| 51 | 54 | while (cThreadPoolTest.GetTaskCount() < TEST_TASK_NUM) { |
| 52 | 55 | WelsSleep (1); |
| 53 | 56 | } |
| 57 | + | |
| 58 | + for (i = 0; i < TEST_TASK_NUM; i++) { | |
| 59 | + delete aTasks[i]; | |
| 60 | + } | |
| 54 | 61 | } |
| 55 | 62 | |