1 /*
2 * Copyright (c) 2022-2023 Huawei Device Co., Ltd.
3 * Licensed under the Apache License, Version 2.0 (the "License");
4 * you may not use this file except in compliance with the License.
5 * You may obtain a copy of the License at
6 *
7 * http://www.apache.org/licenses/LICENSE-2.0
8 *
9 * Unless required by applicable law or agreed to in writing, software
10 * distributed under the License is distributed on an "AS IS" BASIS,
11 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
12 * See the License for the specific language governing permissions and
13 * limitations under the License.
14 */
15
16 #include <gtest/gtest.h>
17 #include <string>
18 #include <vector>
19 #include "iplugin.h"
20 #include "permission_manager.h"
21
22 using namespace testing::ext;
23
24 namespace OHOS {
25 namespace EDM {
26 namespace TEST {
27 constexpr int POLICY_CODE = 100;
28 class PermissionManagerTest : public testing::Test {
29 protected:
30 // Sets up the test fixture.
31 void SetUp() override;
32
33 // Tears down the test fixture.
34 void TearDown() override;
35 };
36
SetUp()37 void PermissionManagerTest::SetUp()
38 {
39 }
40
TearDown()41 void PermissionManagerTest::TearDown()
42 {
43 PermissionManager::DestroyInstance();
44 }
45
46 /**
47 * @tc.name: TestAddPermission
48 * @tc.desc: Test PermissionManager AddPermission func.
49 * @tc.type: FUNC
50 */
51 HWTEST_F(PermissionManagerTest, TestAddPermission, TestSize.Level1)
52 {
53 ASSERT_EQ(PermissionManager::GetInstance()->AddPermission(
54 std::string("ohos.permission.EDM_TEST_PERMISSION"), IPlugin::PermissionType::NORMAL_DEVICE_ADMIN, POLICY_CODE),
55 ERR_OK);
56 }
57
58 /**
59 * @tc.name: GetReqPermission01
60 * @tc.desc: Test PermissionManager ReqPermission func.
61 * @tc.type: FUNC
62 */
63 HWTEST_F(PermissionManagerTest, GetReqPermission01, TestSize.Level1)
64 {
65 PermissionManager::GetInstance()->AddPermission(
66 std::string("ohos.permission.EDM_TEST_PERMISSION"), IPlugin::PermissionType::NORMAL_DEVICE_ADMIN, POLICY_CODE);
67 std::vector<std::string> permission = {
68 "ohos.permission.EDM_TEST_PERMISSION", "ohos.permission.EMD_TEST_PERMISSION_FAIL" };
69 std::vector<std::string> reqPermission;
70 PermissionManager::GetInstance()->GetAdminGrantedPermission(permission, AdminType::NORMAL, reqPermission);
71 ASSERT_TRUE(reqPermission.size() == 1);
72 }
73
74 /**
75 * @tc.name: GetReqPermission02
76 * @tc.desc: Test PermissionManager ReqPermission func.
77 * @tc.type: FUNC
78 */
79 HWTEST_F(PermissionManagerTest, GetReqPermission02, TestSize.Level1)
80 {
81 PermissionManager::GetInstance()->AddPermission(
82 std::string("ohos.permission.EDM_TEST_ENT_PERMISSION"),
83 IPlugin::PermissionType::SUPER_DEVICE_ADMIN, POLICY_CODE);
84 std::vector<std::string> permission = {
85 "ohos.permission.EDM_TEST_ENT_PERMISSION", "ohos.permission.EMD_TEST_PERMISSION_FAIL" };
86 std::vector<std::string> reqPermission;
87 PermissionManager::GetInstance()->GetAdminGrantedPermission(permission, AdminType::NORMAL, reqPermission);
88 ASSERT_TRUE(reqPermission.empty());
89 PermissionManager::GetInstance()->GetAdminGrantedPermission(permission, AdminType::ENT, reqPermission);
90 ASSERT_TRUE(reqPermission.size() == 1);
91 }
92
93 /**
94 * @tc.name: GetReqPermission03
95 * @tc.desc: Test PermissionManager ReqPermission func.
96 * @tc.type: FUNC
97 */
98 HWTEST_F(PermissionManagerTest, GetReqPermission03, TestSize.Level1)
99 {
100 PermissionManager::GetInstance()->AddPermission(
101 std::string("ohos.permission.EDM_TEST_ENT_PERMISSION"),
102 IPlugin::PermissionType::SUPER_DEVICE_ADMIN, POLICY_CODE);
103 std::vector<std::string> permission = {
104 "ohos.permission.EDM_TEST_ENT_PERMISSION", "ohos.permission.EMD_TEST_PERMISSION_FAIL" };
105 std::vector<std::string> reqPermission;
106 PermissionManager::GetInstance()->GetAdminGrantedPermission(permission, AdminType::NORMAL, reqPermission);
107 ASSERT_TRUE(reqPermission.empty());
108 PermissionManager::GetInstance()->GetAdminGrantedPermission(permission, AdminType::BYOD, reqPermission);
109 ASSERT_TRUE(reqPermission.empty());
110 PermissionManager::GetInstance()->GetAdminGrantedPermission(permission, AdminType::ENT, reqPermission);
111 ASSERT_TRUE(reqPermission.size() == 1);
112 }
113 } // namespace TEST
114 } // namespace EDM
115 } // namespace OHOS
116