1 /* 2 * Copyright (c) 2022-2023 Huawei Device Co., Ltd. 3 * Licensed under the Apache License, Version 2.0 (the "License"); 4 * you may not use this file except in compliance with the License. 5 * You may obtain a copy of the License at 6 * 7 * http://www.apache.org/licenses/LICENSE-2.0 8 * 9 * Unless required by applicable law or agreed to in writing, software 10 * distributed under the License is distributed on an "AS IS" BASIS, 11 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 12 * See the License for the specific language governing permissions and 13 * limitations under the License. 14 */ 15 16 #ifndef FOUNDATION_ACE_FRAMEWORKS_CORE_PIPELINE_NG_TEST_MOCK_MOCK_TASK_EXECUTOR_H 17 #define FOUNDATION_ACE_FRAMEWORKS_CORE_PIPELINE_NG_TEST_MOCK_MOCK_TASK_EXECUTOR_H 18 19 #include "gmock/gmock.h" 20 21 #include "base/thread/task_executor.h" 22 #include "base/utils/utils.h" 23 24 namespace OHOS::Ace { 25 class MockTaskExecutor : public TaskExecutor { 26 public: 27 MOCK_METHOD(void, AddTaskObserver, (Task && callback), (override)); 28 MOCK_METHOD(void, RemoveTaskObserver, (), (override)); 29 MOCK_METHOD(Task, WrapTaskWithTraceId, (Task && task, int32_t id), (const override)); 30 OnPostTask(Task && task,TaskType type,uint32_t delayTime)31 bool OnPostTask(Task&& task, TaskType type, uint32_t delayTime) const override 32 { 33 CHECK_NULL_RETURN(task, false); 34 task(); 35 return true; 36 } 37 WillRunOnCurrentThread(TaskType type)38 bool WillRunOnCurrentThread(TaskType type) const override 39 { 40 return true; 41 } 42 }; 43 } // namespace OHOS::Ace 44 #endif // FOUNDATION_ACE_FRAMEWORKS_CORE_PIPELINE_NG_TEST_MOCK_MOCK_TASK_EXECUTOR_H 45