1 /*
2 * Copyright (c) 2025 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 "mock_ability_manager_client.h"
17
18 namespace OHOS {
19 namespace AAFwk {
20 static std::shared_ptr<UIExtensionHostInfo> g_uiExtensionRootHostInfo;
21 static std::shared_ptr<UIExtensionSessionInfo> g_uiExtensionSessionInfo;
22
ClearAll()23 void MockAbilityManagerClient::ClearAll()
24 {
25 g_uiExtensionRootHostInfo = nullptr;
26 g_uiExtensionSessionInfo = nullptr;
27 }
28
SetUIExtensionRootHostInfo(const UIExtensionHostInfo & hostInfo)29 void MockAbilityManagerClient::SetUIExtensionRootHostInfo(const UIExtensionHostInfo& hostInfo)
30 {
31 g_uiExtensionRootHostInfo = std::make_shared<UIExtensionHostInfo>();
32 g_uiExtensionRootHostInfo->elementName_.SetBundleName(hostInfo.elementName_.GetBundleName());
33 }
34
SetUIExtensionSessionInfo(const UIExtensionSessionInfo & sessionInfo)35 void MockAbilityManagerClient::SetUIExtensionSessionInfo(const UIExtensionSessionInfo& sessionInfo)
36 {
37 g_uiExtensionSessionInfo = std::make_shared<UIExtensionSessionInfo>();
38 g_uiExtensionSessionInfo->hostElementName.SetBundleName(sessionInfo.hostElementName.GetBundleName());
39 }
40
GetUIExtensionRootHostInfo(const sptr<IRemoteObject> token,UIExtensionHostInfo & hostInfo,int32_t userId)41 ErrCode AbilityManagerClient::GetUIExtensionRootHostInfo(const sptr<IRemoteObject> token,
42 UIExtensionHostInfo& hostInfo, int32_t userId)
43 {
44 if (g_uiExtensionRootHostInfo) {
45 hostInfo.elementName_.SetBundleName(g_uiExtensionRootHostInfo->elementName_.GetBundleName());
46 }
47 return 0;
48 }
49
GetUIExtensionSessionInfo(const sptr<IRemoteObject> token,UIExtensionSessionInfo & sessionInfo,int32_t userId)50 ErrCode AbilityManagerClient::GetUIExtensionSessionInfo(const sptr<IRemoteObject> token,
51 UIExtensionSessionInfo& sessionInfo, int32_t userId)
52 {
53 if (g_uiExtensionSessionInfo) {
54 sessionInfo.hostElementName.SetBundleName(g_uiExtensionSessionInfo->hostElementName.GetBundleName());
55 }
56 return 0;
57 }
58 }
59 }