Add test for native sorting in Solr.
[yaz-moved-to-github.git] / test / test_icu.c
index 2ad005c..de31121 100644 (file)
@@ -1,10 +1,10 @@
 /* This file is part of the YAZ toolkit.
- * Copyright (C) 1995-2010 Index Data
+ * Copyright (C) 1995-2012 Index Data
  * See the file LICENSE for details.
  */
 
 /* DO NOT EDIT THIS FILE IF YOUR EDITOR DOES NOT SUPPORT UTF-8 */
+
 
 #if HAVE_CONFIG_H
 #include "config.h"
@@ -21,6 +21,7 @@
 
 #if YAZ_HAVE_ICU
 #include <yaz/icu_I18N.h>
+#include <unicode/uclean.h>
 
 #if YAZ_POSIX_THREADS
 #include <pthread.h>
@@ -43,8 +44,8 @@ static int icu_termmap_cmp(const void *vp1, const void *vp2)
     struct icu_termmap *itmp2 = *(struct icu_termmap **) vp2;
 
     int cmp = 0;
-    
-    cmp = strcmp((const char *)itmp1->sort_key, 
+
+    cmp = strcmp((const char *)itmp1->sort_key,
                  (const char *)itmp2->sort_key);
     return cmp;
 }
@@ -70,12 +71,12 @@ static int test_icu_casemap(const char * locale, char action,
 
     /* perform case mapping */
     icu_utf16_casemap(dest16, src16, locale, action, &status);
-  
+
     /* converting to UTF8 */
     icu_utf16_to_utf8(dest8, dest16, &status);
 
     /* determine success */
-    if (dest8->utf8 
+    if (dest8->utf8
         && (dest8->utf8_len == strlen(chk8cstr))
         && !strcmp(chk8cstr, (const char *) dest8->utf8))
         success = 1;
@@ -88,18 +89,18 @@ static int test_icu_casemap(const char * locale, char action,
         yaz_log(YLOG_WARN, "test_icu_casemap failed");
         yaz_log(YLOG_LOG, "Original string:   '%s' (%d)",
                 src8cstr, src8cstr_len);
-        yaz_log(YLOG_LOG, "icu_casemap '%s:%c' '%s' (%d)", 
+        yaz_log(YLOG_LOG, "icu_casemap '%s:%c' '%s' (%d)",
                 locale, action, dest8->utf8, dest8->utf8_len);
         yaz_log(YLOG_LOG, "expected string:   '%s' (%d)",
                 chk8cstr, chk8cstr_len);
     }
-  
+
     /* clean the buffers */
     icu_buf_utf8_destroy(src8);
     icu_buf_utf8_destroy(dest8);
     icu_buf_utf16_destroy(src16);
     icu_buf_utf16_destroy(dest16);
-    
+
     return success;
 }
 
@@ -109,38 +110,38 @@ static void check_icu_casemap(void)
 
     /* successful tests */
     YAZ_CHECK(test_icu_casemap("en", 'l',
-                               "A ReD fOx hunTS sQUirriLs", 
+                               "A ReD fOx hunTS sQUirriLs",
                                "a red fox hunts squirrils"));
-    
+
     YAZ_CHECK(test_icu_casemap("en", 'u',
-                               "A ReD fOx hunTS sQUirriLs", 
+                               "A ReD fOx hunTS sQUirriLs",
                                "A RED FOX HUNTS SQUIRRILS"));
-    
+
     YAZ_CHECK(test_icu_casemap("en", 'f',
-                               "A ReD fOx hunTS sQUirriLs", 
+                               "A ReD fOx hunTS sQUirriLs",
                                "a red fox hunts squirrils"));
-    
+
     YAZ_CHECK(test_icu_casemap("en", 't',
-                               "A ReD fOx hunTS sQUirriLs", 
+                               "A ReD fOx hunTS sQUirriLs",
                                "A Red Fox Hunts Squirrils"));
-    
+
     /* Locale 'da' */
 
     /* success expected */
     YAZ_CHECK(test_icu_casemap("da", 'l',
-                               "åh ÆbLE, øs fLØde i Åen efter bLåBærGRødeN", 
+                               "åh ÆbLE, øs fLØde i Åen efter bLåBærGRødeN",
                                "åh æble, øs fløde i åen efter blåbærgrøden"));
 
     YAZ_CHECK(test_icu_casemap("da", 'u',
-                               "åh ÆbLE, øs fLØde i Åen efter bLåBærGRødeN", 
+                               "åh ÆbLE, øs fLØde i Åen efter bLåBærGRødeN",
                                "ÅH ÆBLE, ØS FLØDE I ÅEN EFTER BLÅBÆRGRØDEN"));
 
     YAZ_CHECK(test_icu_casemap("da", 'f',
-                               "åh ÆbLE, øs fLØde i Åen efter bLåBærGRødeN", 
+                               "åh ÆbLE, øs fLØde i Åen efter bLåBærGRødeN",
                                "åh æble, øs fløde i åen efter blåbærgrøden"));
 
     YAZ_CHECK(test_icu_casemap("da", 't',
-                               "åh ÆbLE, øs fLØde i Åen efter bLåBærGRødeN", 
+                               "åh ÆbLE, øs fLØde i Åen efter bLåBærGRødeN",
                                "Åh Æble, Øs Fløde I Åen Efter Blåbærgrøden"));
 
     /* Locale 'de' */
@@ -178,20 +179,20 @@ static int test_icu_sortmap(const char * locale, int src_list_len,
 
     struct icu_termmap * list[src_list_len];
 
-    UCollator *coll = ucol_open(locale, &status); 
+    UCollator *coll = ucol_open(locale, &status);
     icu_check_status(status);
 
     if (U_FAILURE(status))
         return 0;
 
     /* assigning display terms and sort keys using buf 8 and buf16 */
-    for (i = 0; i < src_list_len; i++) 
+    for (i = 0; i < src_list_len; i++)
     {
 
         list[i] = (struct icu_termmap *) malloc(sizeof(struct icu_termmap));
 
         /* copy display term */
-        strcpy(list[i]->disp_term, src_list[i]);    
+        strcpy(list[i]->disp_term, src_list[i]);
 
         /* transforming to UTF16 */
         icu_utf16_from_utf8_cstr(buf16, list[i]->disp_term, &status);
@@ -200,10 +201,10 @@ static int test_icu_sortmap(const char * locale, int src_list_len,
         /* computing sortkeys */
         icu_sortkey8_from_utf16(coll, buf8, buf16, &status);
         icu_check_status(status);
-    
+
         /* assigning sortkeys */
-        memcpy(list[i]->sort_key, buf8->utf8, buf8->utf8_len);    
-    } 
+        memcpy(list[i]->sort_key, buf8->utf8, buf8->utf8_len);
+    }
 
     /* do the sorting */
     qsort(list, src_list_len, sizeof(struct icu_termmap *), icu_termmap_cmp);
@@ -218,30 +219,30 @@ static int test_icu_sortmap(const char * locale, int src_list_len,
 
     if (!success)
     {
-        yaz_log(YLOG_LOG, "ERROR"); 
-        yaz_log(YLOG_LOG, "Input str:'%s':", locale); 
+        yaz_log(YLOG_LOG, "ERROR");
+        yaz_log(YLOG_LOG, "Input str:'%s':", locale);
         for (i = 0; i < src_list_len; i++) {
-            yaz_log(YLOG_LOG, "  '%s'", list[i]->disp_term); 
+            yaz_log(YLOG_LOG, "  '%s'", list[i]->disp_term);
         }
-        yaz_log(YLOG_LOG, "ICU sort: '%s':", locale); 
+        yaz_log(YLOG_LOG, "ICU sort: '%s':", locale);
         for (i = 0; i < src_list_len; i++) {
-            yaz_log(YLOG_LOG, " '%s'", list[i]->disp_term); 
+            yaz_log(YLOG_LOG, " '%s'", list[i]->disp_term);
         }
-        yaz_log(YLOG_LOG, "Expected: '%s':", locale); 
+        yaz_log(YLOG_LOG, "Expected: '%s':", locale);
         for (i = 0; i < src_list_len; i++) {
-            yaz_log(YLOG_LOG, " '%s'", chk_list[i]); 
+            yaz_log(YLOG_LOG, " '%s'", chk_list[i]);
         }
     }
-  
+
     for (i = 0; i < src_list_len; i++)
         free(list[i]);
-   
+
     ucol_close(coll);
 
     icu_buf_utf8_destroy(buf8);
     icu_buf_utf16_destroy(buf16);
 
-    return success;  
+    return success;
 }
 
 static void check_icu_sortmap(void)
@@ -255,7 +256,7 @@ static void check_icu_sortmap(void)
     YAZ_CHECK(test_icu_sortmap("en_CA", en_1_len, en_1_src, en_1_cck));
     YAZ_CHECK(test_icu_sortmap("en_GB", en_1_len, en_1_src, en_1_cck));
     YAZ_CHECK(test_icu_sortmap("en_US", en_1_len, en_1_src, en_1_cck));
-    
+
     /* successful tests */
     {
         size_t da_1_len = 6;
@@ -280,7 +281,7 @@ static int test_icu_normalizer(const char * rules8cstr,
                                const char * chk8cstr)
 {
     int success = 0;
-    
+
     UErrorCode status = U_ZERO_ERROR;
 
     struct icu_buf_utf16 * src16 = icu_buf_utf16_create(0);
@@ -289,7 +290,7 @@ static int test_icu_normalizer(const char * rules8cstr,
     struct icu_transform * transform
         = icu_transform_create(rules8cstr, 'f', 0, &status);
     icu_check_status(status);
-    
+
     icu_utf16_from_utf8_cstr(src16, src8cstr, &status);
     icu_check_status(status);
 
@@ -300,7 +301,7 @@ static int test_icu_normalizer(const char * rules8cstr,
     icu_check_status(status);
 
 
-    if (!strcmp((const char *) dest8->utf8, 
+    if (!strcmp((const char *) dest8->utf8,
                (const char *) chk8cstr))
         success = 1;
     else
@@ -326,7 +327,7 @@ static void check_icu_normalizer(void)
     YAZ_CHECK(test_icu_normalizer("[:Punctuation:] Any-Remove",
                                   "Don't shoot!",
                                   "Dont shoot"));
-    
+
     YAZ_CHECK(test_icu_normalizer("[:Control:] Any-Remove",
                                   "Don't\n shoot!",
                                   "Don't shoot!"));
@@ -338,7 +339,7 @@ static void check_icu_normalizer(void)
     YAZ_CHECK(test_icu_normalizer("Lower; [:^Letter:] Remove",
                                   "Don't shoot!",
                                   "dontshoot"));
-    
+
     YAZ_CHECK(test_icu_normalizer("[:^Number:] Remove",
                                   "Monday 15th of April",
                                   "15"));
@@ -374,7 +375,7 @@ static int test_icu_tokenizer(const char * locale, char action,
     YAZ_CHECK(tokenizer);
 
     /* attach text buffer to tokenizer */
-    icu_tokenizer_attach(tokenizer, src16, &status);    
+    icu_tokenizer_attach(tokenizer, src16, &status);
     icu_check_status(status);
 
     /* perform work on tokens */
@@ -399,25 +400,25 @@ static int test_icu_tokenizer(const char * locale, char action,
     icu_buf_utf16_destroy(src16);
     icu_buf_utf16_destroy(tkn16);
     icu_buf_utf8_destroy(tkn8);
-        
+
     return success;
 }
 
 static void check_icu_tokenizer(void)
 {
-    const char * en_str 
+    const char * en_str
         = "O Romeo, Romeo! wherefore art thou Romeo?";
-    
+
     YAZ_CHECK(test_icu_tokenizer("en", 's', en_str, 2));
     YAZ_CHECK(test_icu_tokenizer("en", 'l', en_str, 7));
     YAZ_CHECK(test_icu_tokenizer("en", 'w', en_str, 16));
     YAZ_CHECK(test_icu_tokenizer("en", 'c', en_str, 41));
 
     {
-        const char * da_str 
+        const char * da_str
             = "Blåbærtærte. Denne kage stammer fra Finland. "
             "Den er med blåbær, men alle sommerens forskellige bær kan bruges.";
-        
+
         YAZ_CHECK(test_icu_tokenizer("da", 's', da_str, 3));
         YAZ_CHECK(test_icu_tokenizer("dar", 'l', da_str, 17));
         YAZ_CHECK(test_icu_tokenizer("da", 'w', da_str, 37));
@@ -427,12 +428,12 @@ static void check_icu_tokenizer(void)
 
 static void check_icu_chain(void)
 {
-    const char * en_str 
+    const char * en_str
         = "O Romeo, Romeo! wherefore art thou\t Romeo?";
 
     UErrorCode status = U_ZERO_ERROR;
     struct icu_chain * chain = 0;
-    
+
     const char * xml_str = "<icu locale=\"en\">"
         "<transform rule=\"[:Control:] Any-Remove\"/>"
         "<tokenize rule=\"l\"/>"
@@ -441,7 +442,7 @@ static void check_icu_chain(void)
         "<casemap rule=\"l\"/>"
         "</icu>";
 
-    
+
     xmlDoc *doc = xmlParseMemory(xml_str, strlen(xml_str));
     xmlNode *xml_node = xmlDocGetRootElement(doc);
     YAZ_CHECK(xml_node);
@@ -450,6 +451,8 @@ static void check_icu_chain(void)
 
     xmlFreeDoc(doc);
     YAZ_CHECK(chain);
+    if (!chain)
+        return;
 
     YAZ_CHECK(icu_chain_assign_cstr(chain, en_str, &status));
 
@@ -485,7 +488,7 @@ static void check_bug_1140(void)
 {
     UErrorCode status = U_ZERO_ERROR;
     struct icu_chain * chain = 0;
-    
+
     const char * xml_str = "<icu locale=\"en\">"
 
         /* if the first rule is normalize instead. Then it works */
@@ -498,7 +501,7 @@ static void check_bug_1140(void)
         "<casemap rule=\"l\"/>"
         "</icu>";
 
-    
+
     xmlDoc *doc = xmlParseMemory(xml_str, strlen(xml_str));
     xmlNode *xml_node = xmlDocGetRootElement(doc);
     YAZ_CHECK(xml_node);
@@ -507,7 +510,9 @@ static void check_bug_1140(void)
 
     xmlFreeDoc(doc);
     YAZ_CHECK(chain);
-    
+    if (!chain)
+        return;
+
     YAZ_CHECK(icu_chain_assign_cstr(
                   chain,  "O Romeo, Romeo! wherefore art thou\t Romeo?",
                   &status));
@@ -520,7 +525,7 @@ static void check_bug_1140(void)
            icu_chain_token_norm(chain),
            icu_chain_token_display(chain)); */
     }
-    
+
 
     YAZ_CHECK_EQ(icu_chain_token_number(chain), 7);
 
@@ -551,7 +556,7 @@ static void check_chain_empty_token(void)
         "<tokenize rule=\"w\"/>"
         "<transform rule=\"[[:WhiteSpace:][:Punctuation:]] Remove\"/>"
         "</icu>";
-    
+
     xmlDoc *doc = xmlParseMemory(xml_str, strlen(xml_str));
     xmlNode *xml_node = xmlDocGetRootElement(doc);
     YAZ_CHECK(xml_node);
@@ -560,7 +565,7 @@ static void check_chain_empty_token(void)
 
     xmlFreeDoc(doc);
     YAZ_CHECK(chain);
-    
+
     YAZ_CHECK(icu_chain_assign_cstr(
                   chain,  "a string with 15 tokenss and 8 displays",
                   &status));
@@ -586,7 +591,7 @@ static void check_chain_empty_chain(void)
 
     const char * xml_str = "<icu locale=\"en\">"
         "</icu>";
-    
+
     const char * src8 = "some 5487 weired !¤%&(/& sTuFf";
     char * dest8 = 0;
 
@@ -598,7 +603,7 @@ static void check_chain_empty_chain(void)
 
     xmlFreeDoc(doc);
     YAZ_CHECK(chain);
-    
+
     YAZ_CHECK(icu_chain_assign_cstr(
                   chain,  src8,
                   &status));
@@ -616,7 +621,7 @@ static void check_chain_empty_chain(void)
 
     dest8 = (char *) icu_chain_token_norm(chain);
     YAZ_CHECK_EQ(strcmp(src8, dest8), 0);
-    
+
     icu_chain_destroy(chain);
 }
 
@@ -631,7 +636,7 @@ static void check_icu_iter1(void)
         "<tokenize rule=\"w\"/>"
         "<transform rule=\"[[:WhiteSpace:][:Punctuation:]] Remove\"/>"
         "</icu>";
-    
+
     xmlDoc *doc = xmlParseMemory(xml_str, strlen(xml_str));
     YAZ_CHECK(doc);
     if (!doc)
@@ -641,11 +646,11 @@ static void check_icu_iter1(void)
     if (!xml_node)
         return ;
 
-    chain = icu_chain_xml_config(xml_node, 0, &status);
+    chain = icu_chain_xml_config(xml_node, 1, &status);
 
     xmlFreeDoc(doc);
     YAZ_CHECK(chain);
-    
+
     iter = icu_iter_create(chain);
     icu_iter_first(iter, "a string with 15 tokens and 8 displays");
     YAZ_CHECK(iter);
@@ -663,7 +668,7 @@ static int test_iter(struct icu_chain *chain, const char *input,
                      const char *expected)
 {
     yaz_icu_iter_t iter = icu_iter_create(chain);
-    WRBUF result, second;
+    WRBUF result, second, sort_result;
     int success = 1;
 
     if (!iter)
@@ -678,15 +683,27 @@ static int test_iter(struct icu_chain *chain, const char *input,
         return 0;
     }
 
+    sort_result = wrbuf_alloc();
     result = wrbuf_alloc();
     icu_iter_first(iter, input);
     while (icu_iter_next(iter))
     {
+        const char *sort_str = icu_iter_get_sortkey(iter);
+        if (sort_str)
+        {
+            wrbuf_puts(sort_result, "[");
+            wrbuf_puts_escaped(sort_result, sort_str);
+            wrbuf_puts(sort_result, "]");
+        }
+        else
+        {
+            wrbuf_puts(sort_result, "[NULL]");
+        }
         wrbuf_puts(result, "[");
         wrbuf_puts(result, icu_iter_get_norm(iter));
         wrbuf_puts(result, "]");
     }
-
+    yaz_log(YLOG_LOG, "sortkey=%s", wrbuf_cstr(sort_result));
     second = wrbuf_alloc();
     icu_iter_first(iter, input);
     while (icu_iter_next(iter))
@@ -714,6 +731,7 @@ static int test_iter(struct icu_chain *chain, const char *input,
 
     wrbuf_destroy(result);
     wrbuf_destroy(second);
+    wrbuf_destroy(sort_result);
     return success;
 }
 
@@ -721,8 +739,8 @@ static void *iter_thread(void *p)
 {
     struct icu_chain *chain = (struct icu_chain *) p;
     int i;
-    
-    for (i = 0; i < 10000; i++)
+
+    for (i = 0; i < 1000; i++)
     {
         YAZ_CHECK(test_iter(chain, "Adobe Acrobat Reader, 1991-1999.",
                             "[adobe][acrobat][reader][1991][][1999][]"));
@@ -745,6 +763,7 @@ static void check_iter_threads(struct icu_chain *chain)
         pthread_join(t[i], 0);
 #endif
 }
+
 static void check_icu_iter2(void)
 {
     UErrorCode status = U_ZERO_ERROR;
@@ -769,24 +788,69 @@ static void check_icu_iter2(void)
     if (!xml_node)
         return ;
 
-    chain = icu_chain_xml_config(xml_node, 0, &status);
+    chain = icu_chain_xml_config(xml_node, 1, &status);
 
     xmlFreeDoc(doc);
     YAZ_CHECK(chain);
-    
+    if (!chain)
+        return;
+
     YAZ_CHECK(test_iter(chain, "Adobe Acrobat Reader, 1991-1999.",
                         "[adobe][acrobat][reader][1991][][1999][]"));
 
+    YAZ_CHECK(test_iter(chain, "Νόταρης, Γιάννης Σωτ",
+                        "[νόταρης][γιάννης][σωτ]"));
+
     check_iter_threads(chain);
 
     icu_chain_destroy(chain);
 }
 
+static void check_icu_iter3(void)
+{
+    UErrorCode status = U_ZERO_ERROR;
+    struct icu_chain * chain = 0;
+    xmlNode *xml_node;
+
+    const char * xml_str =
+        "<icu_chain id=\"sort\" locale=\"el\">\n"
+        "<transform rule=\"[:Control:] Any-Remove\"/>\n"
+        "<transform rule=\"[[:Control:][:WhiteSpace:][:Punctuation:]] Remove\"/>\n"
+        "<transform rule=\"NFD; [:Nonspacing Mark:] Remove; NFC\"/>\n"
+        "<casemap rule=\"l\"/>\n"
+        "<display/>\n"
+        "</icu_chain>\n";
+
+    xmlDoc *doc = xmlParseMemory(xml_str, strlen(xml_str));
+    YAZ_CHECK(doc);
+    if (!doc)
+        return;
+    xml_node = xmlDocGetRootElement(doc);
+    YAZ_CHECK(xml_node);
+    if (!xml_node)
+        return ;
+
+    chain = icu_chain_xml_config(xml_node, 1, &status);
+
+    xmlFreeDoc(doc);
+    YAZ_CHECK(chain);
+    if (!chain)
+        return;
+
+    YAZ_CHECK(test_iter(chain, "Adobe Acrobat Reader, 1991-1999.",
+                        "[adobeacrobatreader19911999]"));
+
+    YAZ_CHECK(test_iter(chain, "Νόταρης, Γιάννης Σωτ",
+                        "[νοταρηςγιαννηςσωτ]"));
+
+    icu_chain_destroy(chain);
+}
+
 #endif /* YAZ_HAVE_ICU */
 
 int main(int argc, char **argv)
 {
-    YAZ_CHECK_INIT(argc, argv); 
+    YAZ_CHECK_INIT(argc, argv);
     YAZ_CHECK_LOG();
 
 #if YAZ_HAVE_ICU
@@ -800,16 +864,18 @@ int main(int argc, char **argv)
     check_chain_empty_chain();
     check_icu_iter1();
     check_icu_iter2();
-    
+    check_icu_iter3();
+
     check_bug_1140();
 
+    u_cleanup();
 #else /* YAZ_HAVE_ICU */
 
     yaz_log(YLOG_LOG, "ICU unit tests omitted");
     YAZ_CHECK(0 == 0);
 
 #endif /* YAZ_HAVE_ICU */
-   
+
     YAZ_CHECK_TERM;
 }