From f383c007eb47fc16ffc35829b4fe09247ebefb31 Mon Sep 17 00:00:00 2001 From: yuwenze Date: Wed, 11 Jun 2025 20:14:27 +0800 Subject: [PATCH] add ut Signed-off-by: yuwenze Change-Id: Ia41583ef4bace878953690d70a645508be6d065f --- .../ability_context_impl_test.cpp | 47 +++++++++++++++++++ 1 file changed, 47 insertions(+) diff --git a/test/unittest/frameworks_kits_ability_ability_runtime_test/ability_context_impl_test.cpp b/test/unittest/frameworks_kits_ability_ability_runtime_test/ability_context_impl_test.cpp index f4e16d11a8f..105a6ff8e48 100644 --- a/test/unittest/frameworks_kits_ability_ability_runtime_test/ability_context_impl_test.cpp +++ b/test/unittest/frameworks_kits_ability_ability_runtime_test/ability_context_impl_test.cpp @@ -31,6 +31,7 @@ #include "mock_lifecycle_observer.h" #include "mock_serviceability_manager_service.h" #include "scene_board_judgement.h" +#include "session/host/include/session.h" #include "sys_mgr_client.h" #include "system_ability_definition.h" @@ -2205,6 +2206,52 @@ HWTEST_F(AbilityContextImplTest, RevokeDelegator_0600, Function | MediumTest | L } } +/** + * @tc.number: Ability_Context_Impl_RevokeDelegator_0700 + * @tc.name: RevokeDelegator + * @tc.desc: Verify that function RevokeDelegator. + */ +HWTEST_F(AbilityContextImplTest, RevokeDelegator_0700, Function | MediumTest | Level1) +{ + auto context = std::make_unique(); + context->isHook_ = true; + context->hookOff_ = false; + AAFwk::AbilityManagerClient::GetInstance()->proxy_ = g_mockAbilityMs; + Rosen::SessionInfo info; + sptr session = new Rosen::Session(info); + context->SetWeakSessionToken(session->AsObject()); + auto result = context->RevokeDelegator(); + if (!Rosen::SceneBoardJudgement::IsSceneBoardEnabled()) { + EXPECT_EQ(result, AAFwk::ERR_CAPABILITY_NOT_SUPPORT); + } else { + EXPECT_EQ(result, ERR_INVALID_VALUE); + } +} + +/** + * @tc.number: Ability_Context_Impl_RevokeDelegator_0800 + * @tc.name: RevokeDelegator + * @tc.desc: Verify that function RevokeDelegator. + */ +HWTEST_F(AbilityContextImplTest, RevokeDelegator_0800, Function | MediumTest | Level1) +{ + auto context = std::make_unique(); + context->isHook_ = true; + context->hookOff_ = false; + AAFwk::AbilityManagerClient::GetInstance()->proxy_ = g_mockAbilityMs; + Rosen::SessionInfo info; + sptr session = new Rosen::Session(info); + context->SetWeakSessionToken(session->AsObject()); + std::shared_ptr abilityCallback = std::make_shared(); + context->RegisterAbilityCallback(abilityCallback); + auto result = context->RevokeDelegator(); + if (!Rosen::SceneBoardJudgement::IsSceneBoardEnabled()) { + EXPECT_EQ(result, AAFwk::ERR_CAPABILITY_NOT_SUPPORT); + } else { + EXPECT_EQ(result, ERR_OK); + } +} + /** * @tc.number: Ability_Context_Impl_AddCompletionHandler_0100 * @tc.name: AddCompletionHandler -- Gitee