diff --git a/interfaces/ipc/test/addwhitelistinfoscallbackunittest/BUILD.gn b/interfaces/ipc/test/addwhitelistinfoscallbackunittest/BUILD.gn
index eecbd18..a974bb9 100644
--- a/interfaces/ipc/test/addwhitelistinfoscallbackunittest/BUILD.gn
+++ b/interfaces/ipc/test/addwhitelistinfoscallbackunittest/BUILD.gn
@@ -93,6 +93,8 @@ ohos_unittest("add_white_list_infos_callbackl_test") {
 
   external_deps = [
     "access_token:libaccesstoken_sdk",
+    "access_token:libnativetoken",
+    "access_token:libtoken_setproc",
     "access_token:libtokenid_sdk",
     "c_utils:utils",
     "distributed_hardware_fwk:libdhfwk_sdk",
diff --git a/interfaces/ipc/test/addwhitelistinfoscallbackunittest/add_white_list_infos_call_back_test.cpp b/interfaces/ipc/test/addwhitelistinfoscallbackunittest/add_white_list_infos_call_back_test.cpp
index 5b7c14e..6ffb823 100644
--- a/interfaces/ipc/test/addwhitelistinfoscallbackunittest/add_white_list_infos_call_back_test.cpp
+++ b/interfaces/ipc/test/addwhitelistinfoscallbackunittest/add_white_list_infos_call_back_test.cpp
@@ -15,13 +15,35 @@
 
 #include "add_white_list_infos_call_back_test.h"
 
+#include "accesstoken_kit.h"
 #include "dinput_errcode.h"
+#include "nativetoken_kit.h"
+#include "token_setproc.h"
+#include "softbus_common.h"
 
 namespace OHOS {
 namespace DistributedHardware {
 namespace DistributedInput {
 void AddWhiteListInfosCallbackTest::SetUp()
 {
+    uint64_t tokenId;
+    const char *perms[3];
+    perms[0] = "ohos.permission.ENABLE_DISTRIBUTED_HARDWARE";
+    perms[1] = OHOS_PERMISSION_DISTRIBUTED_DATASYNC;
+    perms[2] = "ohos.permission.ACCESS_DISTRIBUTED_HARDWARE";
+    NativeTokenInfoParams infoInstance = {
+        .dcapsNum = 0,
+        .permsNum = 3,
+        .aclsNum = 0,
+        .dcaps = NULL,
+        .perms = perms,
+        .acls = NULL,
+        .processName = "dinput",
+        .aplStr = "system_core",
+    };
+    tokenId = GetAccessTokenId(&infoInstance);
+    SetSelfTokenID(tokenId);
+    OHOS::Security::AccessToken::AccessTokenKit::ReloadNativeTokenInfo();
 }
 
 void AddWhiteListInfosCallbackTest::TearDown()
diff --git a/interfaces/ipc/test/dinputsourcecallbackunittest/BUILD.gn b/interfaces/ipc/test/dinputsourcecallbackunittest/BUILD.gn
index 13dfbe5..df58c62 100644
--- a/interfaces/ipc/test/dinputsourcecallbackunittest/BUILD.gn
+++ b/interfaces/ipc/test/dinputsourcecallbackunittest/BUILD.gn
@@ -93,6 +93,8 @@ ohos_unittest("dinput_source_callback_test") {
 
   external_deps = [
     "access_token:libaccesstoken_sdk",
+    "access_token:libnativetoken",
+    "access_token:libtoken_setproc",
     "access_token:libtokenid_sdk",
     "c_utils:utils",
     "distributed_hardware_fwk:libdhfwk_sdk",
diff --git a/interfaces/ipc/test/dinputsourcecallbackunittest/dinput_source_callback_unittest.cpp b/interfaces/ipc/test/dinputsourcecallbackunittest/dinput_source_callback_unittest.cpp
index fc10ad8..30a8de3 100644
--- a/interfaces/ipc/test/dinputsourcecallbackunittest/dinput_source_callback_unittest.cpp
+++ b/interfaces/ipc/test/dinputsourcecallbackunittest/dinput_source_callback_unittest.cpp
@@ -16,14 +16,37 @@
 #include "dinput_source_callback_unittest.h"
 
 #include <memory>
+
+#include "accesstoken_kit.h"
 #include "dinput_errcode.h"
 #include "dinput_ipc_interface_code.h"
+#include "nativetoken_kit.h"
+#include "token_setproc.h"
+#include "softbus_common.h"
 
 namespace OHOS {
 namespace DistributedHardware {
 namespace DistributedInput {
 void DInputSourceCallBackTest::SetUp()
 {
+    uint64_t tokenId;
+    const char *perms[3];
+    perms[0] = "ohos.permission.ENABLE_DISTRIBUTED_HARDWARE";
+    perms[1] = OHOS_PERMISSION_DISTRIBUTED_DATASYNC;
+    perms[2] = "ohos.permission.ACCESS_DISTRIBUTED_HARDWARE";
+    NativeTokenInfoParams infoInstance = {
+        .dcapsNum = 0,
+        .permsNum = 3,
+        .aclsNum = 0,
+        .dcaps = NULL,
+        .perms = perms,
+        .acls = NULL,
+        .processName = "dinput",
+        .aplStr = "system_core",
+    };
+    tokenId = GetAccessTokenId(&infoInstance);
+    SetSelfTokenID(tokenId);
+    OHOS::Security::AccessToken::AccessTokenKit::ReloadNativeTokenInfo();
 }
 
 void DInputSourceCallBackTest::TearDown()