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 SAMGR_SERVICES_SAMGR_NATIVE_TEST_UNITTEST_INCLUDE_SA_STATUS_CHANGE_MOCK_H
17 #define SAMGR_SERVICES_SAMGR_NATIVE_TEST_UNITTEST_INCLUDE_SA_STATUS_CHANGE_MOCK_H
18 
19 #include "system_ability_status_change_stub.h"
20 #include "system_ability_load_callback_stub.h"
21 
22 namespace OHOS {
23 class SaStatusChangeMock : public SystemAbilityStatusChangeStub {
24 public:
25     void OnAddSystemAbility(int32_t systemAbilityId, const std::string& deviceId) override;
26     void OnRemoveSystemAbility(int32_t systemAbilityId, const std::string& deviceId) override;
27 };
28 
29 class SystemAbilityLoadCallbackMock : public SystemAbilityLoadCallbackStub {
30 public:
31     void OnLoadSystemAbilitySuccess(int32_t systemAbilityId, const sptr<IRemoteObject>& remoteObject) override;
32 
33     void OnLoadSystemAbilityFail(int32_t systemAbilityId) override;
34 
GetSystemAbilityId()35     int32_t GetSystemAbilityId() const
36     {
37         return currSystemAbilityId;
38     }
39 
GetRemoteObject()40     sptr<IRemoteObject> GetRemoteObject() const
41     {
42         return currRemoteObject;
43     }
44 private:
45     int32_t currSystemAbilityId = 0;
46     sptr<IRemoteObject> currRemoteObject;
47 };
48 }
49 #endif /* SAMGR_SERVICES_SAMGR_NATIVE_TEST_UNITTEST_INCLUDE_SA_STATUS_CHANGE_MOCK_H */
50