1 /* 2 * Copyright (c) 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 #ifndef CLOUD_DB_SYNC_UTILS_TEST_H 17 #define CLOUD_DB_SYNC_UTILS_TEST_H 18 19 #include <gtest/gtest.h> 20 #include <iostream> 21 #include "cloud/cloud_storage_utils.h" 22 #include "cloud/cloud_db_constant.h" 23 #include "distributeddb_data_generate_unit_test.h" 24 #include "distributeddb_tools_unit_test.h" 25 #include "process_system_api_adapter_impl.h" 26 #include "relational_store_instance.h" 27 #include "relational_store_manager.h" 28 #include "runtime_config.h" 29 #include "sqlite_relational_store.h" 30 #include "sqlite_relational_utils.h" 31 #include "store_observer.h" 32 #include "time_helper.h" 33 #include "virtual_asset_loader.h" 34 #include "virtual_cloud_data_translate.h" 35 #include "virtual_cloud_db.h" 36 37 namespace DistributedDB { 38 using namespace DistributedDBUnitTest; 39 using CloudSyncStatusCallback = std::function<void(const std::map<std::string, SyncProcess> &onProcess)>; 40 41 class CloudDBSyncUtilsTest { 42 public: 43 static void SetStorePath(const std::string &path); 44 45 static void InitSyncUtils(const std::vector<Field> &cloudField, RelationalStoreObserverUnitTest *&observer, 46 std::shared_ptr<VirtualCloudDb> &virtualCloudDb, std::shared_ptr<VirtualAssetLoader> &virtualAssetLoader, 47 RelationalStoreDelegate *&delegate); 48 49 static void CreateUserDBAndTable(sqlite3 *&db, std::string sql); 50 51 static void InsertCloudTableRecord(int64_t begin, int64_t count, int64_t photoSize, bool assetIsNull, 52 std::shared_ptr<VirtualCloudDb> &virtualCloudDb); 53 54 static void UpdateCloudTableRecord(int64_t begin, int64_t count, int64_t photoSize, bool assetIsNull, 55 std::shared_ptr<VirtualCloudDb> &virtualCloudDb); 56 57 static void DeleteCloudTableRecordByGid(int64_t begin, int64_t count, 58 std::shared_ptr<VirtualCloudDb> &virtualCloudDb); 59 60 static void DeleteUserTableRecord(sqlite3 *&db, int64_t begin, int64_t count); 61 62 static void GetCallback(SyncProcess &syncProcess, CloudSyncStatusCallback &callback, 63 std::vector<SyncProcess> &expectProcess); 64 65 static void CheckCloudTotalCount(std::vector<int64_t> expectCounts, 66 const std::shared_ptr<VirtualCloudDb> &virtualCloudDb); 67 68 static void WaitForSyncFinish(SyncProcess &syncProcess, const int64_t &waitTime); 69 70 static void callSync(const std::vector<std::string> &tableNames, SyncMode mode, DBStatus dbStatus, 71 RelationalStoreDelegate *&delegate); 72 73 static void CloseDb(RelationalStoreObserverUnitTest *&observer, 74 std::shared_ptr<VirtualCloudDb> &virtualCloudDb, RelationalStoreDelegate *&delegate); 75 76 static int QueryCountCallback(void *data, int count, char **colValue, char **colName); 77 78 static void CheckDownloadResult(sqlite3 *&db, std::vector<int64_t> expectCounts, const std::string &keyStr); 79 80 static void CheckLocalRecordNum(sqlite3 *&db, const std::string &tableName, int count); 81 82 static void GetCloudDbSchema(const std::string &tableName, const std::vector<Field> &cloudField, 83 DataBaseSchema &dataBaseSchema); 84 85 static void InitStoreProp(const std::string &storePath, const std::string &appId, const std::string &userId, 86 const std::string &storeId, RelationalDBProperties &properties); 87 88 static void CheckCount(sqlite3 *db, const std::string &sql, int64_t count); 89 90 static void GetHashKey(const std::string &tableName, const std::string &condition, sqlite3 *db, 91 std::vector<std::vector<uint8_t>> &hashKey); 92 }; 93 } 94 95 #endif // CLOUD_DB_SYNC_UTILS_TEST_H 96