1 /*
2  * Copyright (c) 2021-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 #ifndef DISRIBUTED_INPUT_SOURCEHANDLER_TEST_H
17 #define DISRIBUTED_INPUT_SOURCEHANDLER_TEST_H
18 
19 #include <functional>
20 #include <iostream>
21 #include <thread>
22 
23 #include <gtest/gtest.h>
24 #include <refbase.h>
25 
26 #include "idistributed_hardware_source.h"
27 #include "system_ability_load_callback_stub.h"
28 
29 #include "constants_dinput.h"
30 #include "distributed_input_kit.h"
31 #include "distributed_input_source_handler.h"
32 #include "i_distributed_source_input.h"
33 #include "register_d_input_call_back_stub.h"
34 #include "unregister_d_input_call_back_stub.h"
35 
36 namespace OHOS {
37 namespace DistributedHardware {
38 namespace DistributedInput {
39 class DistributedInputSourceHandlerTest : public testing::Test {
40 public:
41     static void SetUpTestCase();
42     static void TearDownTestCase();
43     void SetUp() override;
44     void TearDown() override;
45     int32_t CheckSystemAbility();
46 
47     class TestRegisterDInputCallback : public OHOS::DistributedHardware::RegisterCallback {
48     public:
49         TestRegisterDInputCallback() = default;
50         virtual ~TestRegisterDInputCallback() = default;
51         int32_t OnRegisterResult(const std::string &devId, const std::string &dhId, int32_t status,
52             const std::string &data) override;
53     };
54 
55     class TestUnregisterDInputCallback : public OHOS::DistributedHardware::UnregisterCallback {
56     public:
57         TestUnregisterDInputCallback() = default;
58         virtual ~TestUnregisterDInputCallback() = default;
59         int32_t OnUnregisterResult(const std::string &devId, const std::string &dhId, int32_t status,
60         const std::string &data) override;
61     };
62 };
63 } // namespace DistributedInput
64 } // namespace DistributedHardware
65 } // namespace OHOS
66 
67 #endif // DISRIBUTED_INPUT_SOURCEHANDLER_TEST_H
68