1 /* 2 * Copyright (c) 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 DISC_MANAGER_MOCK_H 17 #define DISC_MANAGER_MOCK_H 18 19 #include <gmock/gmock.h> 20 21 #include "disc_manager.h" 22 #include "lnn_ohos_account_adapter.h" 23 24 namespace OHOS { 25 class DiscManagerInterface { 26 public: DiscManagerInterface()27 DiscManagerInterface() {}; ~DiscManagerInterface()28 virtual ~DiscManagerInterface() {}; 29 30 virtual int32_t DiscPublishService(const char *packageName, const PublishInfo *info, int32_t callingPid) = 0; 31 virtual int32_t DiscStartScan(DiscModule moduleId, const PublishInfo *info, int32_t callingPid) = 0; 32 virtual int32_t DiscUnPublishService(const char *packageName, int32_t publishId, int32_t callingPid) = 0; 33 virtual int32_t DiscUnpublish(DiscModule moduleId, int32_t publishId, int32_t callingPid) = 0; 34 virtual int32_t DiscStartDiscovery( 35 const char *packageName, const SubscribeInfo *info, const IServerDiscInnerCallback *cb, int32_t callingPid) = 0; 36 virtual int32_t DiscSetDiscoverCallback(DiscModule moduleId, const DiscInnerCallback *callback) = 0; 37 virtual int32_t DiscStartAdvertise(DiscModule moduleId, const SubscribeInfo *info, int32_t callingPid) = 0; 38 virtual int32_t DiscStopDiscovery(const char *packageName, int32_t subscribeId, int32_t callingPid) = 0; 39 virtual int32_t DiscStopAdvertise(DiscModule moduleId, int32_t subscribeId, int32_t callingPid) = 0; 40 }; 41 class DiscManagerInterfaceMock : public DiscManagerInterface { 42 public: 43 DiscManagerInterfaceMock(); 44 ~DiscManagerInterfaceMock() override; 45 MOCK_METHOD3(DiscPublishService, int32_t(const char *, const PublishInfo *, int32_t)); 46 MOCK_METHOD3(DiscStartScan, int32_t(DiscModule, const PublishInfo *, int32_t)); 47 MOCK_METHOD3(DiscUnPublishService, int32_t(const char *, int32_t, int32_t)); 48 MOCK_METHOD3(DiscUnpublish, int32_t(DiscModule, int32_t, int32_t)); 49 MOCK_METHOD4(DiscStartDiscovery, 50 int32_t(const char *, const SubscribeInfo *, const IServerDiscInnerCallback *, int32_t)); 51 MOCK_METHOD2(DiscSetDiscoverCallback, int32_t(DiscModule, const DiscInnerCallback *)); 52 MOCK_METHOD3(DiscStartAdvertise, int32_t(DiscModule, const SubscribeInfo *, int32_t)); 53 MOCK_METHOD3(DiscStopDiscovery, int32_t(const char *, int32_t, int32_t)); 54 MOCK_METHOD3(DiscStopAdvertise, int32_t(DiscModule, int32_t, int32_t)); 55 }; 56 } // namespace OHOS 57 #endif // DISC_MANAGER_MOCK_H 58