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 
54     int SimInsert(const std::shared_ptr<DataShare::DataShareHelper> &helper) const;
55     int SimUpdate(const std::shared_ptr<DataShare::DataShareHelper> &helper) const;
56     int SimSelect(const std::shared_ptr<DataShare::DataShareHelper> &helper) const;
57     int SimDelete(const std::shared_ptr<DataShare::DataShareHelper> &helper) const;
58     int SmsInsert(const std::shared_ptr<DataShare::DataShareHelper> &helper) const;
59     int SmsBatchInsert(const std::shared_ptr<DataShare::DataShareHelper> &helper) const;
60     int SmsUpdate(const std::shared_ptr<DataShare::DataShareHelper> &helper) const;
61     int SmsSelect(const std::shared_ptr<DataShare::DataShareHelper> &helper) const;
62     int SmsDelete(const std::shared_ptr<DataShare::DataShareHelper> &helper) const;
63     int PdpProfileInsert(const std::shared_ptr<DataShare::DataShareHelper> &helper) const;
64     int PdpProfileUpdate(const std::shared_ptr<DataShare::DataShareHelper> &helper) const;
65     int PdpProfileUpdate2(const std::shared_ptr<DataShare::DataShareHelper> &helper) const;
66     int PdpProfileUpdate3(const std::shared_ptr<DataShare::DataShareHelper> &helper) const;
67     int PdpProfileUpdate4(const std::shared_ptr<DataShare::DataShareHelper> &helper) const;
68     int PdpProfileSelect(const std::shared_ptr<DataShare::DataShareHelper> &helper) const;
69     int PdpProfileSelect2(const std::shared_ptr<DataShare::DataShareHelper> &helper) const;
70     int PdpProfileDelete(const std::shared_ptr<DataShare::DataShareHelper> &helper) const;
71     int OpKeyInsert(const std::shared_ptr<DataShare::DataShareHelper> &helper) const;
72     int OpKeyUpdate(const std::shared_ptr<DataShare::DataShareHelper> &helper) const;
73     int OpKeySelect(const std::shared_ptr<DataShare::DataShareHelper> &helper) const;
74     int OpKeyDelete(const std::shared_ptr<DataShare::DataShareHelper> &helper) const;
75     int GlobalParamsNumMatchInsert(const std::shared_ptr<DataShare::DataShareHelper> &helper) const;
76     int GlobalParamsNumMatchUpdate(const std::shared_ptr<DataShare::DataShareHelper> &helper) const;
77     int GlobalParamsNumMatchSelect(const std::shared_ptr<DataShare::DataShareHelper> &helper) const;
78     int GlobalParamsNumMatchDelete(const std::shared_ptr<DataShare::DataShareHelper> &helper) const;
79     int GlobalEccInsert(const std::shared_ptr<DataShare::DataShareHelper> &helper) const;
80     int GlobalEccUpdate(const std::shared_ptr<DataShare::DataShareHelper> &helper) const;
81     int GlobalEccSelect(const std::shared_ptr<DataShare::DataShareHelper> &helper) const;
82     int GlobalEccDelete(const std::shared_ptr<DataShare::DataShareHelper> &helper) const;
83 
84 private:
85     std::shared_ptr<DataShare::DataShareHelper> simDataHelper = nullptr;
86     std::shared_ptr<DataShare::DataShareHelper> smsDataHelper = nullptr;
87     std::shared_ptr<DataShare::DataShareHelper> pdpProfileDataHelper = nullptr;
88     std::shared_ptr<DataShare::DataShareHelper> opKeyDataHelper = nullptr;
89     std::shared_ptr<DataShare::DataShareHelper> globalParamsDataHelper = nullptr;
90 };
91 } // namespace Telephony
92 } // namespace OHOS
93 #endif // STATE_REGISTER_GTEST_H