diff --git a/test/unittest/BUILD.gn b/test/unittest/BUILD.gn index 9d78bcc48f68770bf90fa56c4b7680e7d1adaf7f..9bab8b29458c9600a674742069ba123509ffce02 100644 --- a/test/unittest/BUILD.gn +++ b/test/unittest/BUILD.gn @@ -93,6 +93,9 @@ ohos_unittest("DeviceUsageStatsTest") { ] external_deps = [ + "access_token:libaccesstoken_sdk", + "access_token:libnativetoken", + "access_token:libtoken_setproc", "c_utils:utils", "eventhandler:libeventhandler", "ffrt:libffrt", @@ -127,6 +130,9 @@ ohos_unittest("DeviceUsageStatsMultiTest") { ] external_deps = [ + "access_token:libaccesstoken_sdk", + "access_token:libnativetoken", + "access_token:libtoken_setproc", "c_utils:utils", "eventhandler:libeventhandler", "ffrt:libffrt", @@ -164,6 +170,8 @@ ohos_unittest("DeviceUsageStatsServiceTest") { "ability_runtime:app_manager", "ability_runtime:wantagent_innerkits", "access_token:libaccesstoken_sdk", + "access_token:libnativetoken", + "access_token:libtoken_setproc", "bundle_framework:appexecfwk_core", "c_utils:utils", "common_event_service:cesfwk_innerkits", diff --git a/test/unittest/device_usage_statistics_multi_test.cpp b/test/unittest/device_usage_statistics_multi_test.cpp index 01e908f62fc0ef02bdc6236409d99493a44e4d0b..2c37a5ea7e6d161bd6d1f30de2c3514a3998509d 100644 --- a/test/unittest/device_usage_statistics_multi_test.cpp +++ b/test/unittest/device_usage_statistics_multi_test.cpp @@ -32,12 +32,16 @@ #include "app_group_callback_proxy.h" #include "bundle_active_log.h" #include "iapp_group_callback.h" +#include "accesstoken_kit.h" +#include "token_setproc.h" +#include "nativetoken_kit.h" using namespace testing::ext; using namespace testing::mt; namespace OHOS { namespace DeviceUsageStats { +using namespace Security::AccessToken; #ifdef __aarch64__ static std::string g_defaultBundleName = "com.huawei.hmos.camera"; #else @@ -67,6 +71,23 @@ public: void DeviceUsageStatisticsMultiTest::SetUpTestCase(void) { + static const char *perms[] = { + "ohos.permission.BUNDLE_ACTIVE_INFO", + }; + uint64_t tokenId; + NativeTokenInfoParams infoInstance = { + .dcapsNum = 0, + .permsNum = 1, + .aclsNum = 0, + .dcaps = nullptr, + .perms = perms, + .acls = nullptr, + .processName = "DeviceUsageStatisticsMultiTest", + .aplStr = "system_core", + }; + tokenId = GetAccessTokenId(&infoInstance); + SetSelfTokenID(tokenId); + AccessTokenKit::ReloadNativeTokenInfo(); } void DeviceUsageStatisticsMultiTest::TearDownTestCase(void) diff --git a/test/unittest/device_usage_statistics_service_test.cpp b/test/unittest/device_usage_statistics_service_test.cpp index 134203bdb6b5c3b4a3f3056664877f35f3dc9776..f45ba49df6424e180a672d2c1fb5f5732f7c9ebc 100644 --- a/test/unittest/device_usage_statistics_service_test.cpp +++ b/test/unittest/device_usage_statistics_service_test.cpp @@ -29,11 +29,15 @@ #include "bundle_active_group_controller.h" #include "bundle_active_log.h" #include "bundle_active_config_reader.h" +#include "accesstoken_kit.h" +#include "token_setproc.h" +#include "nativetoken_kit.h" using namespace testing::ext; namespace OHOS { namespace DeviceUsageStats { +using namespace Security::AccessToken; class DeviceUsageStatisticsServiceTest : public testing::Test { public: static void SetUpTestCase(void); @@ -47,6 +51,23 @@ std::shared_ptr DeviceUsageStatisticsServiceTest::bundleActive void DeviceUsageStatisticsServiceTest::SetUpTestCase(void) { + static const char *perms[] = { + "ohos.permission.BUNDLE_ACTIVE_INFO", + }; + uint64_t tokenId; + NativeTokenInfoParams infoInstance = { + .dcapsNum = 0, + .permsNum = 1, + .aclsNum = 0, + .dcaps = nullptr, + .perms = perms, + .acls = nullptr, + .processName = "DeviceUsageStatisticsServiceTest", + .aplStr = "system_core", + }; + tokenId = GetAccessTokenId(&infoInstance); + SetSelfTokenID(tokenId); + AccessTokenKit::ReloadNativeTokenInfo(); bundleActiveCore_ = std::make_shared(); bundleActiveCore_->Init(); bundleActiveCore_->InitBundleGroupController(); diff --git a/test/unittest/device_usage_statistics_test.cpp b/test/unittest/device_usage_statistics_test.cpp index 63e75e744a52728e9634b8fe46bd3bf7c67be270..20783e94a728cce75edcef41529f15bd1580822c 100644 --- a/test/unittest/device_usage_statistics_test.cpp +++ b/test/unittest/device_usage_statistics_test.cpp @@ -30,11 +30,15 @@ #include "app_group_callback_proxy.h" #include "iapp_group_callback.h" #include "bundle_active_log.h" +#include "accesstoken_kit.h" +#include "token_setproc.h" +#include "nativetoken_kit.h" using namespace testing::ext; namespace OHOS { namespace DeviceUsageStats { +using namespace Security::AccessToken; #ifdef __aarch64__ static std::string g_defaultBundleName = "com.huawei.hmos.camera"; #else @@ -62,6 +66,23 @@ public: void DeviceUsageStatisticsTest::SetUpTestCase(void) { + static const char *perms[] = { + "ohos.permission.BUNDLE_ACTIVE_INFO", + }; + uint64_t tokenId; + NativeTokenInfoParams infoInstance = { + .dcapsNum = 0, + .permsNum = 1, + .aclsNum = 0, + .dcaps = nullptr, + .perms = perms, + .acls = nullptr, + .processName = "DeviceUsageStatisticsTest", + .aplStr = "system_core", + }; + tokenId = GetAccessTokenId(&infoInstance); + SetSelfTokenID(tokenId); + AccessTokenKit::ReloadNativeTokenInfo(); } void DeviceUsageStatisticsTest::TearDownTestCase(void) @@ -205,7 +226,7 @@ HWTEST_F(DeviceUsageStatisticsTest, DeviceUsageStatisticsTest_IsBundleUsePeriod_ bool result = false; int32_t errCode = BundleActiveClient::GetInstance().IsBundleUsePeriod(result, g_defaultBundleName, g_defaultUserId); EXPECT_EQ(result, false); - EXPECT_EQ(errCode, ERR_PERMISSION_DENIED); + EXPECT_EQ(errCode, 0); } /*