1 /* 2 * Copyright (c) 2021-2022 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 DATA_STORAGE_GTEST_H 17 #define DATA_STORAGE_GTEST_H 18 19 #include <iostream> 20 #include <securec.h> 21 22 #include "abs_shared_result_set.h" 23 #include "accesstoken_kit.h" 24 #include "datashare_helper.h" 25 #include "datashare_predicates.h" 26 #include "gtest/gtest.h" 27 #include "iservice_registry.h" 28 #include "nativetoken_kit.h" 29 #include "system_ability_definition.h" 30 #include "token_setproc.h" 31 #include "values_bucket.h" 32 33 namespace OHOS { 34 namespace Telephony { 35 class DataStorageGtest : public testing::Test { 36 public: 37 static void SetUpTestCase(); 38 static void TearDownTestCase(); 39 void SetUp(); 40 void TearDown(); 41 42 using RequestFuncType = int (DataStorageGtest::*) 43 (const std::shared_ptr<DataShare::DataShareHelper> &helper) const; 44 std::map<char, RequestFuncType> requestFuncMap_; 45 46 std::shared_ptr<DataShare::DataShareHelper> CreateDataShareHelper( 47 int32_t systemAbilityId, std::string &uri) const; 48 std::shared_ptr<DataShare::DataShareHelper> CreateSimHelper(); 49 std::shared_ptr<DataShare::DataShareHelper> CreateSmsHelper(); 50 std::shared_ptr<DataShare::DataShareHelper> CreatePdpProfileHelper(); 51 std::shared_ptr<DataShare::DataShareHelper> CreateOpKeyHelper(); 52 std::shared_ptr<DataShare::DataShareHelper> CreateGlobalParamsHelper(); 53 std::shared_ptr<DataShare::DataShareHelper> CreateOpKeyVersionCustHelper(); 54 std::shared_ptr<DataShare::DataShareHelper> CreateOpKeyVersionChipHelper(); 55 56 int SimInsert(const std::shared_ptr<DataShare::DataShareHelper> &helper) const; 57 int SimUpdate(const std::shared_ptr<DataShare::DataShareHelper> &helper) const; 58 int SimSelect(const std::shared_ptr<DataShare::DataShareHelper> &helper) const; 59 int SimDelete(const std::shared_ptr<DataShare::DataShareHelper> &helper) const; 60 int SmsInsert(const std::shared_ptr<DataShare::DataShareHelper> &helper) const; 61 int SmsBatchInsert(const std::shared_ptr<DataShare::DataShareHelper> &helper) const; 62 int SmsUpdate(const std::shared_ptr<DataShare::DataShareHelper> &helper) const; 63 int SmsSelect(const std::shared_ptr<DataShare::DataShareHelper> &helper) const; 64 int SmsDelete(const std::shared_ptr<DataShare::DataShareHelper> &helper) const; 65 int PdpProfileInsert(const std::shared_ptr<DataShare::DataShareHelper> &helper) const; 66 int PdpProfileUpdate(const std::shared_ptr<DataShare::DataShareHelper> &helper) const; 67 int PdpProfileUpdate2(const std::shared_ptr<DataShare::DataShareHelper> &helper) const; 68 int PdpProfileUpdate3(const std::shared_ptr<DataShare::DataShareHelper> &helper) const; 69 int PdpProfileUpdate4(const std::shared_ptr<DataShare::DataShareHelper> &helper) const; 70 int PdpProfileSelect(const std::shared_ptr<DataShare::DataShareHelper> &helper) const; 71 int PdpProfileSelect2(const std::shared_ptr<DataShare::DataShareHelper> &helper) const; 72 int PdpProfileDelete(const std::shared_ptr<DataShare::DataShareHelper> &helper) const; 73 int OpKeyInsert(const std::shared_ptr<DataShare::DataShareHelper> &helper) const; 74 int OpKeyUpdate(const std::shared_ptr<DataShare::DataShareHelper> &helper) const; 75 int OpKeySelect(const std::shared_ptr<DataShare::DataShareHelper> &helper) const; 76 int OpKeyDelete(const std::shared_ptr<DataShare::DataShareHelper> &helper) const; 77 int GlobalParamsNumMatchInsert(const std::shared_ptr<DataShare::DataShareHelper> &helper) const; 78 int GlobalParamsNumMatchUpdate(const std::shared_ptr<DataShare::DataShareHelper> &helper) const; 79 int GlobalParamsNumMatchSelect(const std::shared_ptr<DataShare::DataShareHelper> &helper) const; 80 int GlobalParamsNumMatchDelete(const std::shared_ptr<DataShare::DataShareHelper> &helper) const; 81 int GlobalEccInsert(const std::shared_ptr<DataShare::DataShareHelper> &helper) const; 82 int GlobalEccUpdate(const std::shared_ptr<DataShare::DataShareHelper> &helper) const; 83 int GlobalEccSelect(const std::shared_ptr<DataShare::DataShareHelper> &helper) const; 84 int GlobalEccDelete(const std::shared_ptr<DataShare::DataShareHelper> &helper) const; 85 int OpKeyVersionQuery(const std::shared_ptr<DataShare::DataShareHelper> &helper, Uri uri) const; 86 int PseBaseStationInsert(const std::shared_ptr<DataShare::DataShareHelper> &helper) const; 87 int PseBaseStationUpdate(const std::shared_ptr<DataShare::DataShareHelper> &helper) const; 88 bool PseBaseStationQuery(const std::shared_ptr<DataShare::DataShareHelper> &helper) const; 89 90 private: 91 std::shared_ptr<DataShare::DataShareHelper> simDataHelper = nullptr; 92 std::shared_ptr<DataShare::DataShareHelper> smsDataHelper = nullptr; 93 std::shared_ptr<DataShare::DataShareHelper> pdpProfileDataHelper = nullptr; 94 std::shared_ptr<DataShare::DataShareHelper> opKeyDataHelper = nullptr; 95 std::shared_ptr<DataShare::DataShareHelper> globalParamsDataHelper = nullptr; 96 std::shared_ptr<DataShare::DataShareHelper> opKeyVersionCustHelper = nullptr; 97 std::shared_ptr<DataShare::DataShareHelper> opKeyVersionChipHelper = nullptr; 98 }; 99 } // namespace Telephony 100 } // namespace OHOS 101 #endif // STATE_REGISTER_GTEST_H