1 /*
2 * Copyright (c) 2021 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 "rdb_helper.h"
17
18 #include <gtest/gtest.h>
19
20 #include <string>
21
22 #include "common.h"
23 #include "rdb_errno.h"
24 #include "rdb_open_callback.h"
25
26 using namespace testing::ext;
27 using namespace OHOS::NativeRdb;
28
29 class OpenCallback : public RdbOpenCallback {
30 public:
OnCreate(RdbStore & store)31 int OnCreate(RdbStore &store) override
32 {
33 return E_OK;
34 }
OnUpgrade(RdbStore & store,int oldVersion,int newVersion)35 int OnUpgrade(RdbStore &store, int oldVersion, int newVersion) override
36 {
37 return E_OK;
38 }
39 };
40
41 class RdbHelperTest : public testing::Test {
42 public:
43 static void SetUpTestCase(void);
44 static void TearDownTestCase(void);
45 void SetUp();
46 void TearDown();
47 static const std::string rdbStorePath;
48 };
49 const std::string RdbHelperTest::rdbStorePath = RDB_TEST_PATH + std::string("rdbhelper.db");
50
SetUpTestCase(void)51 void RdbHelperTest::SetUpTestCase(void)
52 {
53 }
54
TearDownTestCase(void)55 void RdbHelperTest::TearDownTestCase(void)
56 {
57 RdbHelper::DeleteRdbStore(rdbStorePath);
58 }
59
SetUp(void)60 void RdbHelperTest::SetUp(void)
61 {
62 }
63
TearDown(void)64 void RdbHelperTest::TearDown(void)
65 {
66 }
67
68 class RdbHelperTestWrongSqlOpenCallback : public RdbOpenCallback {
69 public:
70 int OnCreate(RdbStore &store) override;
71 int OnUpgrade(RdbStore &store, int oldVersion, int newVersion) override;
72 static const std::string WRONG_SQL_TEST;
73 };
74
75 class RdbHelperTestOpenCallback : public RdbOpenCallback {
76 public:
77 int OnCreate(RdbStore &store) override;
78 int OnUpgrade(RdbStore &store, int oldVersion, int newVersion) override;
79 static const std::string CREATE_TABLE_TEST;
80 };
81
82 const std::string RdbHelperTestWrongSqlOpenCallback::WRONG_SQL_TEST = "CREATE TABL IF NOT EXISTS test "
83 "(id INTEGER PRIMARY KEY AUTOINCREMENT, "
84 "name TEXT NOT NULL, age INTEGER, salary REAL, "
85 "blobType BLOB)";
86 const std::string RdbHelperTestOpenCallback::CREATE_TABLE_TEST = "CREATE TABLE IF NOT EXISTS test "
87 "(id INTEGER PRIMARY KEY AUTOINCREMENT, "
88 "name TEXT NOT NULL, age INTEGER, salary REAL, "
89 "blobType BLOB)";
90
OnCreate(RdbStore & store)91 int RdbHelperTestWrongSqlOpenCallback::OnCreate(RdbStore &store)
92 {
93 return store.ExecuteSql(WRONG_SQL_TEST);
94 }
95
OnCreate(RdbStore & store)96 int RdbHelperTestOpenCallback::OnCreate(RdbStore &store)
97 {
98 return store.ExecuteSql(CREATE_TABLE_TEST);
99 }
100
OnUpgrade(RdbStore & store,int oldVersion,int newVersion)101 int RdbHelperTestWrongSqlOpenCallback::OnUpgrade(RdbStore &store, int oldVersion, int newVersion)
102 {
103 return E_OK;
104 }
105
OnUpgrade(RdbStore & store,int oldVersion,int newVersion)106 int RdbHelperTestOpenCallback::OnUpgrade(RdbStore &store, int oldVersion, int newVersion)
107 {
108 return E_OK;
109 }
110
111 /**
112 * @tc.name: DeleteDatabaseCache_001
113 * @tc.desc: delete db cache
114 * @tc.type: FUNC
115 * @tc.require:
116 * @tc.author:
117 */
118 HWTEST_F(RdbHelperTest, DeleteDatabaseCache_001, TestSize.Level1)
119 {
120 int errCode = E_OK;
121 RdbStoreConfig config(RdbHelperTest::rdbStorePath);
122 RdbHelperTestWrongSqlOpenCallback helper;
123 std::shared_ptr<RdbStore> rdbStore = RdbHelper::GetRdbStore(config, 1, helper, errCode);
124 EXPECT_EQ(rdbStore, nullptr);
125 }
126
127 /**
128 * @tc.name: DeleteDatabase_001
129 * @tc.desc: delete db file
130 * @tc.type: FUNC
131 */
132 HWTEST_F(RdbHelperTest, DeleteDatabase_001, TestSize.Level1)
133 {
134 int errCode = E_OK;
135 RdbStoreConfig config1(RdbHelperTest::rdbStorePath);
136 RdbStoreConfig config2("test");
137 RdbStoreConfig config3("");
138 RdbHelperTestOpenCallback helper;
139 std::shared_ptr<RdbStore> rdbStore = RdbHelper::GetRdbStore(config1, 1, helper, errCode);
140 EXPECT_NE(rdbStore, nullptr);
141 int ret1 = RdbHelper::DeleteRdbStore(config1);
142 EXPECT_EQ(ret1, E_OK);
143 int ret2 = RdbHelper::DeleteRdbStore(config2);
144 EXPECT_EQ(ret2, E_OK);
145 int ret3 = RdbHelper::DeleteRdbStore(config3);
146 EXPECT_EQ(ret3, E_INVALID_FILE_PATH);
147 }
148
149 /**
150 * @tc.name: DeleteDatabase_002
151 * @tc.desc: DeleteRdbStore if the dbFile is not exists
152 * @tc.type: FUNC
153 */
154 HWTEST_F(RdbHelperTest, DeleteDatabase_002, TestSize.Level1)
155
156 {
157 int errCode = E_OK;
158 RdbStoreConfig config(RdbHelperTest::rdbStorePath);
159 RdbHelperTestOpenCallback helper;
160 std::shared_ptr<RdbStore> rdbStore = RdbHelper::GetRdbStore(config, 1, helper, errCode);
161 EXPECT_NE(rdbStore, nullptr);
162
163 remove(rdbStorePath.c_str());
164
165 int ret = RdbHelper::DeleteRdbStore("rdbhelper.db");
166 EXPECT_EQ(ret, E_OK);
167 std::string shmFileName = rdbStorePath + "-shm";
168 std::string walFileName = rdbStorePath + "-wal";
169 EXPECT_NE(access(shmFileName.c_str(), F_OK), 0);
170 EXPECT_NE(access(walFileName.c_str(), F_OK), 0);
171 }
172
173 /**
174 * @tc.name: DeleteDatabase_003
175 * @tc.desc: DeleteRdbStore if the dbFile is not exists
176 * @tc.type: FUNC
177 */
178 HWTEST_F(RdbHelperTest, DeleteDatabase_003, TestSize.Level1)
179 {
180 int errCode = E_OK;
181 RdbStoreConfig config(RdbHelperTest::rdbStorePath);
182 RdbHelperTestOpenCallback helper;
183 std::shared_ptr<RdbStore> rdbStore = RdbHelper::GetRdbStore(config, 1, helper, errCode);
184 EXPECT_NE(rdbStore, nullptr);
185
186 remove(rdbStorePath.c_str());
187
188 int ret = RdbHelper::DeleteRdbStore(config);
189
190 EXPECT_EQ(ret, E_OK);
191 std::string shmFileName = rdbStorePath + "-shm";
192 std::string walFileName = rdbStorePath + "-wal";
193 EXPECT_NE(access(shmFileName.c_str(), F_OK), 0);
194 EXPECT_NE(access(walFileName.c_str(), F_OK), 0);
195 }
196
197 /**
198 * @tc.name: getrdbstore_001
199 * @tc.desc: get db file with a invalid path
200 * @tc.type: FUNC
201 */
202 HWTEST_F(RdbHelperTest, GetDatabase_001, TestSize.Level0)
203 {
204 int errCode = E_OK;
205 RdbStoreConfig config("/invalid/invalid/test.db");
206 OpenCallback helper;
207 auto store = RdbHelper::GetRdbStore(config, 1, helper, errCode);
208 EXPECT_EQ(store, nullptr);
209 EXPECT_EQ(errCode, E_INVALID_FILE_PATH);
210 }
211
212 HWTEST_F(RdbHelperTest, GetDatabase_002, TestSize.Level0)
213 {
214 const std::string dbPath = RDB_TEST_PATH + "GetDatabase.db";
215 RdbStoreConfig config(dbPath);
216 std::string bundleName = "com.ohos.config.GetDatabase";
217 config.SetBundleName(bundleName);
218 config.SetArea(1);
219 config.SetEncryptStatus(true);
220
221 RdbHelper::DeleteRdbStore(config);
222
223 int errCode = E_OK;
224
225 RdbHelperTestOpenCallback helper;
226 std::shared_ptr<RdbStore> rdbStore1 = RdbHelper::GetRdbStore(config, 1, helper, errCode);
227 EXPECT_EQ(errCode, E_OK);
228 EXPECT_NE(rdbStore1, nullptr);
229
230 std::shared_ptr<RdbStore> rdbStore2 = RdbHelper::GetRdbStore(config, 1, helper, errCode);
231 EXPECT_EQ(errCode, E_OK);
232 EXPECT_NE(rdbStore2, nullptr);
233
234 EXPECT_EQ(rdbStore1, rdbStore2);
235 }
236
237 HWTEST_F(RdbHelperTest, GetDatabase_003, TestSize.Level0)
238 {
239 const std::string dbPath = RDB_TEST_PATH + "GetDatabase.db";
240 RdbStoreConfig config(dbPath);
241 std::string bundleName = "com.ohos.config.GetDatabase";
242 config.SetBundleName(bundleName);
243 config.SetArea(1);
244 config.SetEncryptStatus(true);
245
246 RdbHelper::DeleteRdbStore(config);
247
248 int errCode = E_OK;
249
250 RdbHelperTestOpenCallback helper;
251 std::shared_ptr<RdbStore> rdbStore1 = RdbHelper::GetRdbStore(config, 1, helper, errCode);
252 EXPECT_EQ(errCode, E_OK);
253 ASSERT_NE(rdbStore1, nullptr);
254
255 config.SetEncryptStatus(false);
256 std::shared_ptr<RdbStore> rdbStore2 = RdbHelper::GetRdbStore(config, 1, helper, errCode);
257 EXPECT_EQ(errCode, E_OK);
258 ASSERT_NE(rdbStore2, nullptr);
259
260 EXPECT_NE(rdbStore1, rdbStore2);
261 }
262