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 #include <gtest/gtest.h>
17 #include <gmock/gmock.h>
18 #include "ability_manager_adapter.h"
19
20 #include "account_log_wrapper.h"
21 #include "app_account_authorization_extension_service.h"
22 #include "app_account_authorization_extension_stub.h"
23 #include "mock_app_account_authenticator_stub.h"
24 #include "string_wrapper.h"
25
26 namespace OHOS {
27 namespace AccountSA {
28 namespace {
29 const std::string STRING_BUNDLE = "com.example.name";
30 const std::string STRING_NORMAL_BUNDLENAME = "com.example.normal.bundle";
31 const std::string STRING_ABILITY_NAME_WITH_CONNECT_FAILED = "com.example.MainAbilityWithConnectFailed";
32 const std::string STRING_ABILITY_NAME_WITH_NO_PROXY = "com.example.MainAbilityWithNoProxy";
33 } // namespace
34 using namespace AAFwk;
35
36 class MockAppAccountAuthorizationExtensionService final : public AppAccountAuthorizationExtensionStub {
37 public:
StartAuthorization(const AuthorizationRequest & request)38 ErrCode StartAuthorization(const AuthorizationRequest &request)
39 {
40 std::string testValue = request.parameters.GetStringParam("keyStr");
41 if (testValue.size() != 0) {
42 AAFwk::WantParams errResult;
43 return ERR_JS_SYSTEM_SERVICE_EXCEPTION;
44 }
45 AAFwk::WantParams errResult;
46 AsyncCallbackError businessError;
47 request.callback->OnResult(businessError, errResult);
48 return ERR_OK;
49 }
50 };
51
GetInstance()52 AbilityManagerAdapter *AbilityManagerAdapter::GetInstance()
53 {
54 static AbilityManagerAdapter *instance = new (std::nothrow) AbilityManagerAdapter();
55 return instance;
56 }
57
AbilityManagerAdapter()58 AbilityManagerAdapter::AbilityManagerAdapter()
59 {}
60
~AbilityManagerAdapter()61 AbilityManagerAdapter::~AbilityManagerAdapter()
62 {}
63
ConnectAbility(const AAFwk::Want & want,const sptr<AAFwk::IAbilityConnection> & connect,const sptr<IRemoteObject> & callerToken,int32_t userId)64 ErrCode AbilityManagerAdapter::ConnectAbility(const AAFwk::Want &want, const sptr<AAFwk::IAbilityConnection> &connect,
65 const sptr<IRemoteObject> &callerToken, int32_t userId)
66 {
67 if (want.GetBundle() == STRING_BUNDLE) {
68 sptr<AccountSA::MockAppAccountAuthenticator> mockServicePtr_ =
69 new (std::nothrow) AccountSA::MockAppAccountAuthenticator();
70 int resultCode = ERR_OK;
71 AppExecFwk::ElementName element = want.GetElement();
72 connect->OnAbilityConnectDone(element, mockServicePtr_, resultCode);
73 } else if (want.GetBundle() == STRING_NORMAL_BUNDLENAME) {
74 ACCOUNT_LOGI("mock enter bundleName = %{public}s", want.GetBundle().c_str());
75 int resultCode = ERR_OK;
76 AppExecFwk::ElementName element = want.GetElement();
77 sptr<MockAppAccountAuthorizationExtensionService> authorizationService =
78 new (std::nothrow) MockAppAccountAuthorizationExtensionService();
79 if (authorizationService == nullptr) {
80 return ERR_JS_SYSTEM_SERVICE_EXCEPTION;
81 }
82 ACCOUNT_LOGI("mock enter GetAbilityName = %{public}s", element.GetAbilityName().c_str());
83 if (element.GetAbilityName() == STRING_ABILITY_NAME_WITH_CONNECT_FAILED) {
84 return ERR_JS_SYSTEM_SERVICE_EXCEPTION;
85 }
86 if (element.GetAbilityName() == STRING_ABILITY_NAME_WITH_NO_PROXY) {
87 connect->OnAbilityConnectDone(element, nullptr, resultCode);
88 return ERR_OK;
89 }
90 connect->OnAbilityConnectDone(element, authorizationService, resultCode);
91 } else {
92 int resultCode = ERR_OK;
93 AppExecFwk::ElementName element = want.GetElement();
94 connect->OnAbilityConnectDone(element, nullptr, resultCode);
95 }
96 return ERR_OK;
97 }
98
DisconnectAbility(const sptr<AAFwk::IAbilityConnection> & connect)99 ErrCode AbilityManagerAdapter::DisconnectAbility(const sptr<AAFwk::IAbilityConnection> &connect)
100 {
101 return ERR_OK;
102 }
103
StartUser(int accountId,const sptr<AAFwk::IUserCallback> & callback)104 ErrCode AbilityManagerAdapter::StartUser(int accountId, const sptr<AAFwk::IUserCallback> &callback)
105 {
106 return ERR_OK;
107 }
108
StopUser(int accountId,const sptr<AAFwk::IUserCallback> & callback)109 ErrCode AbilityManagerAdapter::StopUser(int accountId, const sptr<AAFwk::IUserCallback> &callback)
110 {
111 return ERR_OK;
112 }
113
LogoutUser(int accountId)114 ErrCode AbilityManagerAdapter::LogoutUser(int accountId)
115 {
116 return ERR_OK;
117 }
118
Connect()119 void Connect()
120 {
121 }
122
DoConnectAbility(const sptr<IRemoteObject> proxy,const Want & want,const sptr<IAbilityConnection> & connect,const sptr<IRemoteObject> & callerToken,int32_t userId=-1)123 ErrCode DoConnectAbility(
124 const sptr<IRemoteObject> proxy,
125 const Want &want,
126 const sptr<IAbilityConnection> &connect,
127 const sptr<IRemoteObject> &callerToken,
128 int32_t userId = -1)
129 {
130 return ERR_OK;
131 }
132
GetAbilityManager()133 sptr<IRemoteObject> GetAbilityManager()
134 {
135 sptr<IRemoteObject> iRemoteObject;
136 return iRemoteObject;
137 }
138
ResetProxy(const wptr<IRemoteObject> & remote)139 void ResetProxy(const wptr<IRemoteObject>& remote)
140 {}
141 } // namespace AccountSA
142 } // namespace OHOS