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_DM_SOFTBUS_CACHE_MOCK_H 16 #define OHOS_DM_SOFTBUS_CACHE_MOCK_H 17 18 #include <string> 19 #include <gmock/gmock.h> 20 21 #include "dm_softbus_cache.h" 22 23 namespace OHOS { 24 namespace DistributedHardware { 25 class DmSoftbusCache { 26 public: 27 virtual ~DmSoftbusCache() = default; 28 public: 29 virtual int32_t GetUdidByUdidHash(const std::string &udidHash, std::string &udid) = 0; 30 virtual int32_t GetUdidFromCache(const char *networkId, std::string &udid) = 0; 31 virtual int32_t GetNetworkIdFromCache(const std::string &udid, std::string &networkId) = 0; 32 virtual int32_t GetLocalDeviceInfo(DmDeviceInfo &nodeInfo) = 0; 33 virtual int32_t GetUuidFromCache(const char *networkId, std::string &uuid) = 0; 34 virtual bool CheckIsOnline(const std::string &deviceId) = 0; 35 virtual int32_t GetDeviceInfoFromCache(std::vector<DmDeviceInfo> &deviceInfoList) = 0; 36 virtual int32_t GetDeviceNameFromCache(const std::string &udid, std::string &deviceName) = 0; 37 public: 38 static inline std::shared_ptr<DmSoftbusCache> dmSoftbusCache = nullptr; 39 }; 40 41 class SoftbusCacheMock : public DmSoftbusCache { 42 public: 43 MOCK_METHOD(int32_t, GetUdidByUdidHash, (const std::string &, std::string &)); 44 MOCK_METHOD(int32_t, GetUdidFromCache, (const char *, std::string &)); 45 MOCK_METHOD(int32_t, GetNetworkIdFromCache, (const std::string &, std::string &)); 46 MOCK_METHOD(int32_t, GetLocalDeviceInfo, (DmDeviceInfo &)); 47 MOCK_METHOD(int32_t, GetUuidFromCache, (const char *, std::string &)); 48 MOCK_METHOD(bool, CheckIsOnline, (const std::string &)); 49 MOCK_METHOD(int32_t, GetDeviceInfoFromCache, (std::vector<DmDeviceInfo> &)); 50 MOCK_METHOD(int32_t, GetDeviceNameFromCache, (const std::string &, std::string &)); 51 }; 52 } 53 } 54 #endif 55