diff --git a/frameworks/wantagent/test/unittest/pending_want_test/BUILD.gn b/frameworks/wantagent/test/unittest/pending_want_test/BUILD.gn
index 39561893289f55d9dc85c5df35e90c78a1caba58..80252d5fd32de52f48041e160953f428229e36a0 100644
--- a/frameworks/wantagent/test/unittest/pending_want_test/BUILD.gn
+++ b/frameworks/wantagent/test/unittest/pending_want_test/BUILD.gn
@@ -20,6 +20,7 @@ module_output_path = "ans_standard/wantagent"
 ohos_unittest("pending_want_test") {
   module_out_path = module_output_path
   include_dirs = [
+    "//aafwk/standard/frameworks/kits/appkit/native/ability_runtime/context/",
     "//foundation/aafwk/standard/frameworks/kits/appkit/native/ability_runtime",
     "//foundation/aafwk/standard/services/abilitymgr/include",
     "//foundation/distributedschedule/samgr/adapter/interfaces/innerkits/include/",
diff --git a/frameworks/wantagent/test/unittest/pending_want_test/pending_want_test.cpp b/frameworks/wantagent/test/unittest/pending_want_test/pending_want_test.cpp
index 4f66a28dcc39fa3bd42e63c5bab3c49a1111cfe2..bee296c8eca4d0ea934580a28ae88dfddceeec99 100644
--- a/frameworks/wantagent/test/unittest/pending_want_test/pending_want_test.cpp
+++ b/frameworks/wantagent/test/unittest/pending_want_test/pending_want_test.cpp
@@ -20,6 +20,7 @@
 #include "completed_dispatcher.h"
 #include "context/context.h"
 #include "context_container.h"
+#include "context_impl.h"
 #include "element_name.h"
 #include "event_handler.h"
 #include "iservice_registry.h"
@@ -49,6 +50,7 @@
 #include "ohos/aafwk/base/short_wrapper.h"
 #include "ohos/aafwk/base/string_wrapper.h"
 
+
 using namespace testing::ext;
 using namespace OHOS::AAFwk;
 using namespace OHOS;
@@ -102,7 +104,8 @@ Want PendingWantTest::MakeWant(std::string deviceId, std::string abilityName, st
 
 std::shared_ptr<OHOS::AbilityRuntime::Context> GetAppContext()
 {
-    return OHOS::AbilityRuntime::Context::GetApplicationContext();
+    std::shared_ptr<AbilityRuntime::Context> context = std::make_shared<AbilityRuntime::ContextImpl>();
+    return context;
 }
 
 void PendingWantTest::SetUpTestCase(void)
@@ -170,26 +173,10 @@ HWTEST_F(PendingWantTest, PendingWant_0400, Function | MediumTest | Level1)
     want->SetElement(element);
     unsigned int flags = 1;
     flags |= FLAG_NO_CREATE;
-    std::shared_ptr<PendingWant> pendingWant = PendingWant::GetAbility(GetAppContext(), requestCode, want, flags);
+    std::shared_ptr<PendingWant> pendingWant = PendingWant::GetAbility(nullptr, requestCode, want, flags);
     EXPECT_EQ(pendingWant, nullptr);
 }
 
-/*
- * @tc.number    : PendingWant_0500
- * @tc.name      : PendingWant GetAbility
- * @tc.desc      : 1.Get pendingWant (START_ABILITY)
- */
-HWTEST_F(PendingWantTest, PendingWant_0500, Function | MediumTest | Level1)
-{
-    int requestCode = 10;
-    std::shared_ptr<Want> want = std::make_shared<Want>();
-    ElementName element("device", "bundleName", "abilityName");
-    want->SetElement(element);
-    unsigned int flags = 1;
-    flags |= FLAG_NO_CREATE;
-    std::shared_ptr<PendingWant> pendingWant = PendingWant::GetAbility(GetAppContext(), requestCode, want, flags);
-    EXPECT_NE(pendingWant, nullptr);
-}
 
 /*
  * @tc.number    : PendingWant_0600
diff --git a/frameworks/wantagent/test/unittest/want_agent_helper_test/BUILD.gn b/frameworks/wantagent/test/unittest/want_agent_helper_test/BUILD.gn
index 12ec08065e3c538ddbcb370835c2161363c75910..926b3dcf1baaa87f75af1fea2fe2730a47543cac 100644
--- a/frameworks/wantagent/test/unittest/want_agent_helper_test/BUILD.gn
+++ b/frameworks/wantagent/test/unittest/want_agent_helper_test/BUILD.gn
@@ -20,6 +20,7 @@ module_output_path = "ans_standard/wantagent"
 ohos_unittest("want_agent_helper_test") {
   module_out_path = module_output_path
   include_dirs = [
+    "//aafwk/standard/frameworks/kits/appkit/native/ability_runtime/context/",
     "//foundation/aafwk/standard/frameworks/kits/appkit/native/ability_runtime",
     "//foundation/aafwk/standard/services/abilitymgr/include",
   ]
diff --git a/frameworks/wantagent/test/unittest/want_agent_helper_test/want_agent_helper_test.cpp b/frameworks/wantagent/test/unittest/want_agent_helper_test/want_agent_helper_test.cpp
index f555062ad2331e5fbdb0fca08149d222827aa6ca..b92fb2d2b23eafd51c33f293576c1d38fef37974 100644
--- a/frameworks/wantagent/test/unittest/want_agent_helper_test/want_agent_helper_test.cpp
+++ b/frameworks/wantagent/test/unittest/want_agent_helper_test/want_agent_helper_test.cpp
@@ -35,6 +35,7 @@
 #include "want_receiver_stub.h"
 #include "want_sender_stub.h"
 #include "ohos/aafwk/base/bool_wrapper.h"
+#include "context_impl.h"
 
 using namespace testing::ext;
 using namespace OHOS::AAFwk;
@@ -249,7 +250,7 @@ HWTEST_F(WantAgentHelperTest, WantAgentHelper_1100, Function | MediumTest | Leve
 HWTEST_F(WantAgentHelperTest, WantAgentHelper_1200, Function | MediumTest | Level1)
 {
     std::shared_ptr<WantAgentHelper> wantAgentHelper = std::make_shared<WantAgentHelper>();
-    std::shared_ptr<AbilityRuntime::Context> context = OHOS::AbilityRuntime::Context::GetApplicationContext();
+    std::shared_ptr<AbilityRuntime::Context> context = std::make_shared<AbilityRuntime::ContextImpl>();
     std::shared_ptr<Want> want = std::make_shared<Want>();
     ElementName element("device", "bundleName", "abilityName");
     want->SetElement(element);
@@ -273,7 +274,7 @@ HWTEST_F(WantAgentHelperTest, WantAgentHelper_1200, Function | MediumTest | Leve
 HWTEST_F(WantAgentHelperTest, WantAgentHelper_1300, Function | MediumTest | Level1)
 {
     std::shared_ptr<WantAgentHelper> wantAgentHelper = std::make_shared<WantAgentHelper>();
-    std::shared_ptr<AbilityRuntime::Context> context = OHOS::AbilityRuntime::Context::GetApplicationContext();
+    std::shared_ptr<AbilityRuntime::Context> context = std::make_shared<AbilityRuntime::ContextImpl>();
     std::shared_ptr<Want> want = std::make_shared<Want>();
     ElementName element("device", "bundleName", "abilityName");
     want->SetElement(element);
@@ -297,7 +298,7 @@ HWTEST_F(WantAgentHelperTest, WantAgentHelper_1300, Function | MediumTest | Leve
 HWTEST_F(WantAgentHelperTest, WantAgentHelper_1400, Function | MediumTest | Level1)
 {
     std::shared_ptr<WantAgentHelper> wantAgentHelper = std::make_shared<WantAgentHelper>();
-    std::shared_ptr<AbilityRuntime::Context> context = OHOS::AbilityRuntime::Context::GetApplicationContext();
+    std::shared_ptr<AbilityRuntime::Context> context = std::make_shared<AbilityRuntime::ContextImpl>();
     std::shared_ptr<Want> want = std::make_shared<Want>();
     ElementName element("device", "bundleName", "abilityName");
     want->SetElement(element);
@@ -321,7 +322,7 @@ HWTEST_F(WantAgentHelperTest, WantAgentHelper_1400, Function | MediumTest | Leve
 HWTEST_F(WantAgentHelperTest, WantAgentHelper_1500, Function | MediumTest | Level1)
 {
     std::shared_ptr<WantAgentHelper> wantAgentHelper = std::make_shared<WantAgentHelper>();
-    std::shared_ptr<AbilityRuntime::Context> context = OHOS::AbilityRuntime::Context::GetApplicationContext();
+    std::shared_ptr<AbilityRuntime::Context> context = std::make_shared<AbilityRuntime::ContextImpl>();
     std::shared_ptr<Want> want = std::make_shared<Want>();
     ElementName element("device", "bundleName", "abilityName");
     want->SetElement(element);
@@ -345,7 +346,7 @@ HWTEST_F(WantAgentHelperTest, WantAgentHelper_1500, Function | MediumTest | Leve
 HWTEST_F(WantAgentHelperTest, WantAgentHelper_1600, Function | MediumTest | Level1)
 {
     std::shared_ptr<WantAgentHelper> wantAgentHelper = std::make_shared<WantAgentHelper>();
-    std::shared_ptr<AbilityRuntime::Context> context = OHOS::AbilityRuntime::Context::GetApplicationContext();
+    std::shared_ptr<AbilityRuntime::Context> context = std::make_shared<AbilityRuntime::ContextImpl>();
     std::shared_ptr<Want> want = std::make_shared<Want>();
     ElementName element("device", "bundleName", "abilityName");
     want->SetElement(element);