diff --git a/services/dbms/test/unittest/dbms_services_kit_test/BUILD.gn b/services/dbms/test/unittest/dbms_services_kit_test/BUILD.gn index 858e0d2da019df58b4b001ce2e17dbcce5f6e9e5..0b2d61f8da05b63d90ec40547902f9b0f95465ff 100644 --- a/services/dbms/test/unittest/dbms_services_kit_test/BUILD.gn +++ b/services/dbms/test/unittest/dbms_services_kit_test/BUILD.gn @@ -40,6 +40,8 @@ ohos_unittest("DbmsServicesKitTest") { external_deps = [ "ability_base:want", "access_token:libaccesstoken_sdk", + "access_token:libnativetoken", + "access_token:libtoken_setproc", "access_token:libtokenid_sdk", "bundle_framework:appexecfwk_base", "bundle_framework:appexecfwk_core", diff --git a/services/dbms/test/unittest/dbms_services_kit_test/dbms_services_kit_test.cpp b/services/dbms/test/unittest/dbms_services_kit_test/dbms_services_kit_test.cpp index 84ff5330da89eada08710f295aa88736fb35ee02..534568c86498fbbcc7c0c0e21efb4a3ecc1f1e7f 100644 --- a/services/dbms/test/unittest/dbms_services_kit_test/dbms_services_kit_test.cpp +++ b/services/dbms/test/unittest/dbms_services_kit_test/dbms_services_kit_test.cpp @@ -22,6 +22,7 @@ #include #include +#include "accesstoken_kit.h" #include "appexecfwk_errors.h" #include "dbms_device_manager.h" #include "distributed_ability_info.h" @@ -34,7 +35,10 @@ #include "event_report.h" #include "image_compress.h" #include "json_util.h" +#include "nativetoken_kit.h" +#include "token_setproc.h" #include "service_control.h" +#include "softbus_common.h" using namespace testing::ext; using namespace std::chrono_literals; @@ -93,6 +97,26 @@ void DbmsServicesKitTest::TearDownTestCase() void DbmsServicesKitTest::SetUp() { + const int32_t PERMS_NUM = 3; + const int32_t PERMS_INDEX_TWO = 2; + uint64_t tokenId; + const char *perms[PERMS_NUM]; + perms[0] = OHOS_PERMISSION_DISTRIBUTED_SOFTBUS_CENTER; + perms[1] = OHOS_PERMISSION_DISTRIBUTED_DATASYNC; + perms[PERMS_INDEX_TWO] = "ohos.permission.ACCESS_SERVICE_DM"; + NativeTokenInfoParams infoInstance = { + .dcapsNum = 0, + .permsNum = PERMS_NUM, + .aclsNum = 0, + .dcaps = NULL, + .perms = perms, + .acls = NULL, + .processName = "dsoftbus_service", + .aplStr = "system_core", + }; + tokenId = GetAccessTokenId(&infoInstance); + SetSelfTokenID(tokenId); + OHOS::Security::AccessToken::AccessTokenKit::ReloadNativeTokenInfo(); std::string strExtra = std::to_string(402); auto extraArgv = strExtra.c_str(); ServiceControlWithExtra("d-bms", START, &extraArgv, 1); @@ -1525,6 +1549,6 @@ HWTEST_F(DbmsServicesKitTest, VerifyCallingPermission_0200, Function | MediumTes auto distributedBms = GetDistributedBms(); EXPECT_NE(distributedBms, nullptr); int res = distributedBms->VerifyCallingPermission(""); - EXPECT_FALSE(res); + EXPECT_TRUE(res); } } // OHOS \ No newline at end of file