1 /* 2 * Copyright (C) 2024 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 OHOS_DEVICEPROFILE_CONNECTOR_MOCK_H 16 #define OHOS_DEVICEPROFILE_CONNECTOR_MOCK_H 17 18 #include <string> 19 #include <gmock/gmock.h> 20 21 #include "deviceprofile_connector.h" 22 #include "dm_device_info.h" 23 24 namespace OHOS { 25 namespace DistributedHardware { 26 class DmDeviceProfileConnector { 27 public: 28 virtual ~DmDeviceProfileConnector() = default; 29 public: 30 virtual std::vector<DistributedDeviceProfile::AccessControlProfile> GetAllAccessControlProfile() = 0; 31 virtual int32_t HandleDevUnBindEvent(int32_t remoteUserId, const std::string &remoteUdid, 32 const std::string &localUdid) = 0; 33 virtual int32_t HandleAccountLogoutEvent(int32_t remoteUserId, const std::string &remoteAccountHash, 34 const std::string &remoteUdid, const std::string &localUdid) = 0; 35 virtual uint32_t CheckBindType(std::string trustDeviceId, std::string requestDeviceId) = 0; 36 virtual std::map<int32_t, int32_t> GetUserIdAndBindLevel(const std::string &localUdid, 37 const std::string &peerUdid) = 0; 38 virtual bool DeleteAclForAccountLogOut(const std::string &localUdid, int32_t localUserId, 39 const std::string &peerUdid, int32_t peerUserId) = 0; 40 virtual OHOS::DistributedHardware::ProcessInfo HandleAppUnBindEvent(int32_t remoteUserId, 41 const std::string &remoteUdid, int32_t tokenId, const std::string &localUdid) = 0; 42 virtual std::multimap<std::string, int32_t> GetDevIdAndUserIdByActHash(const std::string &localUdid, 43 const std::string &peerUdid, int32_t peerUserId, const std::string &peerAccountHash) = 0; 44 virtual std::vector<int32_t> SyncAclByBindType(std::string pkgName, std::vector<int32_t> bindTypeVec, 45 std::string localDeviceId, std::string targetDeviceId) = 0; 46 virtual std::vector<int32_t> GetBindTypeByPkgName(std::string pkgName, std::string requestDeviceId, 47 std::string trustUdid) = 0; 48 virtual DmOfflineParam DeleteAccessControlList(const std::string &pkgName, const std::string &localDeviceId, 49 const std::string &remoteDeviceId, int32_t bindLevel, const std::string &extra) = 0; 50 virtual void DeleteAclForRemoteUserRemoved(std::string peerUdid, int32_t peerUserId, 51 std::vector<int32_t> &userIds) = 0; 52 virtual OHOS::DistributedHardware::ProcessInfo HandleAppUnBindEvent(int32_t remoteUserId, 53 const std::string &remoteUdid, int32_t tokenId, const std::string &localUdid, int32_t peerTokenId) = 0; 54 virtual std::unordered_map<std::string, DmAuthForm> GetAppTrustDeviceList(const std::string &pkgName, 55 const std::string &deviceId) = 0; 56 virtual int32_t CheckDeviceInfoPermission(const std::string &localUdid, const std::string &peerDeviceId) = 0; 57 virtual int32_t GetLocalServiceInfoByBundleNameAndPinExchangeType(const std::string& bundleName, 58 int32_t pinExchangeType, DistributedDeviceProfile::LocalServiceInfo &serviceInfo) = 0; 59 virtual int32_t PutSessionKey(const std::vector<unsigned char> &sessionKeyArray, int32_t &sessionKeyId) = 0; 60 virtual int32_t PutLocalServiceInfo(const DistributedDeviceProfile::LocalServiceInfo &localServiceInfo) = 0; 61 virtual int32_t DeleteLocalServiceInfo(const std::string &bundleName, int32_t pinExchangeType) = 0; 62 virtual int32_t UpdateLocalServiceInfo(const DistributedDeviceProfile::LocalServiceInfo &localServiceInfo) = 0; 63 virtual bool CheckAclStatusAndForegroundNotMatch(const std::string &localUdid, 64 const std::vector<int32_t> &foregroundUserIds, const std::vector<int32_t> &backgroundUserIds) = 0; 65 virtual std::map<std::string, int32_t> GetDeviceIdAndBindLevel(std::vector<int32_t> userIds, 66 const std::string &localUdid) = 0; 67 public: 68 static inline std::shared_ptr<DmDeviceProfileConnector> dmDeviceProfileConnector = nullptr; 69 }; 70 71 class DeviceProfileConnectorMock : public DmDeviceProfileConnector { 72 public: 73 MOCK_METHOD(std::vector<DistributedDeviceProfile::AccessControlProfile>, GetAllAccessControlProfile, ()); 74 MOCK_METHOD(int32_t, HandleDevUnBindEvent, (int32_t, const std::string &, const std::string &)); 75 MOCK_METHOD(int32_t, HandleAccountLogoutEvent, (int32_t, const std::string &, const std::string &, 76 const std::string &)); 77 MOCK_METHOD(uint32_t, CheckBindType, (std::string, std::string)); 78 MOCK_METHOD((std::map<int32_t, int32_t>), GetUserIdAndBindLevel, (const std::string &, const std::string &)); 79 MOCK_METHOD(bool, DeleteAclForAccountLogOut, (const std::string &, int32_t, const std::string &, int32_t)); 80 MOCK_METHOD((OHOS::DistributedHardware::ProcessInfo), HandleAppUnBindEvent, (int32_t, const std::string &, 81 int32_t, const std::string &)); 82 MOCK_METHOD((std::multimap<std::string, int32_t>), GetDevIdAndUserIdByActHash, (const std::string &, 83 const std::string &, int32_t, const std::string &)); 84 MOCK_METHOD((std::vector<int32_t>), SyncAclByBindType, (std::string, std::vector<int32_t>, std::string, 85 std::string)); 86 MOCK_METHOD((std::vector<int32_t>), GetBindTypeByPkgName, (std::string, std::string, std::string)); 87 MOCK_METHOD(DmOfflineParam, DeleteAccessControlList, (const std::string &, const std::string &, const std::string &, 88 int32_t, const std::string &)); 89 MOCK_METHOD(void, DeleteAclForRemoteUserRemoved, (std::string, int32_t, std::vector<int32_t> &)); 90 MOCK_METHOD((OHOS::DistributedHardware::ProcessInfo), HandleAppUnBindEvent, (int32_t, const std::string &, 91 int32_t, const std::string &, int32_t)); 92 MOCK_METHOD((std::unordered_map<std::string, DmAuthForm>), GetAppTrustDeviceList, 93 (const std::string &, const std::string &)); 94 MOCK_METHOD(int32_t, CheckDeviceInfoPermission, (const std::string &, const std::string &)); 95 MOCK_METHOD(int32_t, GetLocalServiceInfoByBundleNameAndPinExchangeType, (const std::string&, int32_t, 96 DistributedDeviceProfile::LocalServiceInfo &serviceInfo)); 97 MOCK_METHOD(int32_t, PutSessionKey, (const std::vector<unsigned char> &, int32_t&)); 98 MOCK_METHOD(int32_t, PutLocalServiceInfo, (const DistributedDeviceProfile::LocalServiceInfo &)); 99 MOCK_METHOD(int32_t, DeleteLocalServiceInfo, (const std::string &, int32_t)); 100 MOCK_METHOD(int32_t, UpdateLocalServiceInfo, (const DistributedDeviceProfile::LocalServiceInfo &)); 101 MOCK_METHOD(bool, CheckAclStatusAndForegroundNotMatch, 102 (const std::string &, (const std::vector<int32_t> &), (const std::vector<int32_t> &))); 103 MOCK_METHOD((std::map<std::string, int32_t>), GetDeviceIdAndBindLevel, 104 ((std::vector<int32_t>), const std::string &)); 105 }; 106 } 107 } 108 #endif 109