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: getrdbstore_001
151 * @tc.desc: get db file with a invalid path
152 * @tc.type: FUNC
153 */
154 HWTEST_F(RdbHelperTest, GetDatabase_001, TestSize.Level0)
155 {
156 int errCode = E_OK;
157 RdbStoreConfig config("/invalid/invalid/test.db");
158 OpenCallback helper;
159 auto store = RdbHelper::GetRdbStore(config, 1, helper, errCode);
160 EXPECT_EQ(store, nullptr);
161 EXPECT_EQ(errCode, E_INVALID_FILE_PATH);
162 }
163
164 HWTEST_F(RdbHelperTest, GetDatabase_002, TestSize.Level0)
165 {
166 const std::string dbPath = RDB_TEST_PATH + "GetDatabase.db";
167 RdbStoreConfig config(dbPath);
168 std::string bundleName = "com.ohos.config.GetDatabase";
169 config.SetBundleName(bundleName);
170 config.SetArea(1);
171 config.SetEncryptStatus(true);
172
173 RdbHelper::DeleteRdbStore(config);
174
175 int errCode = E_OK;
176
177 RdbHelperTestOpenCallback helper;
178 std::shared_ptr<RdbStore> rdbStore1 = RdbHelper::GetRdbStore(config, 1, helper, errCode);
179 EXPECT_EQ(errCode, E_OK);
180 EXPECT_NE(rdbStore1, nullptr);
181
182 std::shared_ptr<RdbStore> rdbStore2 = RdbHelper::GetRdbStore(config, 1, helper, errCode);
183 EXPECT_EQ(errCode, E_OK);
184 EXPECT_NE(rdbStore2, nullptr);
185
186 EXPECT_EQ(rdbStore1, rdbStore2);
187 }
188
189 HWTEST_F(RdbHelperTest, GetDatabase_003, TestSize.Level0)
190 {
191 const std::string dbPath = RDB_TEST_PATH + "GetDatabase.db";
192 RdbStoreConfig config(dbPath);
193 std::string bundleName = "com.ohos.config.GetDatabase";
194 config.SetBundleName(bundleName);
195 config.SetArea(1);
196 config.SetEncryptStatus(true);
197
198 RdbHelper::DeleteRdbStore(config);
199
200 int errCode = E_OK;
201
202 RdbHelperTestOpenCallback helper;
203 std::shared_ptr<RdbStore> rdbStore1 = RdbHelper::GetRdbStore(config, 1, helper, errCode);
204 EXPECT_EQ(errCode, E_OK);
205 ASSERT_NE(rdbStore1, nullptr);
206
207 config.SetEncryptStatus(false);
208 std::shared_ptr<RdbStore> rdbStore2 = RdbHelper::GetRdbStore(config, 1, helper, errCode);
209 EXPECT_EQ(errCode, E_OK);
210 ASSERT_NE(rdbStore2, nullptr);
211
212 EXPECT_NE(rdbStore1, rdbStore2);
213 }
214