1 /*
2  * Copyright (c) 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 #ifndef MOCK_THREAD_POOL_H
16 #define MOCK_THREAD_POOL_H
17 #include <gmock/gmock.h>
18 #include "thread_pool_test_stub.h"
19 
20 namespace DistributedDB {
21 class MockThreadPool : public ThreadPoolTestStub {
22 public:
23     MOCK_METHOD1(Execute, TaskId(const Task &));
24     MOCK_METHOD2(Execute, TaskId(const Task &, Duration));
25     MOCK_METHOD2(Schedule, TaskId(const Task &, Duration));
26     MOCK_METHOD3(Schedule, TaskId(const Task &, Duration, Duration));
27     MOCK_METHOD4(Schedule, TaskId(const Task &, Duration, Duration, uint64_t));
28     MOCK_METHOD2(Remove, bool(const TaskId &, bool));
29     MOCK_METHOD2(Reset, TaskId(const TaskId &, Duration));
30 };
31 }
32 #endif // MOCK_THREAD_POOL_H
33