1 /*
2 * Copyright (c) 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 #include <gtest/gtest.h>
17
18 #include <vector>
19
20 #include "abs_rdb_predicates.h"
21 #include "datashare_value_object.h"
22 #include "datashare_values_bucket.h"
23 #include "rdb_data_ability_utils.h"
24 #include "refbase.h"
25 #include "values_bucket.h"
26
27 using namespace testing::ext;
28 using namespace OHOS::NativeRdb;
29 using namespace OHOS::RdbDataAbilityAdapter;
30 using namespace OHOS::DataShare;
31
32 class DataAbilityUtilsTest : public testing::Test {
33 public:
34 static void SetUpTestCase(void);
35 static void TearDownTestCase(void);
36 void SetUp();
37 void TearDown();
38 };
39
SetUpTestCase(void)40 void DataAbilityUtilsTest::SetUpTestCase(void)
41 {
42 }
TearDownTestCase(void)43 void DataAbilityUtilsTest::TearDownTestCase(void)
44 {
45 }
SetUp(void)46 void DataAbilityUtilsTest::SetUp(void)
47 {
48 }
TearDown(void)49 void DataAbilityUtilsTest::TearDown(void)
50 {
51 }
52
53 /* *
54 * @tc.name: DataAbilityUtilsTest_001
55 * @tc.desc: test ToDataShareValuesBucket()
56 * @tc.type: FUNC
57 * @tc.require:
58 */
59 HWTEST_F(DataAbilityUtilsTest, DataAbilityUtilsTest_001, TestSize.Level1)
60 {
61 ValuesBucket valuesBucket;
62 valuesBucket.PutString("name", "Tom");
63 valuesBucket.PutInt("age", 10);
64 valuesBucket.PutDouble("weight", 50.5);
65 valuesBucket.PutBlob("data", { 1, 2, 3 });
66
67 auto dataShareValuesBucket = RdbDataAbilityUtils::ToDataShareValuesBucket(valuesBucket);
68
69 bool isEmpty = dataShareValuesBucket.IsEmpty();
70 EXPECT_EQ(isEmpty, false);
71
72 auto iter = dataShareValuesBucket.valuesMap.find("name");
73 if (iter != dataShareValuesBucket.valuesMap.end()) {
74 std::string strValue = std::get<std::string>(iter->second);
75 EXPECT_EQ(strValue, "Tom");
76 } else {
77 EXPECT_EQ(true, false);
78 }
79
80 if (iter != dataShareValuesBucket.valuesMap.end()) {
81 iter = dataShareValuesBucket.valuesMap.find("age");
82 int64_t intValue = std::get<int64_t>(iter->second);
83 EXPECT_EQ(intValue, 10);
84 } else {
85 EXPECT_EQ(true, false);
86 }
87
88 if (iter != dataShareValuesBucket.valuesMap.end()) {
89 iter = dataShareValuesBucket.valuesMap.find("weight");
90 double doubleValue = std::get<double>(iter->second);
91 EXPECT_EQ(doubleValue, 50.5);
92 } else {
93 EXPECT_EQ(true, false);
94 }
95
96 if (iter != dataShareValuesBucket.valuesMap.end()) {
97 iter = dataShareValuesBucket.valuesMap.find("data");
98 std::vector<uint8_t> uint8Vec = std::get<std::vector<uint8_t>>(iter->second);
99 EXPECT_EQ(uint8Vec[0], 1);
100 } else {
101 EXPECT_EQ(true, false);
102 }
103 }
104
105 /* *
106 * @tc.name: DataAbilityUtilsTest_002
107 * @tc.desc: test ToDataSharePredicates()
108 * @tc.type: FUNC
109 * @tc.require:
110 */
111 HWTEST_F(DataAbilityUtilsTest, DataAbilityUtilsTest_002, TestSize.Level1)
112 {
113 std::string order = "age";
114 auto *predicates = new DataAbilityPredicates();
115 predicates->EqualTo("name", "Tom")->OrderByDesc("id")->Limit(2);
116 predicates->SetOrder(order);
117 predicates->Distinct();
118
119 std::string whereClause = "name = ? ";
120 auto dataSharePredicates = RdbDataAbilityUtils::ToDataSharePredicates(*predicates);
121 std::string dataShareWhereClause = dataSharePredicates.GetWhereClause();
122 EXPECT_EQ(dataShareWhereClause, whereClause);
123
124 std::string dataShareOrder = dataSharePredicates.GetOrder();
125 EXPECT_EQ(dataShareOrder, order);
126 }