From de90c45ed67ec69fd984f9d75a768a53ad9e6156 Mon Sep 17 00:00:00 2001 From: zxin8218 <15929048406@163.com> Date: Sat, 13 Sep 2025 18:10:10 +0800 Subject: [PATCH] feat: add ut Signed-off-by: zxin8218 <15929048406@163.com> --- .../interface_test/power_mgr_client_test.cpp | 23 +++++++++++-------- 1 file changed, 14 insertions(+), 9 deletions(-) diff --git a/test/unittest/src/interface_test/power_mgr_client_test.cpp b/test/unittest/src/interface_test/power_mgr_client_test.cpp index 61869238..a2b5fb41 100644 --- a/test/unittest/src/interface_test/power_mgr_client_test.cpp +++ b/test/unittest/src/interface_test/power_mgr_client_test.cpp @@ -1461,8 +1461,6 @@ HWTEST_F(PowerMgrClientTest, PowerMgrClient058, TestSize.Level2) powerMgrClient.OverrideScreenOffTime(PowerStateMachine::DEFAULT_SLEEP_TIME_MS); POWER_HILOGI(LABEL_TEST, "PowerMgrClient058 function end!"); } - -#ifdef POWER_MANAGER_TAKEOVER_SUSPEND class TestTakeOverSuspendCallback : public ITakeOverSuspendCallback { public: TestTakeOverSuspendCallback() = default; @@ -1486,9 +1484,13 @@ class TestTakeOverSuspendCallback : public ITakeOverSuspendCallback { HWTEST_F(PowerMgrClientTest, PowerMgrClient059, TestSize.Level0) { POWER_HILOGI(LABEL_TEST, "PowerMgrClient059 function start!"); auto& powerMgrClinet = PowerMgrClient::GetInstance(); - sptr callback = new TestTakeOverSuspendCallback(); + sptr callback = nullptr; TakeOverSuspendPriority priority = TakeOverSuspendPriority::LOW; - powerMgrClinet.RegisterSuspendTakeoverCallback(callback, priority); + bool ret = powerMgrClinet.RegisterSuspendTakeoverCallback(callback, priority); + EXPECT_FALSE(ret); + callback = new TestTakeOverSuspendCallback(); + ret = powerMgrClinet.RegisterSuspendTakeoverCallback(callback, priority); + EXPECT_FALSE(ret); POWER_HILOGI(LABEL_TEST, "PowerMgrClient059 function end!"); } @@ -1500,16 +1502,19 @@ HWTEST_F(PowerMgrClientTest, PowerMgrClient059, TestSize.Level0) { HWTEST_F(PowerMgrClientTest, PowerMgrClient060, TestSize.Level0) { POWER_HILOGI(LABEL_TEST, "PowerMgrClient060 function start!"); auto& powerMgrClinet = PowerMgrClient::GetInstance(); - sptr callback = new TestTakeOverSuspendCallback(); + sptr callback = nullptr; TakeOverSuspendPriority priority = TakeOverSuspendPriority::HIGH; + bool ret = powerMgrClinet.UnRegisterSuspendTakeoverCallback(callback); + EXPECT_FALSE(ret); + callback = new TestTakeOverSuspendCallback(); // first register callback - powerMgrClinet.RegisterSuspendTakeoverCallback(callback, priority); + ret = powerMgrClinet.RegisterSuspendTakeoverCallback(callback, priority); + EXPECT_FALSE(ret); // then test unregister callback - powerMgrClinet.UnRegisterSuspendTakeoverCallback(callback); + ret = powerMgrClinet.UnRegisterSuspendTakeoverCallback(callback); + EXPECT_FALSE(ret); POWER_HILOGI(LABEL_TEST, "PowerMgrClient060 function end!"); } -#endif - /** * @tc.name: PowerMgrClient061 -- Gitee