1 /*
2  * Copyright (c) 2021-2022 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 MODULETEST_OHOS_ABILITY_RUNTIME_MOCK_APP_MGR_CLIENT_H
17 #define MODULETEST_OHOS_ABILITY_RUNTIME_MOCK_APP_MGR_CLIENT_H
18 
19 #include <gmock/gmock.h>
20 #include "app_mgr_client.h"
21 #include "param.h"
22 
23 namespace OHOS {
24 namespace AppExecFwk {
25 class MockAppMgrClient : public AppMgrClient {
26 public:
27     MockAppMgrClient();
28     ~MockAppMgrClient();
29     MOCK_METHOD4(LoadAbility, AppMgrResultCode(const AbilityInfo&, const ApplicationInfo&,
30         const AAFwk::Want&, AbilityRuntime::LoadParam));
31     MOCK_METHOD2(TerminateAbility, AppMgrResultCode(const sptr<IRemoteObject>&, bool));
32     MOCK_METHOD2(UpdateAbilityState, AppMgrResultCode(const sptr<IRemoteObject>& token, const AbilityState state));
33     MOCK_METHOD2(KillApplication, AppMgrResultCode(const std::string&, const bool clearPageStack));
34     MOCK_METHOD1(KillProcessByAbilityToken, AppMgrResultCode(const sptr<IRemoteObject>& token));
35     MOCK_METHOD1(KillProcessesByUserId, AppMgrResultCode(int32_t userId));
36     MOCK_METHOD1(AbilityAttachTimeOut, void(const sptr<IRemoteObject>& token));
37     MOCK_METHOD2(GetRunningProcessInfoByToken, void((const sptr<IRemoteObject>& token,
38         AppExecFwk::RunningProcessInfo& info)));
39     MOCK_METHOD1(GetAllRunningProcesses, AppMgrResultCode(std::vector<RunningProcessInfo>& info));
40     MOCK_METHOD1(GetAllRunningInstanceKeysBySelf, AppMgrResultCode(std::vector<std::string> &instanceKeys));
41     MOCK_METHOD3(GetAllRunningInstanceKeysByBundleName, AppMgrResultCode(const std::string &bundleName,
42         std::vector<std::string> &instanceKeys, int32_t userId));
43 
44     AppMgrResultCode GetProcessRunningInfosByUserId(std::vector<RunningProcessInfo>& info, int32_t userId);
45 
46     AppMgrResultCode AbilityBehaviorAnalysis(const sptr<IRemoteObject>& token, const sptr<IRemoteObject>& preToken,
47         const int32_t visibility, const int32_t perceptibility, const int32_t connectionState) override;
48     AppMgrResultCode ConnectAppMgrService() override;
49     AppMgrResultCode RegisterAppStateCallback(const sptr<IAppStateCallback>& callback) override;
50 };
51 }  // namespace AppExecFwk
52 }  // namespace OHOS
53 #endif  // MODULETEST_OHOS_ABILITY_RUNTIME_MOCK_APP_MGR_CLIENT_H
54