1 // Copyright (C) 2016 and later: Unicode, Inc. and others.
2 // License & terms of use: http://www.unicode.org/copyright.html
3 /********************************************************************
4 * COPYRIGHT:
5 * Copyright (c) 1997-2009, International Business Machines Corporation and
6 * others. All Rights Reserved.
7 ********************************************************************/
8
9 #include "unicode/utypes.h"
10
11 #if !UCONFIG_NO_COLLATION
12
13 #include "unicode/coll.h"
14 #include "unicode/tblcoll.h"
15 #include "unicode/unistr.h"
16 #include "unicode/sortkey.h"
17 #include "mnkytst.h"
18 #include "sfwdchit.h"
19
20 #include <stdlib.h>
21 #include <stdio.h>
22 #include <time.h>
23
24 #ifndef MIN
25 #define MIN(x,y) ((x) < (y) ? (x) : (y))
26 #endif
27
28 #ifndef MAX
29 #define MAX(x,y) ((x) > (y) ? (x) : (y))
30 #endif
31
CollationMonkeyTest()32 CollationMonkeyTest::CollationMonkeyTest()
33 : source("-abcdefghijklmnopqrstuvwxyz#&^$@", ""),
34 myCollator(0)
35 {
36 UErrorCode status = U_ZERO_ERROR;
37 myCollator = Collator::createInstance("en_US", status);
38 }
39
~CollationMonkeyTest()40 CollationMonkeyTest::~CollationMonkeyTest()
41 {
42 delete myCollator;
43 }
44
45
46 void
report(UnicodeString & s,UnicodeString & t,int32_t result,int32_t revResult)47 CollationMonkeyTest::report(UnicodeString& s, UnicodeString& t, int32_t result, int32_t revResult)
48 {
49 if (revResult != -result)
50 {
51 UnicodeString msg;
52
53 msg += s;
54 msg += " and ";
55 msg += t;
56 msg += " round trip comparison failed";
57 msg += (UnicodeString) " (result " + result + ", reverse Result " + revResult + ")";
58
59 errln(msg);
60 }
61 }
62
63 int32_t
checkValue(int32_t value)64 CollationMonkeyTest::checkValue(int32_t value)
65 {
66 if (value < 0)
67 {
68 return -value;
69 }
70
71 return value;
72 }
73
TestCollationKey()74 void CollationMonkeyTest::TestCollationKey(/* char* par */)
75 {
76 if(source.length() == 0) {
77 errln(UNICODE_STRING("CollationMonkeyTest::TestCollationKey(): source is empty - ICU_DATA not set or data missing?", 92));
78 return;
79 }
80
81 srand( (unsigned)time( NULL ) );
82 int32_t s = checkValue(rand() % source.length());
83 int32_t t = checkValue(rand() % source.length());
84 int32_t slen = checkValue((rand() - source.length()) % source.length());
85 int32_t tlen = checkValue((rand() - source.length()) % source.length());
86 UnicodeString subs, subt;
87
88 source.extract(MIN(s, slen), MAX(s, slen), subs);
89 source.extract(MIN(t, tlen), MAX(t, tlen), subt);
90
91 CollationKey collationKey1, collationKey2;
92 UErrorCode status1 = U_ZERO_ERROR, status2= U_ZERO_ERROR;
93
94 myCollator->setStrength(Collator::TERTIARY);
95 myCollator->getCollationKey(subs, collationKey1, status1);
96 myCollator->getCollationKey(subt, collationKey2, status2);
97 int32_t result = collationKey1.compareTo(collationKey2); // Tertiary
98 int32_t revResult = collationKey2.compareTo(collationKey1); // Tertiary
99 report( subs, subt, result, revResult);
100
101 myCollator->setStrength(Collator::SECONDARY);
102 myCollator->getCollationKey(subs, collationKey1, status1);
103 myCollator->getCollationKey(subt, collationKey2, status2);
104 result = collationKey1.compareTo(collationKey2); // Secondary
105 revResult = collationKey2.compareTo(collationKey1); // Secondary
106 report( subs, subt, result, revResult);
107
108 myCollator->setStrength(Collator::PRIMARY);
109 myCollator->getCollationKey(subs, collationKey1, status1);
110 myCollator->getCollationKey(subt, collationKey2, status2);
111 result = collationKey1.compareTo(collationKey2); // Primary
112 revResult = collationKey2.compareTo(collationKey1); // Primary
113 report(subs, subt, result, revResult);
114
115 UnicodeString msg;
116 UnicodeString addOne(subs);
117 addOne += (UChar32)0xE000;
118
119 myCollator->getCollationKey(subs, collationKey1, status1);
120 myCollator->getCollationKey(addOne, collationKey2, status2);
121 result = collationKey1.compareTo(collationKey2);
122 if (result != -1)
123 {
124 msg += "CollationKey(";
125 msg += subs;
126 msg += ") .LT. CollationKey(";
127 msg += addOne;
128 msg += ") Failed.";
129 errln(msg);
130 }
131
132 msg.remove();
133 result = collationKey2.compareTo(collationKey1);
134 if (result != 1)
135 {
136 msg += "CollationKey(";
137 msg += addOne;
138 msg += ") .GT. CollationKey(";
139 msg += subs;
140 msg += ") Failed.";
141 errln(msg);
142 }
143 }
144
145 void
TestCompare()146 CollationMonkeyTest::TestCompare(/* char* par */)
147 {
148 if(source.length() == 0) {
149 errln(UNICODE_STRING("CollationMonkeyTest::TestCompare(): source is empty - ICU_DATA not set or data missing?", 87));
150 return;
151 }
152
153 /* Seed the random-number generator with current time so that
154 * the numbers will be different every time we run.
155 */
156 srand( (unsigned)time( NULL ) );
157 int32_t s = checkValue(rand() % source.length());
158 int32_t t = checkValue(rand() % source.length());
159 int32_t slen = checkValue((rand() - source.length()) % source.length());
160 int32_t tlen = checkValue((rand() - source.length()) % source.length());
161 UnicodeString subs, subt;
162
163 source.extract(MIN(s, slen), MAX(s, slen), subs);
164 source.extract(MIN(t, tlen), MAX(t, tlen), subt);
165
166 myCollator->setStrength(Collator::TERTIARY);
167 int32_t result = myCollator->compare(subs, subt); // Tertiary
168 int32_t revResult = myCollator->compare(subt, subs); // Tertiary
169 report(subs, subt, result, revResult);
170
171 myCollator->setStrength(Collator::SECONDARY);
172 result = myCollator->compare(subs, subt); // Secondary
173 revResult = myCollator->compare(subt, subs); // Secondary
174 report(subs, subt, result, revResult);
175
176 myCollator->setStrength(Collator::PRIMARY);
177 result = myCollator->compare(subs, subt); // Primary
178 revResult = myCollator->compare(subt, subs); // Primary
179 report(subs, subt, result, revResult);
180
181 UnicodeString msg;
182 UnicodeString addOne(subs);
183 addOne += (UChar32)0xE000;
184
185 result = myCollator->compare(subs, addOne);
186 if (result != -1)
187 {
188 msg += "Test : ";
189 msg += subs;
190 msg += " .LT. ";
191 msg += addOne;
192 msg += " Failed.";
193 errln(msg);
194 }
195
196 msg.remove();
197 result = myCollator->compare(addOne, subs);
198 if (result != 1)
199 {
200 msg += "Test : ";
201 msg += addOne;
202 msg += " .GT. ";
203 msg += subs;
204 msg += " Failed.";
205 errln(msg);
206 }
207 }
TestRules()208 void CollationMonkeyTest::TestRules(/* char* par */){
209 UChar testSourceCases[][10] = {
210 {0x0061, 0x0062, 0x007a, 0},
211 {0x0061, 0x0062, 0x007a, 0},
212 };
213
214 UChar testTargetCases[][10] = {
215 {0x0061, 0x0062, 0x00e4, 0},
216 {0x0061, 0x0062, 0x0061, 0x0308, 0},
217 };
218
219 int i=0;
220 logln("Demo Test 1 : Create a new table collation with rules \"& z < 0x00e4\"");
221 UErrorCode status = U_ZERO_ERROR;
222 Collator *col = Collator::createInstance("en_US", status);
223 const UnicodeString baseRules = ((RuleBasedCollator*)col)->getRules();
224 UnicodeString newRules(" & z < ");
225 newRules.append((UChar)0x00e4);
226 newRules.insert(0, baseRules);
227 RuleBasedCollator *myCollation = new RuleBasedCollator(newRules, status);
228 if (U_FAILURE(status)) {
229 errln( "Demo Test 1 Table Collation object creation failed.");
230 return;
231 }
232 for(i=0; i<2; i++){
233 doTest(myCollation, testSourceCases[i], testTargetCases[i], Collator::LESS);
234 }
235 delete myCollation;
236
237 logln("Demo Test 2 : Create a new table collation with rules \"& z < a 0x0308\"");
238 newRules.remove();
239 newRules.append(" & z < a");
240 newRules.append((UChar)0x0308);
241 newRules.insert(0, baseRules);
242 myCollation = new RuleBasedCollator(newRules, status);
243 if (U_FAILURE(status)) {
244 errln( "Demo Test 1 Table Collation object creation failed.");
245 return;
246 }
247 for(i=0; i<2; i++){
248 doTest(myCollation, testSourceCases[i], testTargetCases[i], Collator::LESS);
249 }
250 delete myCollation;
251 delete col;
252
253 }
254
runIndexedTest(int32_t index,UBool exec,const char * & name,char *)255 void CollationMonkeyTest::runIndexedTest( int32_t index, UBool exec, const char* &name, char* /*par*/ )
256 {
257 if (exec) logln("TestSuite CollationMonkeyTest: ");
258 if(myCollator) {
259 switch (index) {
260 case 0: name = "TestCompare"; if (exec) TestCompare(/* par */); break;
261 case 1: name = "TestCollationKey"; if (exec) TestCollationKey(/* par */); break;
262 case 2: name = "TestRules"; if (exec) TestRules(/* par */); break;
263 default: name = ""; break;
264 }
265 } else {
266 dataerrln("Class collator not instantiated");
267 name = "";
268 }
269 }
270
271 #endif /* #if !UCONFIG_NO_COLLATION */
272