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 618692388a1da947e48055617bdc5c90d8d01fd6..a2b5fb410e0f9fd0e0a488f3a6b706ec2fb24c12 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