From 43ec0b23e8200382514e0d53d9c628fc44e84db2 Mon Sep 17 00:00:00 2001
From: zhaochaoyan <zhaochaoyan1@huawei.com>
Date: Fri, 13 Oct 2023 15:14:45 +0800
Subject: [PATCH] =?UTF-8?q?=E4=BF=AE=E5=A4=8Dinput=E5=A4=B1=E8=B4=A5UT?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

Signed-off-by: zhaochaoyan <zhaochaoyan1@huawei.com>
---
 .../BUILD.gn                                  |  2 ++
 .../add_white_list_infos_call_back_test.cpp   | 22 ++++++++++++++++++
 .../dinputsourcecallbackunittest/BUILD.gn     |  2 ++
 .../dinput_source_callback_unittest.cpp       | 23 +++++++++++++++++++
 4 files changed, 49 insertions(+)

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()
-- 
Gitee