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 "ecmascript/containers/containers_private.h"
17 #include "ecmascript/ecma_string.h"
18 #include "ecmascript/ecma_vm.h"
19 #include "ecmascript/global_env.h"
20 #include "ecmascript/js_function.h"
21 #include "ecmascript/js_handle.h"
22 #include "ecmascript/js_iterator.h"
23 #include "ecmascript/js_api/js_api_hashset.h"
24 #include "ecmascript/js_api/js_api_hashset_iterator.h"
25 #include "ecmascript/js_object-inl.h"
26 #include "ecmascript/js_tagged_value.h"
27 #include "ecmascript/object_factory.h"
28 #include "ecmascript/tests/ecma_test_common.h"
29
30 using namespace panda;
31 using namespace panda::ecmascript;
32
33 namespace panda::test {
34 class JSAPIHashSetTest : public BaseTestWithScope<false> {
35 protected:
CreateHashSet()36 JSAPIHashSet *CreateHashSet()
37 {
38 return EcmaContainerCommon::CreateHashSet(thread);
39 }
40
Update(JSHandle<JSAPIHashSet> & hashSet,JSMutableHandle<JSTaggedValue> value,std::string & myValue,uint32_t numbers)41 void Update(JSHandle<JSAPIHashSet>& hashSet, JSMutableHandle<JSTaggedValue> value, std::string& myValue,
42 uint32_t numbers)
43 {
44 ObjectFactory *factory = thread->GetEcmaVM()->GetFactory();
45 for (uint32_t i = 0; i < numbers; i++) {
46 std::string iValue = myValue + std::to_string(i);
47 value.Update(factory->NewFromStdString(iValue).GetTaggedValue());
48 JSAPIHashSet::Add(thread, hashSet, value);
49 }
50 EXPECT_EQ(hashSet->GetSize(), numbers);
51 }
52 };
53
HWTEST_F_L0(JSAPIHashSetTest,HashSetCreate)54 HWTEST_F_L0(JSAPIHashSetTest, HashSetCreate)
55 {
56 JSAPIHashSet *set = CreateHashSet();
57 EXPECT_TRUE(set != nullptr);
58 }
59
HWTEST_F_L0(JSAPIHashSetTest,HashSetAddAndHas)60 HWTEST_F_L0(JSAPIHashSetTest, HashSetAddAndHas)
61 {
62 constexpr uint32_t NODE_NUMBERS = 8;
63 JSMutableHandle<JSTaggedValue> value(thread, JSTaggedValue::Undefined());
64 JSHandle<JSAPIHashSet> hashSet(thread, CreateHashSet());
65 std::string myValue("myvalue");
66 Update(hashSet, value, myValue, NODE_NUMBERS);
67
68 ObjectFactory *factory = thread->GetEcmaVM()->GetFactory();
69 for (uint32_t i = 0; i < NODE_NUMBERS; i++) {
70 std::string iValue = myValue + std::to_string(i);
71 value.Update(factory->NewFromStdString(iValue).GetTaggedValue());
72
73 // test has
74 JSTaggedValue bHas = hashSet->Has(thread, value.GetTaggedValue());
75 EXPECT_EQ(bHas, JSTaggedValue::True());
76 }
77
78 // test add exception
79 JSHandle<JSTaggedValue> hole(thread, JSTaggedValue::Hole());
80 JSAPIHashSet::Add(thread, hashSet, hole);
81 EXPECT_EXCEPTION();
82
83 // test Has exception
84 JSTaggedValue exceptionHas = hashSet->Has(thread, JSTaggedValue::Hole());
85 EXPECT_EQ(exceptionHas, JSTaggedValue::Exception());
86 EXPECT_EXCEPTION();
87 }
88
HWTEST_F_L0(JSAPIHashSetTest,HashSetRemoveAndHas)89 HWTEST_F_L0(JSAPIHashSetTest, HashSetRemoveAndHas)
90 {
91 constexpr uint32_t NODE_NUMBERS = 8;
92 ObjectFactory *factory = thread->GetEcmaVM()->GetFactory();
93 // test JSAPIHashSet
94 JSHandle<JSAPIHashSet> hashSet(thread, CreateHashSet());
95 std::string myValue("myvalue");
96 JSMutableHandle<JSTaggedValue> value(thread, JSTaggedValue::Undefined());
97 Update(hashSet, value, myValue, NODE_NUMBERS);
98
99 for (uint32_t i = 0; i < NODE_NUMBERS / 2; i++) {
100 std::string iValue = myValue + std::to_string(i);
101 value.Update(factory->NewFromStdString(iValue).GetTaggedValue());
102 [[maybe_unused]] JSTaggedValue rvalue = JSAPIHashSet::Remove(thread, hashSet, value.GetTaggedValue());
103 }
104 EXPECT_EQ(hashSet->GetSize(), NODE_NUMBERS / 2);
105
106 for (uint32_t i = 0; i < NODE_NUMBERS / 2; i++) {
107 std::string iValue = myValue + std::to_string(i);
108 value.Update(factory->NewFromStdString(iValue).GetTaggedValue());
109
110 // test has
111 JSTaggedValue has = hashSet->Has(thread, value.GetTaggedValue());
112 EXPECT_EQ(has, JSTaggedValue::False());
113 }
114
115 for (uint32_t i = NODE_NUMBERS / 2; i < NODE_NUMBERS; i++) {
116 std::string iValue = myValue + std::to_string(i);
117 value.Update(factory->NewFromStdString(iValue).GetTaggedValue());
118
119 // test has
120 JSTaggedValue has = hashSet->Has(thread, value.GetTaggedValue());
121 EXPECT_EQ(has, JSTaggedValue::True());
122 }
123
124 // test Remove exception
125 JSTaggedValue exceptionValue = JSAPIHashSet::Remove(thread, hashSet, JSTaggedValue::Hole());
126 EXPECT_EQ(exceptionValue, JSTaggedValue::Exception());
127 EXPECT_EXCEPTION();
128 }
129
HWTEST_F_L0(JSAPIHashSetTest,JSAPIHashSetRemoveRBTreeTest)130 HWTEST_F_L0(JSAPIHashSetTest, JSAPIHashSetRemoveRBTreeTest)
131 {
132 std::vector<int> hashCollisionVector = {1224, 1285, 1463, 4307, 5135, 5903, 6603, 6780, 8416, 9401, 9740};
133 uint32_t NODE_NUMBERS = static_cast<uint32_t>(hashCollisionVector.size());
134 constexpr uint32_t REMOVE_NUMBERS = 4;
135 JSHandle<JSAPIHashSet> hashSet(thread, CreateHashSet());
136
137 // test Remove empty
138 JSTaggedValue emptyValue = JSAPIHashSet::Remove(thread, hashSet, JSTaggedValue(0));
139 EXPECT_EQ(emptyValue, JSTaggedValue::False());
140
141 JSMutableHandle<JSTaggedValue> value(thread, JSTaggedValue::Undefined());
142
143 for (size_t i = 0; i < hashCollisionVector.size(); i++) {
144 value.Update(JSTaggedValue(hashCollisionVector[i]));
145 JSAPIHashSet::Add(thread, hashSet, value);
146 }
147
148 // test Remove non-existent value
149 JSTaggedValue nonExistentValue = JSAPIHashSet::Remove(thread, hashSet, JSTaggedValue(0));
150 EXPECT_EQ(nonExistentValue, JSTaggedValue::False());
151
152 // test Remove RBTree
153 for (uint32_t i = 0; i < REMOVE_NUMBERS; i++) {
154 value.Update(JSTaggedValue(hashCollisionVector[i]));
155 JSAPIHashSet::Remove(thread, hashSet, value.GetTaggedValue());
156 }
157 EXPECT_EQ(hashSet->GetSize(), NODE_NUMBERS - REMOVE_NUMBERS);
158
159 for (uint32_t i = 0; i < REMOVE_NUMBERS; i++) {
160 value.Update(JSTaggedValue(hashCollisionVector[i]));
161 JSTaggedValue has = hashSet->Has(thread, value.GetTaggedValue());
162 EXPECT_EQ(has, JSTaggedValue::False());
163 }
164
165 for (uint32_t i = REMOVE_NUMBERS; i < NODE_NUMBERS; i++) {
166 value.Update(JSTaggedValue(hashCollisionVector[i]));
167 JSTaggedValue has = hashSet->Has(thread, value.GetTaggedValue());
168 EXPECT_EQ(has, JSTaggedValue::True());
169 }
170 }
171
HWTEST_F_L0(JSAPIHashSetTest,HashSetClearAddHas)172 HWTEST_F_L0(JSAPIHashSetTest, HashSetClearAddHas)
173 {
174 constexpr uint32_t NODE_NUMBERS = 8;
175 JSMutableHandle<JSTaggedValue> value(thread, JSTaggedValue::Undefined());
176 std::string myValue("myvalue");
177 JSHandle<JSAPIHashSet> hashSet(thread, CreateHashSet());
178 Update(hashSet, value, myValue, NODE_NUMBERS);
179
180 hashSet->Clear(thread);
181 JSTaggedValue isEmpty = hashSet->IsEmpty();
182 EXPECT_EQ(isEmpty, JSTaggedValue::True());
183 }
184
HWTEST_F_L0(JSAPIHashSetTest,JSAPIHashSetIterator)185 HWTEST_F_L0(JSAPIHashSetTest, JSAPIHashSetIterator)
186 {
187 constexpr uint32_t NODE_NUMBERS = 8;
188 ObjectFactory *factory = thread->GetEcmaVM()->GetFactory();
189 JSHandle<JSAPIHashSet> hashSet(thread, CreateHashSet());
190
191 JSMutableHandle<JSTaggedValue> value(thread, JSTaggedValue::Undefined());
192 for (uint32_t i = 0; i < NODE_NUMBERS; i++) {
193 value.Update(JSTaggedValue(i));
194 JSAPIHashSet::Add(thread, hashSet, value);
195 }
196
197 // test value
198 JSHandle<JSTaggedValue> valueIter(factory->NewJSAPIHashSetIterator(hashSet, IterationKind::VALUE));
199 JSMutableHandle<JSTaggedValue> valueIterResult(thread, JSTaggedValue::Undefined());
200 for (uint32_t i = 0; i < NODE_NUMBERS; i++) {
201 valueIterResult.Update(JSIterator::IteratorStep(thread, valueIter).GetTaggedValue());
202 JSHandle<JSTaggedValue> tmpIterValue = JSIterator::IteratorValue(thread, valueIterResult);
203 JSTaggedValue iterValueFlag = hashSet->Has(thread, tmpIterValue.GetTaggedValue());
204 EXPECT_EQ(JSTaggedValue::True(), iterValueFlag);
205 }
206 // test end
207 valueIterResult.Update(JSIterator::IteratorStep(thread, valueIter).GetTaggedValue());
208 EXPECT_EQ(JSTaggedValue::False(), valueIterResult.GetTaggedValue());
209
210 // test key and value
211 JSHandle<JSTaggedValue> indexKey(thread, JSTaggedValue(0));
212 JSHandle<JSTaggedValue> elementKey(thread, JSTaggedValue(1));
213 JSHandle<JSTaggedValue> iter(factory->NewJSAPIHashSetIterator(hashSet, IterationKind::KEY_AND_VALUE));
214 JSMutableHandle<JSTaggedValue> iterResult(thread, JSTaggedValue::Undefined());
215 JSMutableHandle<JSTaggedValue> result(thread, JSTaggedValue::Undefined());
216 for (uint32_t i = 0; i < NODE_NUMBERS; i++) {
217 iterResult.Update(JSIterator::IteratorStep(thread, iter).GetTaggedValue());
218 result.Update(JSIterator::IteratorValue(thread, iterResult).GetTaggedValue());
219 EXPECT_EQ(JSTaggedValue(i), JSObject::GetProperty(thread, result, indexKey).GetValue().GetTaggedValue());
220 JSHandle<JSTaggedValue> tmpValue = JSObject::GetProperty(thread, result, elementKey).GetValue();
221 JSTaggedValue iterValueFlag = hashSet->Has(thread, tmpValue.GetTaggedValue());
222 EXPECT_EQ(JSTaggedValue::True(), iterValueFlag);
223 }
224 }
225
HWTEST_F_L0(JSAPIHashSetTest,JSAPIHashSetIteratorRBTreeTest)226 HWTEST_F_L0(JSAPIHashSetTest, JSAPIHashSetIteratorRBTreeTest)
227 {
228 constexpr uint32_t NODE_NUMBERS = 11;
229 ObjectFactory *factory = thread->GetEcmaVM()->GetFactory();
230 JSHandle<JSAPIHashSet> hashSet(thread, CreateHashSet());
231 JSMutableHandle<JSTaggedValue> value(thread, JSTaggedValue::Undefined());
232 JSHandle<JSTaggedValue> valueStr = thread->GlobalConstants()->GetHandledValueString();
233 std::vector<int> hashCollisionVector = {1224, 1285, 1463, 4307, 5135, 5903, 6603, 6780, 8416, 9401, 9740};
234
235 for (size_t i = 0; i < hashCollisionVector.size(); i++) {
236 value.Update(JSTaggedValue(hashCollisionVector[i]));
237 JSAPIHashSet::Add(thread, hashSet, value);
238 }
239
240 JSHandle<JSAPIHashSetIterator> hashsetIterator = factory->NewJSAPIHashSetIterator(hashSet, IterationKind::VALUE);
241 for (uint32_t i = 0; i < NODE_NUMBERS; i++) {
242 auto ecmaRuntimeCallInfo = TestHelper::CreateEcmaRuntimeCallInfo(thread, JSTaggedValue::Undefined(), 4);
243 ecmaRuntimeCallInfo->SetFunction(JSTaggedValue::Undefined());
244 ecmaRuntimeCallInfo->SetThis(hashsetIterator.GetTaggedValue());
245
246 [[maybe_unused]] auto prev = TestHelper::SetupFrame(thread, ecmaRuntimeCallInfo);
247 JSTaggedValue result = JSAPIHashSetIterator::Next(ecmaRuntimeCallInfo);
248 TestHelper::TearDownFrame(thread, prev);
249
250 JSHandle<JSObject> resultObj(thread, result);
251 if (i <= NODE_NUMBERS - 1U) {
252 EXPECT_TRUE(JSObject::GetProperty(thread, resultObj, valueStr).GetValue()->IsInt());
253 }
254 }
255 }
256 } // namespace panda::test
257