1 /*
2  * Copyright (c) 2022-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 
16 #ifndef OHOS_DM_IMPL_TEST_H
17 #define OHOS_DM_IMPL_TEST_H
18 
19 #include <gtest/gtest.h>
20 #include <refbase.h>
21 
22 #include "device_manager.h"
23 #include "device_manager_callback.h"
24 #include "device_manager_impl.h"
25 #include "mock/mock_ipc_client_proxy.h"
26 
27 namespace OHOS {
28 namespace DistributedHardware {
29 class DeviceManagerImplTest : public testing::Test {
30 public:
31     static void SetUpTestCase();
32     static void TearDownTestCase();
33     void SetUp();
34     void TearDown();
35 
36 private:
37     std::shared_ptr<DiscoveryCallback> test_callback_ = nullptr;
38     std::shared_ptr<PublishCallback> testPublishCallback_ = nullptr;
39 };
40 
41 class DeviceDiscoveryCallback : public DiscoveryCallback {
42 public:
DeviceDiscoveryCallback()43     DeviceDiscoveryCallback() : DiscoveryCallback()
44     {
45     }
~DeviceDiscoveryCallback()46     virtual  ~DeviceDiscoveryCallback()
47     {
48     }
49     void OnDiscoverySuccess(uint16_t subscribeId) override;
50     void OnDiscoveryFailed(uint16_t subscribeId, int32_t failedReason) override;
51     void OnDeviceFound(uint16_t subscribeId, const DmDeviceInfo &deviceInfo) override;
52 };
53 
54 class DevicePublishCallback : public PublishCallback {
55 public:
DevicePublishCallback()56     DevicePublishCallback() : PublishCallback()
57     {
58     }
~DevicePublishCallback()59     virtual ~DevicePublishCallback()
60     {
61     }
62     void OnPublishResult(int32_t publishId, int32_t failedReason) override;
63 };
64 } // namespace DistributedHardware
65 } // namespace OHOS
66 
67 #endif // OHOS_DM_IMPL_TEST_H
68