From 2e5cacb8af0322c51cdda0c7a163a00105f45163 Mon Sep 17 00:00:00 2001
From: l30054665 <lishang21@huawei.com>
Date: Sat, 2 Dec 2023 18:22:40 +0800
Subject: [PATCH] add UT

Signed-off-by: l30054665 <lishang21@huawei.com>
---
 interfaces/inner_kits/test/unittest/BUILD.gn  |  2 ++
 .../unittest/distributed_input_inner_test.cpp | 20 +++++++++++++++++++
 .../unittest/distributed_input_inner_test.h   |  9 +++++++++
 .../mock/mock_distributed_input_client.cpp    | 11 ++++++++++
 4 files changed, 42 insertions(+)

diff --git a/interfaces/inner_kits/test/unittest/BUILD.gn b/interfaces/inner_kits/test/unittest/BUILD.gn
index c8ca062..35e4125 100644
--- a/interfaces/inner_kits/test/unittest/BUILD.gn
+++ b/interfaces/inner_kits/test/unittest/BUILD.gn
@@ -62,6 +62,8 @@ ohos_unittest("distributed_input_inner_test") {
     "${ipc_path}/src/prepare_d_input_call_back_stub.cpp",
     "${ipc_path}/src/register_d_input_call_back_proxy.cpp",
     "${ipc_path}/src/register_d_input_call_back_stub.cpp",
+    "${ipc_path}/src/register_session_state_callback_proxy.cpp",
+    "${ipc_path}/src/register_session_state_callback_stub.cpp",
     "${ipc_path}/src/sharing_dhid_listener_proxy.cpp",
     "${ipc_path}/src/sharing_dhid_listener_stub.cpp",
     "${ipc_path}/src/simulation_event_listener_proxy.cpp",
diff --git a/interfaces/inner_kits/test/unittest/distributed_input_inner_test.cpp b/interfaces/inner_kits/test/unittest/distributed_input_inner_test.cpp
index 125e04c..5b56f2e 100644
--- a/interfaces/inner_kits/test/unittest/distributed_input_inner_test.cpp
+++ b/interfaces/inner_kits/test/unittest/distributed_input_inner_test.cpp
@@ -113,6 +113,13 @@ int32_t DistributedInputInnerTest::TestSimulationEventListenerStub::OnSimulation
     return DH_SUCCESS;
 }
 
+void DistributedInputInnerTest::TestRegisterSessionStateCb::OnResult(const std::string &devId, const uint32_t status)
+{
+    (void)devId;
+    (void)status;
+    return;
+}
+
 int DistributedInputInnerTest::CheckSourceProxy() const
 {
     OHOS::sptr<OHOS::ISystemAbilityManager> systemAbilityManager =
@@ -486,6 +493,19 @@ HWTEST_F(DistributedInputInnerTest, UnregisterSimulationEventListener02, testing
     int32_t ret = DistributedInputKit::UnregisterSimulationEventListener(listener);
     EXPECT_EQ(DH_SUCCESS, ret);
 }
+
+HWTEST_F(DistributedInputInnerTest, RegisterSessionStateCb01, testing::ext::TestSize.Level0)
+{
+    sptr<TestRegisterSessionStateCb> listener(new TestRegisterSessionStateCb());
+    int32_t ret = DistributedInputKit::RegisterSessionStateCb(listener);
+    EXPECT_EQ(DH_SUCCESS, ret);
+}
+
+HWTEST_F(DistributedInputInnerTest, UnregisterSessionStateCb01, testing::ext::TestSize.Level0)
+{
+    int32_t ret = DistributedInputKit::UnregisterSessionStateCb();
+    EXPECT_EQ(DH_SUCCESS, ret);
+}
 } // namespace DistributedInput
 } // namespace DistributedHardware
 } // namespace OHOS
\ No newline at end of file
diff --git a/interfaces/inner_kits/test/unittest/distributed_input_inner_test.h b/interfaces/inner_kits/test/unittest/distributed_input_inner_test.h
index a9ab92b..2060897 100644
--- a/interfaces/inner_kits/test/unittest/distributed_input_inner_test.h
+++ b/interfaces/inner_kits/test/unittest/distributed_input_inner_test.h
@@ -37,6 +37,7 @@
 #include "stop_d_input_call_back_stub.h"
 #include "simulation_event_listener_stub.h"
 #include "unprepare_d_input_call_back_stub.h"
+#include "register_session_state_callback_stub.h"
 
 namespace OHOS {
 namespace DistributedHardware {
@@ -109,6 +110,14 @@ public:
         virtual ~TestSimulationEventListenerStub() = default;
         int32_t OnSimulationEvent(uint32_t type, uint32_t code, int32_t value);
     };
+    
+    class TestRegisterSessionStateCb : public
+        OHOS::DistributedHardware::DistributedInput::RegisterSessionStateCallbackStub {
+    public:
+        TestRegisterSessionStateCb() = default;
+        virtual ~TestRegisterSessionStateCb() = default;
+        void OnResult(const std::string &devId, const uint32_t status);
+    };
 };
 } // namespace DistributedInput
 } // namespace DistributedHardware
diff --git a/interfaces/inner_kits/test/unittest/mock/mock_distributed_input_client.cpp b/interfaces/inner_kits/test/unittest/mock/mock_distributed_input_client.cpp
index 7a1a039..683375c 100644
--- a/interfaces/inner_kits/test/unittest/mock/mock_distributed_input_client.cpp
+++ b/interfaces/inner_kits/test/unittest/mock/mock_distributed_input_client.cpp
@@ -227,6 +227,17 @@ int32_t DistributedInputClient::UnregisterSimulationEventListener(sptr<ISimulati
 {
     return DH_SUCCESS;
 }
+
+int32_t DistributedInputClient::RegisterSessionStateCb(sptr<ISessionStateCallback> callback)
+{
+    (void)callback;
+    return DH_SUCCESS;
+}
+
+int32_t DistributedInputClient::UnregisterSessionStateCb()
+{
+    return DH_SUCCESS;
+}
 } // namespace DistributedInput
 } // namespace DistributedHardware
 } // namespace OHOS
-- 
Gitee