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