From ec9899e072faaac585bc6e936120d39b6aac3f9b Mon Sep 17 00:00:00 2001
From: z30078586 <zhaoqichao5@h-partners.com>
Date: Fri, 22 Aug 2025 10:06:34 +0800
Subject: [PATCH] =?UTF-8?q?=E6=8F=90=E5=8D=87UT=E8=A6=86=E7=9B=96=E7=8E=87?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

Signed-off-by: z30078586 <zhaoqichao5@h-partners.com>
---
 .../src/driver_state_manager_test.cpp         | 12 +++++
 .../src/event_listener_manager_test.cpp       | 44 ++++++++++++++++++-
 2 files changed, 55 insertions(+), 1 deletion(-)

diff --git a/test/unittest/services/src/driver_state_manager_test.cpp b/test/unittest/services/src/driver_state_manager_test.cpp
index 23ed3e068..1e95158d3 100644
--- a/test/unittest/services/src/driver_state_manager_test.cpp
+++ b/test/unittest/services/src/driver_state_manager_test.cpp
@@ -68,6 +68,18 @@ HWTEST_F(DriverStateManagerTest, OnDriverStopTest, TestSize.Level0)
     EXPECT_NO_THROW(DriverStateManager::GetInstance().OnDriverStart());
     EXPECT_NO_THROW(DriverStateManager::GetInstance().OnDriverStart());
 }
+
+HWTEST_F(DriverStateManagerTest, RegisterDriverStartCallbackTest, TestSize.Level0)
+{
+    EXPECT_NO_THROW(DriverStateManager::GetInstance().RegisterDriverStartCallback(nullptr));
+    EXPECT_NO_THROW(DriverStateManager::GetInstance().RegisterDriverStartCallback([]() { return; }));
+}
+
+HWTEST_F(DriverStateManagerTest, RegisterDriverStopCallbackTest, TestSize.Level0)
+{
+    EXPECT_NO_THROW(DriverStateManager::GetInstance().RegisterDriverStopCallback(nullptr));
+    EXPECT_NO_THROW(DriverStateManager::GetInstance().RegisterDriverStopCallback([]() { return; }));
+}
 } // namespace UserAuth
 } // namespace UserIam
 } // namespace OHOS
diff --git a/test/unittest/services/src/event_listener_manager_test.cpp b/test/unittest/services/src/event_listener_manager_test.cpp
index 3ad2f658e..2a191731b 100644
--- a/test/unittest/services/src/event_listener_manager_test.cpp
+++ b/test/unittest/services/src/event_listener_manager_test.cpp
@@ -18,6 +18,7 @@
 
 #include "gtest/gtest.h"
 #include "mock_event_listener.h"
+#include "mock_remote_object.h"
 
 namespace OHOS {
 namespace UserIam {
@@ -62,7 +63,7 @@ HWTEST_F(EventListenerManagerTest, EventListenerManagerTestUnRegistEventListener
     EXPECT_NO_THROW(CredChangeEventListenerManager::GetInstance().UnRegistEventListener(testCallback));
 }
 
-HWTEST_F(EventListenerManagerTest, EventListenerManagerTestRemoveDeathRecipient, TestSize.Level0)
+HWTEST_F(EventListenerManagerTest, EventListenerManagerTestRemoveDeathRecipient_001, TestSize.Level0)
 {
     sptr<IEventListenerCallback> testCallback = new MockEventListener();
     EXPECT_NO_THROW(AuthEventListenerManager::GetInstance().RemoveDeathRecipient(testCallback));
@@ -70,6 +71,47 @@ HWTEST_F(EventListenerManagerTest, EventListenerManagerTestRemoveDeathRecipient,
     EXPECT_NO_THROW(CredChangeEventListenerManager::GetInstance().RemoveDeathRecipient(testCallback));
     EXPECT_NO_THROW(CredChangeEventListenerManager::GetInstance().RemoveDeathRecipient(nullptr));
 }
+
+HWTEST_F(EventListenerManagerTest, EventListenerManagerTestRemoveDeathRecipient_002, TestSize.Level0)
+{
+    auto mockCallbackRemove = new MockEventListener();
+    sptr<MockRemoteObject> obj(new (std::nothrow) MockRemoteObject);
+    EXPECT_CALL(*mockCallbackRemove, AsObject())
+        .WillOnce(Return(obj))
+        .WillRepeatedly(Return(obj));
+    
+    AuthEventListenerManager& authEventListenerManager = AuthEventListenerManager::GetInstance();
+    EXPECT_EQ(authEventListenerManager.RemoveDeathRecipient(mockCallbackRemove), SUCCESS);
+}
+
+HWTEST_F(EventListenerManagerTest, EventListenerManagerTestAddDeathRecipient_001, TestSize.Level0)
+{
+    auto mockCallbackAdd = new MockEventListener();
+    sptr<MockRemoteObject> obj(new (std::nothrow) MockRemoteObject);
+    EXPECT_CALL(*mockCallbackAdd, AsObject())
+        .WillOnce(Return(obj))
+        .WillRepeatedly(Return(obj));
+
+     EXPECT_CALL(*obj, AddDeathRecipient(_))
+        .WillOnce(Return(false));
+    AuthEventListenerManager& authEventListenerManager = AuthEventListenerManager::GetInstance();
+    EXPECT_EQ(authEventListenerManager.AddDeathRecipient(&authEventListenerManager, mockCallbackAdd), GENERAL_ERROR);
+}
+
+HWTEST_F(EventListenerManagerTest, EventListenerManagerTestAddDeathRecipient_002, TestSize.Level0)
+{
+   auto mockCallbackAdd = new MockEventListener();
+    sptr<MockRemoteObject> obj(new (std::nothrow) MockRemoteObject);
+    EXPECT_CALL(*mockCallbackAdd, AsObject())
+        .WillOnce(Return(obj))
+        .WillRepeatedly(Return(obj));
+
+     EXPECT_CALL(*obj, AddDeathRecipient(_))
+        .WillOnce(Return(true));
+
+    AuthEventListenerManager& authEventListenerManager = AuthEventListenerManager::GetInstance();
+    EXPECT_EQ(authEventListenerManager.AddDeathRecipient(&authEventListenerManager, mockCallbackAdd), SUCCESS);
+}
 } // namespace UserAuth
 } // namespace UserIam
 } // namespace OHOS
\ No newline at end of file
-- 
Gitee