1 // Copyright (C) 2016 and later: Unicode, Inc. and others. 2 // License & terms of use: http://www.unicode.org/copyright.html 3 /* 4 ******************************************************************************* 5 * Copyright (C) 2015, International Business Machines Corporation and * 6 * others. All Rights Reserved. * 7 ******************************************************************************* 8 * 9 * File UNIFIEDCACHETEST.CPP 10 * 11 ******************************************************************************** 12 */ 13 #include "cstring.h" 14 #include "intltest.h" 15 #include "unifiedcache.h" 16 #include "unicode/datefmt.h" 17 18 class UCTItem : public SharedObject { 19 public: 20 char *value; 21 UCTItem(const char *x) : value(NULL) { 22 value = uprv_strdup(x); 23 } 24 virtual ~UCTItem() { 25 uprv_free(value); 26 } 27 }; 28 29 class UCTItem2 : public SharedObject { 30 }; 31 32 U_NAMESPACE_BEGIN 33 34 template<> U_EXPORT 35 const UCTItem *LocaleCacheKey<UCTItem>::createObject( 36 const void *context, UErrorCode &status) const { 37 const UnifiedCache *cacheContext = (const UnifiedCache *) context; 38 if (uprv_strcmp(fLoc.getName(), "zh") == 0) { 39 status = U_MISSING_RESOURCE_ERROR; 40 return NULL; 41 } 42 if (uprv_strcmp(fLoc.getLanguage(), fLoc.getName()) != 0) { 43 const UCTItem *item = NULL; 44 if (cacheContext == NULL) { 45 UnifiedCache::getByLocale(fLoc.getLanguage(), item, status); 46 } else { 47 cacheContext->get(LocaleCacheKey<UCTItem>(fLoc.getLanguage()), item, status); 48 } 49 if (U_FAILURE(status)) { 50 return NULL; 51 } 52 return item; 53 } 54 UCTItem *result = new UCTItem(fLoc.getName()); 55 result->addRef(); 56 return result; 57 } 58 59 template<> U_EXPORT 60 const UCTItem2 *LocaleCacheKey<UCTItem2>::createObject( 61 const void * /*unused*/, UErrorCode & /*status*/) const { 62 return NULL; 63 } 64 65 U_NAMESPACE_END 66 67 68 class UnifiedCacheTest : public IntlTest { 69 public: 70 UnifiedCacheTest() { 71 } 72 void runIndexedTest(int32_t index, UBool exec, const char *&name, char *par=0); 73 private: 74 void TestEvictionPolicy(); 75 void TestBounded(); 76 void TestBasic(); 77 void TestError(); 78 void TestHashEquals(); 79 void TestEvictionUnderStress(); 80 }; 81 82 void UnifiedCacheTest::runIndexedTest(int32_t index, UBool exec, const char* &name, char* /*par*/) { 83 TESTCASE_AUTO_BEGIN; 84 TESTCASE_AUTO(TestEvictionPolicy); 85 TESTCASE_AUTO(TestBounded); 86 TESTCASE_AUTO(TestBasic); 87 TESTCASE_AUTO(TestError); 88 TESTCASE_AUTO(TestHashEquals); 89 TESTCASE_AUTO(TestEvictionUnderStress); 90 TESTCASE_AUTO_END; 91 } 92 93 void UnifiedCacheTest::TestEvictionUnderStress() { 94 #if !UCONFIG_NO_FORMATTING 95 int32_t localeCount; 96 const Locale *locales = DateFormat::getAvailableLocales(localeCount); 97 UErrorCode status = U_ZERO_ERROR; 98 const UnifiedCache *cache = UnifiedCache::getInstance(status); 99 int64_t evictedCountBefore = cache->autoEvictedCount(); 100 for (int32_t i = 0; i < localeCount; ++i) { 101 LocalPointer<DateFormat> ptr(DateFormat::createInstanceForSkeleton("yMd", locales[i], status)); 102 } 103 int64_t evictedCountAfter = cache->autoEvictedCount(); 104 if (evictedCountBefore == evictedCountAfter) { 105 dataerrln("%s:%d Items should have been evicted from cache", 106 __FILE__, __LINE__); 107 } 108 #endif /* #if !UCONFIG_NO_FORMATTING */ 109 } 110 111 void UnifiedCacheTest::TestEvictionPolicy() { 112 UErrorCode status = U_ZERO_ERROR; 113 114 // We have to call this first or else calling the UnifiedCache 115 // ctor will fail. This is by design to deter clients from using the 116 // cache API incorrectly by creating their own cache instances. 117 UnifiedCache::getInstance(status); 118 119 // We create our own local UnifiedCache instance to ensure we have 120 // complete control over it. Real clients should never ever create 121 // their own cache! 122 UnifiedCache cache(status); 123 assertSuccess("", status); 124 125 // Don't allow unused entries to exeed more than 100% of in use entries. 126 cache.setEvictionPolicy(0, 100, status); 127 128 static const char *locales[] = { 129 "1", "2", "3", "4", "5", "6", "7", "8", "9", "10", 130 "11", "12", "13", "14", "15", "16", "17", "18", "19", "20"}; 131 132 const UCTItem *usedReferences[] = {NULL, NULL, NULL, NULL, NULL}; 133 const UCTItem *unusedReference = NULL; 134 135 // Add 5 in-use entries 136 for (int32_t i = 0; i < UPRV_LENGTHOF(usedReferences); i++) { 137 cache.get( 138 LocaleCacheKey<UCTItem>(locales[i]), 139 &cache, 140 usedReferences[i], 141 status); 142 } 143 144 // Add 10 not in use entries. 145 for (int32_t i = 0; i < 10; ++i) { 146 cache.get( 147 LocaleCacheKey<UCTItem>( 148 locales[i + UPRV_LENGTHOF(usedReferences)]), 149 &cache, 150 unusedReference, 151 status); 152 } 153 unusedReference->removeRef(); 154 155 // unused count not to exeed in use count 156 assertEquals("", UPRV_LENGTHOF(usedReferences), cache.unusedCount()); 157 assertEquals("", 2*UPRV_LENGTHOF(usedReferences), cache.keyCount()); 158 159 // Free up those used entries. 160 for (int32_t i = 0; i < UPRV_LENGTHOF(usedReferences); i++) { 161 usedReferences[i]->removeRef(); 162 } 163 164 // This should free up all cache items 165 assertEquals("", 0, cache.keyCount()); 166 167 assertSuccess("", status); 168 } 169 170 171 172 void UnifiedCacheTest::TestBounded() { 173 UErrorCode status = U_ZERO_ERROR; 174 175 // We have to call this first or else calling the UnifiedCache 176 // ctor will fail. This is by design to deter clients from using the 177 // cache API incorrectly by creating their own cache instances. 178 UnifiedCache::getInstance(status); 179 180 // We create our own local UnifiedCache instance to ensure we have 181 // complete control over it. Real clients should never ever create 182 // their own cache! 183 UnifiedCache cache(status); 184 assertSuccess("", status); 185 186 // Maximum unused count is 3. 187 cache.setEvictionPolicy(3, 0, status); 188 189 // Our cache will hold up to 3 unused key-value pairs 190 // We test the following invariants: 191 // 1. unusedCount <= 3 192 // 2. cache->get(X) always returns the same reference as long as caller 193 // already holds references to that same object. 194 195 // We first add 5 key-value pairs with two distinct values, "en" and "fr" 196 // keeping all those references. 197 198 const UCTItem *en = NULL; 199 const UCTItem *enGb = NULL; 200 const UCTItem *enUs = NULL; 201 const UCTItem *fr = NULL; 202 const UCTItem *frFr = NULL; 203 cache.get(LocaleCacheKey<UCTItem>("en_US"), &cache, enUs, status); 204 cache.get(LocaleCacheKey<UCTItem>("en"), &cache, en, status); 205 assertEquals("", 1, cache.unusedCount()); 206 cache.get(LocaleCacheKey<UCTItem>("en_GB"), &cache, enGb, status); 207 cache.get(LocaleCacheKey<UCTItem>("fr_FR"), &cache, frFr, status); 208 cache.get(LocaleCacheKey<UCTItem>("fr"), &cache, fr, status); 209 210 // Client holds two unique references, "en" and "fr" the other three 211 // entries are eligible for eviction. 212 assertEquals("", 3, cache.unusedCount()); 213 assertEquals("", 5, cache.keyCount()); 214 215 // Exercise cache more but don't hold the references except for 216 // the last one. At the end of this, we will hold references to one 217 // additional distinct value, so we will have references to 3 distinct 218 // values. 219 const UCTItem *throwAway = NULL; 220 cache.get(LocaleCacheKey<UCTItem>("zn_AA"), &cache, throwAway, status); 221 cache.get(LocaleCacheKey<UCTItem>("sr_AA"), &cache, throwAway, status); 222 cache.get(LocaleCacheKey<UCTItem>("de_AU"), &cache, throwAway, status); 223 224 const UCTItem *deAu(throwAway); 225 deAu->addRef(); 226 227 // Client holds three unique references, "en", "fr", "de" although we 228 // could have a total of 8 entries in the cache maxUnusedCount == 3 229 // so we have only 6 entries. 230 assertEquals("", 3, cache.unusedCount()); 231 assertEquals("", 6, cache.keyCount()); 232 233 // For all the references we have, cache must continue to return 234 // those same references (#2) 235 236 cache.get(LocaleCacheKey<UCTItem>("en"), &cache, throwAway, status); 237 if (throwAway != en) { 238 errln("Expected en to resolve to the same object."); 239 } 240 cache.get(LocaleCacheKey<UCTItem>("en_US"), &cache, throwAway, status); 241 if (throwAway != enUs) { 242 errln("Expected enUs to resolve to the same object."); 243 } 244 cache.get(LocaleCacheKey<UCTItem>("en_GB"), &cache, throwAway, status); 245 if (throwAway != enGb) { 246 errln("Expected enGb to resolve to the same object."); 247 } 248 cache.get(LocaleCacheKey<UCTItem>("fr_FR"), &cache, throwAway, status); 249 if (throwAway != frFr) { 250 errln("Expected frFr to resolve to the same object."); 251 } 252 cache.get(LocaleCacheKey<UCTItem>("fr_FR"), &cache, throwAway, status); 253 cache.get(LocaleCacheKey<UCTItem>("fr"), &cache, throwAway, status); 254 if (throwAway != fr) { 255 errln("Expected fr to resolve to the same object."); 256 } 257 cache.get(LocaleCacheKey<UCTItem>("de_AU"), &cache, throwAway, status); 258 if (throwAway != deAu) { 259 errln("Expected deAu to resolve to the same object."); 260 } 261 262 assertEquals("", 3, cache.unusedCount()); 263 assertEquals("", 6, cache.keyCount()); 264 265 // Now we hold a references to two more distinct values. Cache size 266 // should grow to 8. 267 const UCTItem *es = NULL; 268 const UCTItem *ru = NULL; 269 cache.get(LocaleCacheKey<UCTItem>("es"), &cache, es, status); 270 cache.get(LocaleCacheKey<UCTItem>("ru"), &cache, ru, status); 271 assertEquals("", 3, cache.unusedCount()); 272 assertEquals("", 8, cache.keyCount()); 273 274 // Now release all the references we hold except for 275 // es, ru, and en 276 SharedObject::clearPtr(enGb); 277 SharedObject::clearPtr(enUs); 278 SharedObject::clearPtr(fr); 279 SharedObject::clearPtr(frFr); 280 SharedObject::clearPtr(deAu); 281 SharedObject::clearPtr(es); 282 SharedObject::clearPtr(ru); 283 SharedObject::clearPtr(en); 284 SharedObject::clearPtr(throwAway); 285 286 // Size of cache should magically drop to 3. 287 assertEquals("", 3, cache.unusedCount()); 288 assertEquals("", 3, cache.keyCount()); 289 290 // Be sure nothing happens setting the eviction policy in the middle of 291 // a run. 292 cache.setEvictionPolicy(3, 0, status); 293 assertSuccess("", status); 294 295 } 296 297 void UnifiedCacheTest::TestBasic() { 298 UErrorCode status = U_ZERO_ERROR; 299 const UnifiedCache *cache = UnifiedCache::getInstance(status); 300 assertSuccess("", status); 301 cache->flush(); 302 int32_t baseCount = cache->keyCount(); 303 const UCTItem *en = NULL; 304 const UCTItem *enGb = NULL; 305 const UCTItem *enGb2 = NULL; 306 const UCTItem *enUs = NULL; 307 const UCTItem *fr = NULL; 308 const UCTItem *frFr = NULL; 309 cache->get(LocaleCacheKey<UCTItem>("en"), en, status); 310 cache->get(LocaleCacheKey<UCTItem>("en_US"), enUs, status); 311 cache->get(LocaleCacheKey<UCTItem>("en_GB"), enGb, status); 312 cache->get(LocaleCacheKey<UCTItem>("fr_FR"), frFr, status); 313 cache->get(LocaleCacheKey<UCTItem>("fr"), fr, status); 314 cache->get(LocaleCacheKey<UCTItem>("en_GB"), enGb2, status); 315 SharedObject::clearPtr(enGb2); 316 if (enGb != enUs) { 317 errln("Expected en_GB and en_US to resolve to same object."); 318 } 319 if (fr != frFr) { 320 errln("Expected fr and fr_FR to resolve to same object."); 321 } 322 if (enGb == fr) { 323 errln("Expected en_GB and fr to return different objects."); 324 } 325 assertSuccess("", status); 326 // en_US, en_GB, en share one object; fr_FR and fr don't share. 327 // 5 keys in all. 328 assertEquals("", baseCount + 5, cache->keyCount()); 329 SharedObject::clearPtr(enGb); 330 cache->flush(); 331 332 // Only 2 unique values in the cache. flushing trims cache down 333 // to this minimum size. 334 assertEquals("", baseCount + 2, cache->keyCount()); 335 SharedObject::clearPtr(enUs); 336 SharedObject::clearPtr(en); 337 cache->flush(); 338 // With en_GB and en_US and en cleared there are no more hard references to 339 // the "en" object, so it gets flushed and the keys that refer to it 340 // get removed from the cache. Now we have just one unique value, fr, in 341 // the cache 342 assertEquals("", baseCount + 1, cache->keyCount()); 343 SharedObject::clearPtr(fr); 344 cache->flush(); 345 assertEquals("", baseCount + 1, cache->keyCount()); 346 SharedObject::clearPtr(frFr); 347 cache->flush(); 348 assertEquals("", baseCount + 0, cache->keyCount()); 349 assertSuccess("", status); 350 } 351 352 void UnifiedCacheTest::TestError() { 353 UErrorCode status = U_ZERO_ERROR; 354 const UnifiedCache *cache = UnifiedCache::getInstance(status); 355 assertSuccess("", status); 356 cache->flush(); 357 int32_t baseCount = cache->keyCount(); 358 const UCTItem *zh = NULL; 359 const UCTItem *zhTw = NULL; 360 const UCTItem *zhHk = NULL; 361 362 status = U_ZERO_ERROR; 363 cache->get(LocaleCacheKey<UCTItem>("zh"), zh, status); 364 if (status != U_MISSING_RESOURCE_ERROR) { 365 errln("Expected U_MISSING_RESOURCE_ERROR"); 366 } 367 status = U_ZERO_ERROR; 368 cache->get(LocaleCacheKey<UCTItem>("zh_TW"), zhTw, status); 369 if (status != U_MISSING_RESOURCE_ERROR) { 370 errln("Expected U_MISSING_RESOURCE_ERROR"); 371 } 372 status = U_ZERO_ERROR; 373 cache->get(LocaleCacheKey<UCTItem>("zh_HK"), zhHk, status); 374 if (status != U_MISSING_RESOURCE_ERROR) { 375 errln("Expected U_MISSING_RESOURCE_ERROR"); 376 } 377 // 3 keys in cache zh, zhTW, zhHk all pointing to error placeholders 378 assertEquals("", baseCount + 3, cache->keyCount()); 379 cache->flush(); 380 // error placeholders have no hard references so they always get flushed. 381 assertEquals("", baseCount + 0, cache->keyCount()); 382 } 383 384 void UnifiedCacheTest::TestHashEquals() { 385 LocaleCacheKey<UCTItem> key1("en_US"); 386 LocaleCacheKey<UCTItem> key2("en_US"); 387 LocaleCacheKey<UCTItem> diffKey1("en_UT"); 388 LocaleCacheKey<UCTItem2> diffKey2("en_US"); 389 assertTrue("", key1.hashCode() == key2.hashCode()); 390 assertTrue("", key1.hashCode() != diffKey1.hashCode()); 391 assertTrue("", key1.hashCode() != diffKey2.hashCode()); 392 assertTrue("", diffKey1.hashCode() != diffKey2.hashCode()); 393 assertTrue("", key1 == key2); 394 assertTrue("", key1 != diffKey1); 395 assertTrue("", key1 != diffKey2); 396 assertTrue("", diffKey1 != diffKey2); 397 } 398 399 extern IntlTest *createUnifiedCacheTest() { 400 return new UnifiedCacheTest(); 401 } 402