test/test_sortspec.c

   1 /* This file is part of the YAZ toolkit.
   2  * Copyright (C) Index Data
   3  * See the file LICENSE for details.
   4  */
   5 #if HAVE_CONFIG_H
   6 #include <config.h>
   7 #endif
   8
   9 #include <string.h>
  10 #include <yaz/wrbuf.h>
  11 #include <yaz/sortspec.h>
  12 #include <yaz/log.h>
  13 #include <yaz/test.h>
  14
  15 static int cql(const char *arg, const char *expected_result)
  16 {
  17     ODR odr = odr_createmem(ODR_ENCODE);
  18     Z_SortKeySpecList *sort_spec = yaz_sort_spec(odr, arg);
  19     int ret = 0;
  20
  21     if (!sort_spec)
  22     {
  23         yaz_log(YLOG_WARN, "yaz_sort_spec : parse error: %s", arg);
  24     }
  25     else
  26     {
  27         WRBUF w = wrbuf_alloc();
  28         int r = yaz_sort_spec_to_cql(sort_spec, w);
  29
  30         if (!expected_result && r)
  31             ret = 1;
  32         else if (expected_result && r == 0)
  33         {
  34             if (strcmp(wrbuf_cstr(w), expected_result) == 0)
  35                 ret = 1;
  36             else
  37             {
  38                 yaz_log(YLOG_WARN, "sort: diff: %s", arg);
  39                 yaz_log(YLOG_WARN, " expected %s", expected_result);
  40                 yaz_log(YLOG_WARN, " got      %s", wrbuf_cstr(w));
  41             }
  42         }
  43         else if (r)
  44         {
  45             yaz_log(YLOG_WARN, "sort: diff %s", arg);
  46             yaz_log(YLOG_WARN, " expected %s", expected_result);
  47             yaz_log(YLOG_WARN, " got error %d", r);
  48         }
  49         else if (r == 0)
  50         {
  51             yaz_log(YLOG_WARN, "sort: diff %s", arg);
  52             yaz_log(YLOG_WARN, " expected error");
  53             yaz_log(YLOG_WARN, " got %s", wrbuf_cstr(w));
  54         }
  55         wrbuf_destroy(w);
  56     }
  57     odr_destroy(odr);
  58     return ret;
  59 }
  60
  61 static int type7(const char *arg, const char *expected_result)
  62 {
  63     ODR odr = odr_createmem(ODR_ENCODE);
  64     Z_SortKeySpecList *sort_spec = yaz_sort_spec(odr, arg);
  65     int ret = 0;
  66
  67     if (!sort_spec)
  68     {
  69         yaz_log(YLOG_WARN, "yaz_sort_spec : parse error: %s", arg);
  70     }
  71     else
  72     {
  73         WRBUF w = wrbuf_alloc();
  74         int r;
  75
  76         wrbuf_puts(w, "q");
  77         r = yaz_sort_spec_to_type7(sort_spec, w);
  78
  79         if (!expected_result && r)
  80             ret = 1;
  81         else if (expected_result && r == 0)
  82         {
  83             if (strcmp(wrbuf_cstr(w), expected_result) == 0)
  84                 ret = 1;
  85             else
  86             {
  87                 yaz_log(YLOG_WARN, "sort: diff: %s", arg);
  88                 yaz_log(YLOG_WARN, " expected %s", expected_result);
  89                 yaz_log(YLOG_WARN, " got      %s", wrbuf_cstr(w));
  90             }
  91         }
  92         else if (r)
  93         {
  94             yaz_log(YLOG_WARN, "sort: diff %s", arg);
  95             yaz_log(YLOG_WARN, " expected %s", expected_result);
  96             yaz_log(YLOG_WARN, " got error %d", r);
  97         }
  98         else if (r == 0)
  99         {
 100             yaz_log(YLOG_WARN, "sort: diff %s", arg);
 101             yaz_log(YLOG_WARN, " expected error");
 102             yaz_log(YLOG_WARN, " got %s", wrbuf_cstr(w));
 103         }
 104         wrbuf_destroy(w);
 105     }
 106     odr_destroy(odr);
 107     return ret;
 108 }
 109
 110 static int strategy_sortkeys(const char *arg, const char *expected_result, int (*strategy) (Z_SortKeySpecList *, WRBUF))
 111 {
 112     ODR odr = odr_createmem(ODR_ENCODE);
 113     Z_SortKeySpecList *sort_spec = yaz_sort_spec(odr, arg);
 114     int ret = 0;
 115
 116     if (!sort_spec)
 117     {
 118         yaz_log(YLOG_WARN, "yaz_sort_spec : parse error: %s", arg);
 119     }
 120     else
 121     {
 122         WRBUF w = wrbuf_alloc();
 123         int r = (strategy)(sort_spec, w);
 124
 125         if (!expected_result && r)
 126             ret = 1;
 127         else if (expected_result && r == 0)
 128         {
 129             if (strcmp(wrbuf_cstr(w), expected_result) == 0)
 130                 ret = 1;
 131             else
 132             {
 133                 yaz_log(YLOG_WARN, "sort: diff: %s", arg);
 134                 yaz_log(YLOG_WARN, " expected %s", expected_result);
 135                 yaz_log(YLOG_WARN, " got      %s", wrbuf_cstr(w));
 136             }
 137         }
 138         else if (r)
 139         {
 140             yaz_log(YLOG_WARN, "sort: diff %s", arg);
 141             yaz_log(YLOG_WARN, " expected %s", expected_result);
 142             yaz_log(YLOG_WARN, " got error %d", r);
 143         }
 144         else if (r == 0)
 145         {
 146             yaz_log(YLOG_WARN, "sort: diff %s", arg);
 147             yaz_log(YLOG_WARN, " expected error");
 148             yaz_log(YLOG_WARN, " got %s", wrbuf_cstr(w));
 149         }
 150         wrbuf_destroy(w);
 151     }
 152     odr_destroy(odr);
 153     return ret;
 154 }
 155
 156 static int srw_sortkeys(const char *arg, const char *expected_result) {
 157     return strategy_sortkeys(arg, expected_result, yaz_sort_spec_to_srw_sortkeys);
 158 }
 159
 160 static int solr_sortkeys(const char *arg, const char *expected_result) {
 161     return strategy_sortkeys(arg, expected_result, yaz_sort_spec_to_solr_sortkeys);
 162 }
 163
 164
 165
 166 static int check_sortkeys_to_sort_spec(const char *arg,
 167                                            const char *expected_result, int (*sort_strategy)(const char *, WRBUF))
 168 {
 169     WRBUF w = wrbuf_alloc();
 170     int ret = 0;
 171     int r = sort_strategy(arg, w);
 172
 173     if (!expected_result && r)
 174         ret = 1;
 175     else if (expected_result && r == 0)
 176     {
 177         if (strcmp(wrbuf_cstr(w), expected_result) == 0)
 178             ret = 1;
 179         else
 180         {
 181             yaz_log(YLOG_WARN, "sort: diff: %s", arg);
 182             yaz_log(YLOG_WARN, " expected %s", expected_result);
 183             yaz_log(YLOG_WARN, " got      %s", wrbuf_cstr(w));
 184         }
 185     }
 186     else if (r)
 187     {
 188         yaz_log(YLOG_WARN, "sort: diff %s", arg);
 189         yaz_log(YLOG_WARN, " expected %s", expected_result);
 190         yaz_log(YLOG_WARN, " got error %d", r);
 191     }
 192     else if (r == 0)
 193     {
 194         yaz_log(YLOG_WARN, "sort: diff %s", arg);
 195         yaz_log(YLOG_WARN, " expected error");
 196         yaz_log(YLOG_WARN, " got %s", wrbuf_cstr(w));
 197     }
 198     wrbuf_destroy(w);
 199     return ret;
 200 }
 201
 202 static int check_srw_sortkeys_to_sort_spec(const char *arg, const char *expected_result) {
 203     return check_sortkeys_to_sort_spec(arg, expected_result, yaz_srw_sortkeys_to_sort_spec);
 204 }
 205
 206 static int check_solr_sortkeys_to_sort_spec(const char *arg, const char *expected_result) {
 207     return check_sortkeys_to_sort_spec(arg, expected_result, yaz_solr_sortkeys_to_sort_spec);
 208 }
 209
 210 static void tst(void)
 211 {
 212     YAZ_CHECK(cql("title a",
 213                   " SORTBY title/ascending/ignoreCase"));
 214     YAZ_CHECK(cql("title a date ds",
 215                   " SORTBY title/ascending/ignoreCase"
 216                   " date/descending/respectCase"));
 217     YAZ_CHECK(cql("1=4,2=3 a", 0));
 218     YAZ_CHECK(cql("date a=1900",
 219                   " SORTBY date/ascending/ignoreCase/missingValue=1900"));
 220
 221     YAZ_CHECK(type7("title a",
 222                   "@or q @attr 1=title @attr 7=1 0"));
 223     YAZ_CHECK(type7("title a date ds",
 224                     "@or @or q @attr 1=title @attr 7=1 0"
 225                     " @attr 1=date @attr 7=2 1"));
 226     YAZ_CHECK(type7("1=4,2=3 a",
 227                   "@or q @attr 1=4 @attr 2=3 @attr 7=1 0"));
 228     YAZ_CHECK(type7("date a=1900",
 229                   "@or q @attr 1=date @attr 7=1 0"));
 230
 231     YAZ_CHECK(srw_sortkeys("title a",
 232                            "title,,1,0,highValue"));
 233     YAZ_CHECK(srw_sortkeys("title a date ds",
 234                            "title,,1,0,highValue "
 235                            "date,,0,1,highValue"));
 236     YAZ_CHECK(srw_sortkeys("1=4,2=3 a", 0));
 237     YAZ_CHECK(srw_sortkeys("date a=1900",
 238                            "date,,1,0,1900"));
 239     YAZ_CHECK(check_srw_sortkeys_to_sort_spec(
 240                   "date,,1,0,1900",
 241                   "date ai=1900"));
 242
 243     YAZ_CHECK(solr_sortkeys("title a",
 244                            "title asc"));
 245     YAZ_CHECK(solr_sortkeys("title a date ds",
 246                            "title asc"
 247                             ",date desc"));
 248     YAZ_CHECK(solr_sortkeys("1=4,2=3 a", 0));
 249
 250     YAZ_CHECK(check_solr_sortkeys_to_sort_spec(
 251                   "date asc",
 252                   "date ai"));
 253
 254 }
 255
 256 int main(int argc, char **argv)
 257 {
 258     YAZ_CHECK_INIT(argc, argv);
 259     YAZ_CHECK_LOG();
 260     tst();
 261     YAZ_CHECK_TERM;
 262 }
 263 /*
 264  * Local variables:
 265  * c-basic-offset: 4
 266  * c-file-style: "Stroustrup"
 267  * indent-tabs-mode: nil
 268  * End:
 269  * vim: shiftwidth=4 tabstop=8 expandtab
 270  */
 271