1 /*
2 * Copyright (c) 2024 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
18 #include "access_token_error.h"
19 #include "access_token.h"
20 #define private public
21 #include "access_token_db.h"
22 #include "access_token_open_callback.h"
23 #undef private
24 #include "data_translator.h"
25 #include "token_field_const.h"
26
27 using namespace testing::ext;
28
29 namespace OHOS {
30 namespace Security {
31 namespace AccessToken {
32 namespace {
33 static constexpr uint32_t NOT_EXSIT_ATM_TYPE = 9;
34 }
35 class AccessTokenDatabaseCoverageTest : public testing::Test {
36 public:
37 static void SetUpTestCase();
38
39 static void TearDownTestCase();
40
41 void SetUp();
42
43 void TearDown();
44 };
45
SetUpTestCase()46 void AccessTokenDatabaseCoverageTest::SetUpTestCase() {}
47
TearDownTestCase()48 void AccessTokenDatabaseCoverageTest::TearDownTestCase() {}
49
SetUp()50 void AccessTokenDatabaseCoverageTest::SetUp() {}
51
TearDown()52 void AccessTokenDatabaseCoverageTest::TearDown() {}
53
54 /*
55 * @tc.name: ToRdbValueBuckets001
56 * @tc.desc: AccessTokenDbUtil::ToRdbValueBuckets
57 * @tc.type: FUNC
58 * @tc.require:
59 */
60 HWTEST_F(AccessTokenDatabaseCoverageTest, ToRdbValueBuckets001, TestSize.Level4)
61 {
62 std::vector<GenericValues> values;
63 GenericValues value;
64 values.emplace_back(value);
65 std::vector<NativeRdb::ValuesBucket> buckets;
66 AccessTokenDbUtil::ToRdbValueBuckets(values, buckets);
67 ASSERT_EQ(true, buckets.empty());
68 }
69
70 /*
71 * @tc.name: TranslationIntoPermissionStatus001
72 * @tc.desc: DataTranslator::TranslationIntoPermissionStatus
73 * @tc.type: FUNC
74 * @tc.require:
75 */
76 HWTEST_F(AccessTokenDatabaseCoverageTest, TranslationIntoPermissionStatus001, TestSize.Level4)
77 {
78 GenericValues value;
79 value.Put(TokenFiledConst::FIELD_GRANT_IS_GENERAL, 1);
80 value.Put(TokenFiledConst::FIELD_PERMISSION_NAME, "ohos.permission.READ_MEDIA");
81 value.Put(TokenFiledConst::FIELD_DEVICE_ID, "local");
82 value.Put(TokenFiledConst::FIELD_GRANT_FLAG, static_cast<int32_t>(PermissionFlag::PERMISSION_ALLOW_THIS_TIME));
83 value.Put(TokenFiledConst::FIELD_GRANT_STATE, static_cast<int32_t>(PermissionState::PERMISSION_GRANTED));
84 ASSERT_EQ(static_cast<int32_t>(PermissionState::PERMISSION_GRANTED),
85 value.GetInt(TokenFiledConst::FIELD_GRANT_STATE));
86
87 PermissionStatus permissionState;
88 DataTranslator::TranslationIntoPermissionStatus(value, permissionState);
89 ASSERT_EQ(static_cast<int32_t>(PermissionState::PERMISSION_DENIED), permissionState.grantStatus);
90 }
91
92 /*
93 * @tc.name: OnCreate001
94 * @tc.desc: AccessTokenOpenCallback::OnCreate
95 * @tc.type: FUNC
96 * @tc.require:
97 */
98 HWTEST_F(AccessTokenDatabaseCoverageTest, OnCreate001, TestSize.Level4)
99 {
100 std::shared_ptr<NativeRdb::RdbStore> db = AccessTokenDb::GetInstance()->GetRdb();
101 AccessTokenOpenCallback callback;
102 ASSERT_EQ(NativeRdb::E_OK, callback.OnCreate(*(db.get())));
103 }
104
105 /*
106 * @tc.name: OnUpgrade001
107 * @tc.desc: AccessTokenOpenCallback::OnUpgrade
108 * @tc.type: FUNC
109 * @tc.require:
110 */
111 HWTEST_F(AccessTokenDatabaseCoverageTest, OnUpgrade001, TestSize.Level4)
112 {
113 std::shared_ptr<NativeRdb::RdbStore> db = AccessTokenDb::GetInstance()->GetRdb();
114 AccessTokenOpenCallback callback;
115 ASSERT_EQ(NativeRdb::E_OK, callback.OnUpgrade(*(db.get()), DATABASE_VERSION_1, DATABASE_VERSION_2));
116 ASSERT_EQ(NativeRdb::E_OK, callback.OnUpgrade(*(db.get()), DATABASE_VERSION_1, DATABASE_VERSION_3));
117 ASSERT_EQ(NativeRdb::E_OK, callback.OnUpgrade(*(db.get()), DATABASE_VERSION_1, DATABASE_VERSION_4));
118 ASSERT_EQ(NativeRdb::E_OK, callback.OnUpgrade(*(db.get()), DATABASE_VERSION_1, DATABASE_VERSION_5));
119 ASSERT_EQ(NativeRdb::E_OK, callback.OnUpgrade(*(db.get()), DATABASE_VERSION_1, DATABASE_VERSION_6));
120
121 ASSERT_EQ(NativeRdb::E_OK, callback.OnUpgrade(*(db.get()), DATABASE_VERSION_2, DATABASE_VERSION_3));
122 ASSERT_EQ(NativeRdb::E_OK, callback.OnUpgrade(*(db.get()), DATABASE_VERSION_2, DATABASE_VERSION_4));
123 ASSERT_EQ(NativeRdb::E_OK, callback.OnUpgrade(*(db.get()), DATABASE_VERSION_2, DATABASE_VERSION_5));
124 ASSERT_EQ(NativeRdb::E_OK, callback.OnUpgrade(*(db.get()), DATABASE_VERSION_2, DATABASE_VERSION_6));
125
126 ASSERT_EQ(NativeRdb::E_OK, callback.OnUpgrade(*(db.get()), DATABASE_VERSION_3, DATABASE_VERSION_4));
127 ASSERT_EQ(NativeRdb::E_OK, callback.OnUpgrade(*(db.get()), DATABASE_VERSION_3, DATABASE_VERSION_5));
128 ASSERT_EQ(NativeRdb::E_OK, callback.OnUpgrade(*(db.get()), DATABASE_VERSION_3, DATABASE_VERSION_6));
129
130 ASSERT_EQ(NativeRdb::E_OK, callback.OnUpgrade(*(db.get()), DATABASE_VERSION_4, DATABASE_VERSION_5));
131 ASSERT_EQ(NativeRdb::E_OK, callback.OnUpgrade(*(db.get()), DATABASE_VERSION_4, DATABASE_VERSION_6));
132
133 ASSERT_EQ(NativeRdb::E_OK, callback.OnUpgrade(*(db.get()), DATABASE_VERSION_5, DATABASE_VERSION_6));
134 }
135
136 /*
137 * @tc.name: Modify001
138 * @tc.desc: AccessTokenDb::Modify
139 * @tc.type: FUNC
140 * @tc.require:
141 */
142 HWTEST_F(AccessTokenDatabaseCoverageTest, Modify001, TestSize.Level4)
143 {
144 AtmDataType type = static_cast<AtmDataType>(NOT_EXSIT_ATM_TYPE);
145 GenericValues modifyValue;
146 GenericValues conditionValue;
147 ASSERT_EQ(AccessTokenError::ERR_PARAM_INVALID,
148 AccessTokenDb::GetInstance()->Modify(type, modifyValue, conditionValue));
149
150 type = AtmDataType::ACCESSTOKEN_HAP_INFO;
151 ASSERT_EQ(AccessTokenError::ERR_PARAM_INVALID,
152 AccessTokenDb::GetInstance()->Modify(type, modifyValue, conditionValue));
153
154 modifyValue.Put(TokenFiledConst::FIELD_PROCESS_NAME, "hdcd");
155 ASSERT_EQ(NativeRdb::E_SQLITE_ERROR, AccessTokenDb::GetInstance()->Modify(type, modifyValue, conditionValue));
156
157 conditionValue.Put(TokenFiledConst::FIELD_PROCESS_NAME, "hdcd");
158 ASSERT_NE(NativeRdb::E_OK, AccessTokenDb::GetInstance()->Modify(type, modifyValue, conditionValue));
159
160 int32_t resultCode = NativeRdb::E_SQLITE_ERROR;
161 int32_t changedRows = 0;
162 NativeRdb::ValuesBucket bucket;
163 AccessTokenDbUtil::ToRdbValueBucket(modifyValue, bucket);
164 NativeRdb::RdbPredicates predicates("hap_token_info_table");
165 AccessTokenDbUtil::ToRdbPredicates(conditionValue, predicates);
166 std::shared_ptr<NativeRdb::RdbStore> db = AccessTokenDb::GetInstance()->GetRdb();
167 ASSERT_EQ(NativeRdb::E_SQLITE_ERROR,
168 AccessTokenDb::GetInstance()->RestoreAndUpdateIfCorrupt(resultCode, changedRows, bucket, predicates, db));
169
170 resultCode = NativeRdb::E_SQLITE_CORRUPT;
171 ASSERT_NE(NativeRdb::E_OK,
172 AccessTokenDb::GetInstance()->RestoreAndUpdateIfCorrupt(resultCode, changedRows, bucket, predicates, db));
173 }
174
175 /*
176 * @tc.name: Find001
177 * @tc.desc: AccessTokenDb::Find
178 * @tc.type: FUNC
179 * @tc.require:
180 */
181 HWTEST_F(AccessTokenDatabaseCoverageTest, Find001, TestSize.Level4)
182 {
183 AtmDataType type = static_cast<AtmDataType>(NOT_EXSIT_ATM_TYPE);
184 GenericValues conditionValue;
185 std::vector<GenericValues> results;
186 ASSERT_EQ(AccessTokenError::ERR_PARAM_INVALID,
187 AccessTokenDb::GetInstance()->Find(type, conditionValue, results));
188
189 type = AtmDataType::ACCESSTOKEN_HAP_INFO;
190 ASSERT_EQ(NativeRdb::E_OK, AccessTokenDb::GetInstance()->Find(type, conditionValue, results));
191
192 conditionValue.Put(TokenFiledConst::FIELD_PROCESS_NAME, "hdcd");
193 ASSERT_EQ(AccessTokenError::ERR_DATABASE_OPERATE_FAILED,
194 AccessTokenDb::GetInstance()->Find(type, conditionValue, results));
195 }
196 } // namespace AccessToken
197 } // namespace Security
198 } // namespace OHOS
199