test/test_rpn2cql.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 <stdlib.h>
  10 #include <stdio.h>
  11
  12 #include <yaz/test.h>
  13 #include <yaz/log.h>
  14 #include <yaz/rpn2cql.h>
  15 #include <yaz/wrbuf.h>
  16 #include <yaz/pquery.h>
  17
  18 static int compare2(cql_transform_t ct, const char *pqf, const char *cql,
  19                     int expected_error)
  20 {
  21     int ret = 0;
  22     ODR odr = odr_createmem(ODR_ENCODE);
  23     WRBUF w = wrbuf_alloc();
  24     Z_RPNQuery *q = p_query_rpn(odr, pqf);
  25
  26     if (q)
  27     {
  28         int r = cql_transform_rpn2cql_wrbuf(ct, w, q);
  29
  30         if (r != 0)
  31         {
  32             const char *addinfo = 0;
  33             int err = cql_transform_error(ct, &addinfo);
  34             /* transform error */
  35             yaz_log(YLOG_LOG, "%s -> Error %d", pqf, r);
  36             if (err == 0)
  37                 ;
  38             else if (err == expected_error)
  39             {
  40                 if (addinfo && cql && !strcmp(addinfo, cql))
  41                     ret = 1;
  42                 else if (!addinfo && !cql)
  43                     ret = 1;
  44             }
  45         }
  46         else if (r == 0)
  47         {
  48             yaz_log(YLOG_LOG, "%s -> %s", pqf, wrbuf_cstr(w));
  49             if (!expected_error)
  50                 ret = 1;
  51             else if (cql && !strcmp(wrbuf_cstr(w), cql))
  52             {
  53                 ret = 1;
  54             }
  55             else
  56             {
  57                 yaz_log(YLOG_WARN, " expected: %s", cql ? cql : "null");
  58                 yaz_log(YLOG_WARN, " got:      %s", wrbuf_cstr(w));
  59             }
  60         }
  61     }
  62     wrbuf_destroy(w);
  63     odr_destroy(odr);
  64     return ret;
  65 }
  66
  67 static int compare(cql_transform_t ct, const char *pqf, const char *cql)
  68 {
  69     return compare2(ct, pqf, cql, 0);
  70 }
  71
  72 static void tst1(void)
  73 {
  74     cql_transform_t ct = cql_transform_create();
  75
  76     YAZ_CHECK(compare(ct, "abc", "abc"));
  77     YAZ_CHECK(compare(ct, "\"a b c\"", "\"a b c\""));
  78     YAZ_CHECK(compare(ct, "@and a b", "a and b"));
  79     YAZ_CHECK(compare(ct, "@or a @and b c", "a or (b and c)"));
  80     YAZ_CHECK(compare(ct, "@or @and a b @and c d", "(a and b) or (c and d)"));
  81     YAZ_CHECK(compare(ct, "@or @or a b @or c d", "(a or b) or (c or d)"));
  82     YAZ_CHECK(compare(ct, "@and @and a b @and c d", "(a and b) and (c and d)"));
  83
  84     YAZ_CHECK(compare(ct, "@attr 1=field abc", "field=abc"));
  85     YAZ_CHECK(compare2(ct, "@attr 1=4 abc", "4", 114)); /* should fail */
  86
  87     cql_transform_define_pattern(ct, "index.title", "1=4");
  88     YAZ_CHECK(compare(ct, "@attr 1=4 abc", "title=abc"));
  89
  90     cql_transform_define_pattern(ct, "index.foo", "1=bar");
  91     YAZ_CHECK(compare(ct, "@attr 1=bar abc", "foo=abc"));
  92     YAZ_CHECK(compare(ct, "@attr bib1 1=bar abc", "foo=abc"));
  93
  94     cql_transform_define_pattern(ct, "index.author", "bib1 1=1003 4=2");
  95     YAZ_CHECK(compare(ct, "@attr 4=2 @attr bib1 1=1003 abc", "author=abc"));
  96
  97     cql_transform_close(ct);
  98 }
  99
 100 static void tst2(void)
 101 {
 102     WRBUF w = wrbuf_alloc();
 103     cql_transform_t ct = 0;
 104     const char *srcdir = getenv("srcdir");
 105     if (srcdir)
 106     {
 107         wrbuf_puts(w, srcdir);
 108         wrbuf_puts(w, "/");
 109     }
 110     wrbuf_puts(w, "../etc/pqf.properties");
 111
 112     ct = cql_transform_open_fname(wrbuf_cstr(w));
 113     YAZ_CHECK(compare(ct, "@attr 1=4 abc", "dc.title=abc"));
 114     YAZ_CHECK(compare(ct, "@attr 1=4 @attr 4=108 abc", "dc.title=/exact abc"));
 115     YAZ_CHECK(compare(ct, "@attr 1=4 @attr 3=1 @attr 6=1 abc", "dc.title=abc"));
 116     YAZ_CHECK(compare(ct, "@attr 1=4 @attr 4=1 @attr 6=1 abc", "dc.title=abc"));
 117     YAZ_CHECK(compare(ct, "@attr 1=1016 abc", "abc"));
 118     /* Date tests */
 119     YAZ_CHECK(compare(ct, "@attr 1=30 @attr 2=1 1980", "dc.date<1980"));
 120     YAZ_CHECK(compare(ct, "@attr 1=30 @attr 2=2 1980", "dc.date<=1980"));
 121     YAZ_CHECK(compare(ct, "@attr 1=30 @attr 2=3 1980", "dc.date=1980"));
 122     YAZ_CHECK(compare(ct, "@attr 1=30 @attr 2=4 1980", "dc.date>=1980"));
 123     YAZ_CHECK(compare(ct, "@attr 1=30 @attr 2=5 1980", "dc.date>1980"));
 124     YAZ_CHECK(compare(ct, "@and @attr 1=30 @attr 2=4 234 @attr 1=30 @attr 2=2 1990", "dc.date>=234 and dc.date<=1990"));
 125
 126     /* Truncation */
 127     YAZ_CHECK(compare(ct, "@attr 5=1 water", "water*"));
 128     YAZ_CHECK(compare(ct, "@attr 5=2 water", "*water"));
 129     YAZ_CHECK(compare(ct, "@attr 5=3 water", "*water*"));
 130     YAZ_CHECK(compare(ct, "@attr 5=100 water", "water"));
 131     YAZ_CHECK(compare(ct, "@attr 5=102 water", "water"));
 132     YAZ_CHECK(compare(ct, "@attr 5=104 water", "water"));
 133
 134     YAZ_CHECK(compare(ct, "@attr 5=102 wat.*er", "wat*er"));
 135     YAZ_CHECK(compare(ct, "@attr 5=104 wat?er", "wat*er"));
 136
 137     YAZ_CHECK(compare(ct, "@attr 5=102 wat.er", "wat?er"));
 138     YAZ_CHECK(compare(ct, "@attr 5=104 wat#er", "wat?er"));
 139     YAZ_CHECK(compare(ct, "@attr 5=102 wat?er", "wat\\?er"));
 140     YAZ_CHECK(compare(ct, "@attr 5=104 wat*er", "wat\\*er"));
 141     YAZ_CHECK(compare(ct, "@attr 5=102 wat#er", "wat#er"));
 142
 143     /* \. is 'eaten' by PQF parser */
 144     YAZ_CHECK(compare(ct, "@attr 5=102 wat\\.er", "wat?er"));
 145
 146     /* Escape sequences */
 147     /* note: escape sequences that survive after PQF parse below */
 148     YAZ_CHECK(compare(ct, "@attr 5=102 wat\\\\?er", "wat\\?er"));
 149     YAZ_CHECK(compare(ct, "@attr 5=104 wat\\\\?er", "wat\\?er"));
 150
 151     YAZ_CHECK(compare(ct, "@attr 5=102 wat\\\\*er", "wat\\*er"));
 152     YAZ_CHECK(compare(ct, "@attr 5=104 wat\\\\*er", "wat\\*er"));
 153
 154     YAZ_CHECK(compare(ct, "wat\\\\#er", "wat#er"));
 155     YAZ_CHECK(compare(ct, "@attr 5=100 wat\\\\#er", "wat#er"));
 156     YAZ_CHECK(compare(ct, "@attr 5=102 wat\\\\#er", "wat#er"));
 157     YAZ_CHECK(compare(ct, "@attr 5=104 wat\\\\#er", "wat#er"));
 158     YAZ_CHECK(compare(ct, "@attr 5=102 wat\\\\.er", "wat.er"));
 159     YAZ_CHECK(compare(ct, "@attr 5=104 wat\\\\.er", "wat.er"));
 160
 161     /* Quoting */
 162     YAZ_CHECK(compare(ct, "@attr 5=100 \"\"", "\"\""));
 163     YAZ_CHECK(compare(ct, "@attr 5=1 \"\"", "\"*\""));
 164     YAZ_CHECK(compare(ct, "@attr 5=2 \"\"", "\"*\""));
 165     YAZ_CHECK(compare(ct, "@attr 5=3 \"\"", "\"**\""));
 166     YAZ_CHECK(compare(ct, "@attr 5=102 \"\"", "\"\""));
 167     YAZ_CHECK(compare(ct, "@attr 5=104 \"\"", "\"\""));
 168
 169     YAZ_CHECK(compare(ct, "@attr 5=1 \"water basket\"", "\"water basket*\""));
 170     YAZ_CHECK(compare(ct, "@attr 5=2 \"water basket\"", "\"*water basket\""));
 171     YAZ_CHECK(compare(ct, "@attr 5=3 \"water basket\"", "\"*water basket*\""));
 172
 173     /* Other */
 174     YAZ_CHECK(compare(ct, "@attr 2=103 @attr 1=_ALLRECORDS 1", "cql.allRecords=1"));
 175     YAZ_CHECK(compare2(ct, "@attr 1=500 abc", "500", 114));
 176     YAZ_CHECK(compare2(ct, "@attr 5=99 x", "99", 120));
 177     cql_transform_close(ct);
 178     wrbuf_destroy(w);
 179 }
 180
 181 int main (int argc, char **argv)
 182 {
 183     YAZ_CHECK_INIT(argc, argv);
 184     YAZ_CHECK_LOG();
 185     tst1();
 186     tst2();
 187     YAZ_CHECK_TERM;
 188 }
 189
 190 /*
 191  * Local variables:
 192  * c-basic-offset: 4
 193  * c-file-style: "Stroustrup"
 194  * indent-tabs-mode: nil
 195  * End:
 196  * vim: shiftwidth=4 tabstop=8 expandtab
 197  */
 198